Update app.py

app.py

@@ -2,85 +2,101 @@ import os
 import gradio as gr
 import requests
 import tempfile
-import logging
-from datetime import datetime
-
-import edge_tts
-import nest_asyncio
 import asyncio
-
+import edge_tts
+from datetime import datetime
+from moviepy.editor import (
+    VideoFileClip, AudioFileClip,
+    concatenate_videoclips, CompositeAudioClip, afx
+)
 from transformers import pipeline
-from moviepy.editor import AudioFileClip, VideoFileClip, concatenate_videoclips
+import nest_asyncio
 
 nest_asyncio.apply()
 
-PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")
-logger = logging.getLogger(__name__)
-
-# Carga modelo
-generator = pipeline("text-generation", model="DeepESP/gpt2-spanish")
-
-# Cargar voces
-loop = asyncio.get_event_loop()
-VOCES = loop.run_until_complete(edge_tts.list_voices())
-VOICE_NAMES = [f"{v['Name']} ({v['Gender']})" for v in VOCES]
-
-def generar_guion(prompt):
-    prompt_text = f"Escribe un guion profesional de YouTube sobre '{prompt}':\n"
-    res = generator(prompt_text, max_new_tokens=300)[0]['generated_text']
-    return res.strip()
-
-def buscar_videos(prompt, max_videos=3):
-    headers = {"Authorization": PEXELS_API_KEY}
-    r = requests.get(f"https://api.pexels.com/videos/search?query={prompt}&per_page={max_videos}", headers=headers)
-    return r.json().get("videos", [])
-
-def crear_video(prompt, voz_nombre):
+# Generador de texto
+generador = pipeline("text-generation", model="gpt2")
+
+# Obtener voces disponibles
+async def get_voices():
+    voices = await edge_tts.list_voices()
+    return voices
+
+VOICES = asyncio.run(get_voices())
+VOICE_OPTIONS = [
+    f"{v['Name']} ({v['Gender']}, {v['Locale']})" for v in VOICES
+]
+VOICE_MAP = {v['Name']: v['ShortName'] for v in VOICES}
+
+# Mock de videos
+def buscar_videos_mock():
+    return [
+        "https://samplelib.com/lib/preview/mp4/sample-5s.mp4",
+        "https://samplelib.com/lib/preview/mp4/sample-10s.mp4"
+    ]
+
+# Mock de música
+def buscar_musica_mock():
+    return "https://samplelib.com/lib/preview/mp3/sample-3s.mp3"
+
+# Función principal
+async def generar_video(prompt, voz_str):
     try:
-        idx = VOICE_NAMES.index(voz_nombre)
-    except:
-        idx = 0
-
-    voz_short = VOCES[idx]['ShortName']
-    guion = generar_guion(prompt)
-
-    # Guardar voz
-    tts = edge_tts.Communicate(guion, voz_short)
-    voz_path = "voz.mp3"
-    loop.run_until_complete(tts.save(voz_path))
-
-    audio_clip = AudioFileClip(voz_path)
-    duracion = audio_clip.duration
-
-    # Descargar videos
-    videos = buscar_videos(prompt)
-    clips = []
-    for v in videos:
-        url = v['video_files'][0]['link']
-        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-        r = requests.get(url, stream=True)
-        for c in r.iter_content(1024*1024):
-            tmp.write(c)
-        tmp.close()
-        clip = VideoFileClip(tmp.name).subclip(0, min(v['duration'], duracion / len(videos)))
-        clips.append(clip)
-
-    final = concatenate_videoclips(clips).set_audio(audio_clip)
-    salida = f"video_{datetime.now().strftime('%H%M%S')}.mp4"
-    final.write_videofile(salida, fps=24)
-
-    return salida
-
-def run(prompt, voz):
-    return crear_video(prompt, voz)
-
-with gr.Blocks() as app:
-    entrada = gr.Textbox(label="Tema del video")
-    voz = gr.Dropdown(VOICE_NAMES, label="Voz", value=VOICE_NAMES[0])
-    boton = gr.Button("Generar Video")
+        # 1. Texto largo
+        texto = generador(prompt, max_length=300, do_sample=True)[0]['generated_text']
+
+        # 2. Voz
+        voz_id = voz_str.split(" ")[0]
+        short_name = VOICE_MAP.get(voz_id, "es-ES-ElviraNeural")
+        voz_path = "voz.mp3"
+        await edge_tts.Communicate(text=texto, voice=short_name).save(voz_path)
+        voz_clip = AudioFileClip(voz_path)
+
+        # 3. Videos
+        video_urls = buscar_videos_mock()
+        clips = []
+        for url in video_urls:
+            r = requests.get(url, stream=True)
+            with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as f:
+                for chunk in r.iter_content(1024 * 1024):
+                    f.write(chunk)
+                f.flush()
+                clips.append(VideoFileClip(f.name).subclip(0, 5))
+
+        video = concatenate_videoclips(clips).set_audio(voz_clip)
+
+        # 4. Música (loop)
+        music_url = buscar_musica_mock()
+        r = requests.get(music_url, stream=True)
+        with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as f:
+            for chunk in r.iter_content(1024 * 1024):
+                f.write(chunk)
+            f.flush()
+            music_clip = AudioFileClip(f.name)
+            music_loop = afx.audio_loop(music_clip, duration=video.duration).volumex(0.3)
+
+        # 5. Composición de audio
+        final_audio = CompositeAudioClip([video.audio, music_loop])
+        video = video.set_audio(final_audio)
+
+        # 6. Guardar
+        output_path = f"video_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
+        video.write_videofile(output_path, fps=24)
+
+        return output_path
+    except Exception as e:
+        return f"Error: {e}"
+
+# Interfaz Gradio
+with gr.Blocks() as demo:
+    prompt = gr.Textbox(label="Prompt", placeholder="Escribe un tema...")
+    voz_dropdown = gr.Dropdown(VOICE_OPTIONS, label="Voz", value=VOICE_OPTIONS[0])
+    btn = gr.Button("Generar video")
     salida = gr.Video(label="Resultado")
 
-    boton.click(fn=run, inputs=[entrada, voz], outputs=salida)
+    btn.click(fn=lambda p, v: asyncio.run(generar_video(p, v)),
+              inputs=[prompt, voz_dropdown],
+              outputs=salida)
 
 if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860)