Update app.py

app.py

@@ -2,45 +2,57 @@ import gradio as gr
 from transformers import pipeline
 from gtts import gTTS
 from PIL import Image, ImageDraw, ImageFont
-import uuid
+import numpy as np
+import cv2
 import os
+import uuid
 
-# Папки
-os.makedirs("facts", exist_ok=True)
+# Создание папки
+os.makedirs("videos", exist_ok=True)
 
-# Фактогенерация
-fact_generator = pipeline("text-generation", model="sberbank-ai/rugpt3small_based_on_gpt2")
+# Генератор текстов
+generator = pipeline("text-generation", model="sberbank-ai/rugpt3small_based_on_gpt2")
+used_ideas = set()
 
-def generate():
-    # 1. Генерация текста
-    fact = fact_generator("Знаете ли вы, что", max_length=50, num_return_sequences=1)[0]["generated_text"]
-    fact = fact.split(".")[0] + "."
+def generate_video():
+    global used_ideas
 
-    # 2. Озвучка
-    audio_path = f"facts/{uuid.uuid4().hex}.mp3"
-    tts = gTTS(fact, lang="ru")
-    tts.save(audio_path)
+    # Генерация уникальной идеи
+    while True:
+        idea = generator("Идея для короткого видео: ", max_length=60, num_return_sequences=1)[0]['generated_text']
+        idea = idea.strip().split(".")[0]
+        if idea not in used_ideas and len(idea) > 20:
+            used_ideas.add(idea)
+            break
 
-    # 3. Картинка
-    image = Image.new("RGB", (1080, 1920), color="black")
-    draw = ImageDraw.Draw(image)
+    # Сохраняем аудио
+    audio_path = f"videos/{uuid.uuid4().hex}.mp3"
+    gTTS(idea, lang="ru").save(audio_path)
+
+    # Создаём изображение
+    img = Image.new("RGB", (1080, 1920), color="black")
+    draw = ImageDraw.Draw(img)
     font = ImageFont.load_default()
-    draw.text((50, 960), fact, font=font, fill="white")
-    image_path = audio_path.replace(".mp3", ".png")
-    image.save(image_path)
-
-    return fact, image_path, audio_path
-
-demo = gr.Interface(
-    fn=generate,
-    inputs=[],
-    outputs=[
-        gr.Textbox(label="Факт"),
-        gr.Image(label="Изображение"),
-        gr.Audio(label="Озвучка", type="filepath")
-    ],
-    title="TikTok-факты на русском",
-)
+    draw.text((50, 900), idea, font=font, fill="white")
+    img_np = np.array(img)
+
+    # Конвертируем в видео (6 сек, 30 fps)
+    video_path = f"videos/{uuid.uuid4().hex}.mp4"
+    out = cv2.VideoWriter(video_path, cv2.VideoWriter_fourcc(*'mp4v'), 30, (1080, 1920))
+
+    for _ in range(180):
+        out.write(cv2.cvtColor(img_np, cv2.COLOR_RGB2BGR))
+    out.release()
+
+    return video_path, idea
+
+# Интерфейс
+with gr.Blocks() as demo:
+    out_text = gr.Textbox(label="🧠 Идея")
+    out_video = gr.Video(label="🎥 Видео")
+    gen_btn = gr.Button("🎬 Сгенерировать")
+
+    gen_btn.click(fn=generate_video, inputs=[], outputs=[out_video, out_text])
 
 demo.launch()