Update app.py

app.py

@@ -14,7 +14,6 @@ import random
 from transformers import pipeline
 import torch
 import asyncio
-from nltk.tokenize import sent_tokenize
 
 nltk.download('punkt', quiet=True)
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
@@ -23,15 +22,12 @@ logger = logging.getLogger(__name__)
 PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")
 MODEL_NAME = "DeepESP/gpt2-spanish"
 
-async def get_voices():
-    try:
-        voices = await edge_tts.list_voices()
-        voice_names = [f"{v['Name']} ({v['Gender']}, {v['LocaleName']})" for v in voices]
-        return voice_names, voices
-    except:
-        return ["Voz Predeterminada (Femenino, es-ES)"], [{'ShortName': 'es-ES-ElviraNeural'}]
+# Obtener voces de edge-tts de forma síncrona (wrapper)
+def get_voices_sync():
+    return asyncio.run(edge_tts.list_voices())
 
-VOICE_NAMES, VOICES = asyncio.run(get_voices())
+VOICES = get_voices_sync()
+VOICE_NAMES = [f"{v['Name']} ({v['Gender']}, {v['LocaleName']})" for v in VOICES]
 
 def generar_guion_profesional(prompt):
     try:
@@ -47,10 +43,12 @@ def generar_guion_profesional(prompt):
             "2. Tres secciones detalladas con subtítulos\n"
             "3. Conclusión impactante\n"
             "Usa un estilo natural para narración:",
+            max_length=1000,
             temperature=0.7,
             top_k=50,
             top_p=0.95,
-            num_return_sequences=1
+            num_return_sequences=1,
+            truncation=True
         )
         guion = response[0]['generated_text']
         if len(guion.split()) < 100:
@@ -90,6 +88,8 @@ def generar_guion_profesional(prompt):
         ¿Listos para profundizar? ¡Empecemos!
         """
 
+from nltk.tokenize import sent_tokenize
+
 def buscar_videos_avanzado(prompt, guion, num_videos=5):
     try:
         oraciones = sent_tokenize(guion)
@@ -125,6 +125,7 @@ def buscar_videos_avanzado(prompt, guion, num_videos=5):
         return response.json().get('videos', [])[:num_videos]
 
 async def crear_video_profesional(prompt, custom_script, voz_index, musica=None):
+    voz_archivo = None
     try:
         guion = custom_script if custom_script else generar_guion_profesional(prompt)
         logger.info(f"Guion generado ({len(guion.split())} palabras)")
@@ -146,13 +147,6 @@ async def crear_video_profesional(prompt, custom_script, voz_index, musica=None)
                 clip = VideoFileClip(temp_video.name).subclip(0, min(10, video['duration']))
                 clips.append(clip)
         video_final = concatenate_videoclips(clips)
-        if musica:
-            musica_clip = AudioFileClip(musica.name)
-            if musica_clip.duration < duracion_total:
-                musica_clip = musica_clip.loop(duration=duracion_total)
-            else:
-                musica_clip = musica_clip.subclip(0, duracion_total)
-            audio = CompositeAudioClip([audio, musica_clip.volumex(0.25)])
         video_final = video_final.set_audio(audio)
         output_path = f"video_output_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
         video_final.write_videofile(output_path, fps=24, threads=2)
@@ -161,25 +155,26 @@ async def crear_video_profesional(prompt, custom_script, voz_index, musica=None)
         logger.error(f"Error crítico: {str(e)}")
         return None
     finally:
-        if os.path.exists("voz.mp3"):
-            os.remove("voz.mp3")
+        if voz_archivo and os.path.exists(voz_archivo):
+            os.remove(voz_archivo)
 
-def run_async_func(prompt, custom_script, voz_index, musica=None):
+def run_async_wrapper(prompt, custom_script, voz, musica):
+    voz_index = VOICE_NAMES.index(voz)
     return asyncio.run(crear_video_profesional(prompt, custom_script, voz_index, musica))
 
 with gr.Blocks(title="Generador de Videos") as app:
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(label="Tema del video")
-            custom_script = gr.TextArea(label="Guión personalizado (opcional)")
+            custom_script = gr.TextArea(label="Guion personalizado (opcional)")
             voz = gr.Dropdown(VOICE_NAMES, label="Voz", value=VOICE_NAMES[0])
             musica = gr.File(label="Música de fondo (opcional)", file_types=["audio"])
             btn = gr.Button("Generar Video", variant="primary")
         with gr.Column():
             output = gr.Video(label="Resultado", format="mp4")
-
+    
     btn.click(
-        fn=lambda p, cs, v, m: asyncio.run(crear_video_profesional(p, cs, VOICE_NAMES.index(v), m)),
+        fn=run_async_wrapper,
         inputs=[prompt, custom_script, voz, musica],
         outputs=output
     )