Update app.py

app.py

@@ -35,7 +35,7 @@ except Exception as e:
     tokenizer = None
 
 def generar_guion_largo(tema, custom_script=None):
-    """Genera un texto largo sobre el tema usando GPT-2"""
+    """Genera un texto largo sobre el tema usando GPT-2 con configuración correcta"""
     if custom_script:
         return custom_script
         
@@ -48,15 +48,17 @@ def generar_guion_largo(tema, custom_script=None):
         
         inputs = tokenizer(prompt, return_tensors="pt", max_length=512, truncation=True)
         
-        # Generar texto
+        # Generar texto con configuración corregida
         outputs = model.generate(
             inputs.input_ids,
             max_length=800,
+            do_sample=True,  # Añadido para habilitar temperature y top_p
             temperature=0.9,
             top_k=50,
             top_p=0.95,
             num_return_sequences=1,
-            pad_token_id=tokenizer.eos_token_id
+            pad_token_id=tokenizer.eos_token_id,
+            early_stopping=True
         )
         
         guion = tokenizer.decode(outputs[0], skip_special_tokens=True)
@@ -64,6 +66,7 @@ def generar_guion_largo(tema, custom_script=None):
         # Limpiar texto
         guion = re.sub(r'<.*?>', '', guion)
         guion = re.sub(r'\n+', '\n', guion)
+        guion = re.sub(r'\s+', ' ', guion).strip()
         
         logger.info(f"Guion generado: {len(guion)} caracteres")
         return guion
@@ -165,19 +168,37 @@ def crear_video(prompt, custom_script, voz_seleccionada, musica=None):
         if not clips:
             raise Exception("No se pudieron cargar videos válidos")
         
-        # 6. Combinar videos
-        video_final = concatenate_videoclips(clips)
-        video_final = video_final.set_audio(audio)
+        # 6. Combinar videos con transiciones suaves
+        final_clip = concatenate_videoclips(clips, method="compose")
+        final_clip = final_clip.set_audio(audio)
         
-        # 7. Exportar video final
+        # 7. Aplicar música de fondo si existe
+        if musica:
+            try:
+                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_final = CompositeAudioClip([
+                    audio.volumex(1.0),
+                    musica_clip.volumex(0.25)
+                ])
+                final_clip = final_clip.set_audio(audio_final)
+            except Exception as e:
+                logger.error(f"Error procesando música: {str(e)}")
+        
+        # 8. Exportar video final
         output_path = f"video_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
-        video_final.write_videofile(
+        final_clip.write_videofile(
             output_path,
             fps=24,
             codec="libx264",
             audio_codec="aac",
             threads=4,
-            preset='fast'
+            preset='fast',
+            ffmpeg_params=['-crf', '23']  # Calidad balanceada
         )
         
         logger.info(f"Video generado exitosamente: {output_path}")
@@ -195,24 +216,42 @@ def crear_video(prompt, custom_script, voz_seleccionada, musica=None):
             if os.path.exists(temp_file):
                 os.remove(temp_file)
 
-# Interfaz simplificada y funcional
-with gr.Blocks(title="Generador de Videos") as app:
+# Interfaz mejorada
+with gr.Blocks(title="Generador de Videos", theme=gr.themes.Soft()) as app:
     gr.Markdown("# 🎬 GENERADOR AUTOMÁTICO DE VIDEOS")
     
     with gr.Row():
         with gr.Column():
-            prompt = gr.Textbox(label="Tema del video", placeholder="Ej: 'La historia de la inteligencia artificial'")
+            prompt = gr.Textbox(
+                label="Tema del video", 
+                placeholder="Ej: 'La historia de la piratería en el Caribe'",
+                max_lines=1
+            )
             custom_script = gr.TextArea(
                 label="Guion personalizado (opcional)", 
                 placeholder="Pega tu guion completo aquí...",
-                lines=5
+                lines=8,
+                max_lines=20
+            )
+            voz = gr.Dropdown(
+                label="Voz Narradora", 
+                choices=VOICES, 
+                value=VOICES[0],
+                interactive=True
+            )
+            musica = gr.File(
+                label="Música de fondo (opcional)", 
+                file_types=["audio"],
+                type="filepath"
             )
-            voz = gr.Dropdown(label="Voz Narradora", choices=VOICES, value=VOICES[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="Video Resultado", format="mp4")
+            output = gr.Video(
+                label="Video Resultado", 
+                format="mp4",
+                interactive=False
+            )
     
     btn.click(
         fn=crear_video,
@@ -221,4 +260,8 @@ with gr.Blocks(title="Generador de Videos") as app:
     )
 
 if __name__ == "__main__":
-    app.launch(server_name="0.0.0.0", server_port=7860)
+    app.launch(
+        server_name="0.0.0.0", 
+        server_port=7860,
+        enable_queue=True
+    )