Update app.py

app.py

@@ -8,11 +8,15 @@ import logging
 import re
 import torch
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
+import warnings
 
 # Configuración inicial
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Suprimir warnings específicos de transformers
+warnings.filterwarnings("ignore", category=UserWarning, module="transformers")
+
 PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")
 
 # Lista de voces válidas
@@ -58,7 +62,7 @@ def generar_guion_largo(tema, custom_script=None):
             top_p=0.95,
             num_return_sequences=1,
             pad_token_id=tokenizer.eos_token_id,
-            early_stopping=True
+            # early_stopping=True  # Eliminado para evitar warnings
         )
         
         guion = tokenizer.decode(outputs[0], skip_special_tokens=True)
@@ -87,14 +91,11 @@ def buscar_videos_pexels(tema, num_videos=4):
             headers=headers,
             timeout=15
         )
-        videos = response.json().get("videos", [])
         
-        # Filtrar videos de alta calidad
-        return sorted(
-            videos,
-            key=lambda x: (x.get('width', 0) * x.get('height', 0), x.get('duration', 0)),
-            reverse=True
-        )[:num_videos]
+        if response.status_code != 200:
+            return []
+            
+        return response.json().get("videos", [])[:num_videos]
     except Exception as e:
         logger.error(f"Error buscando videos: {str(e)}")
         return []
@@ -142,6 +143,9 @@ def crear_video(prompt, custom_script, voz_seleccionada, musica=None):
         for i, video in enumerate(videos_data):
             try:
                 # Seleccionar la mejor calidad disponible
+                if 'video_files' not in video or not video['video_files']:
+                    continue
+                    
                 video_file = max(
                     video['video_files'], 
                     key=lambda x: x.get('width', 0) * x.get('height', 0)
@@ -166,9 +170,11 @@ def crear_video(prompt, custom_script, voz_seleccionada, musica=None):
                 logger.error(f"Error procesando video {i}: {str(e)}")
         
         if not clips:
-            raise Exception("No se pudieron cargar videos válidos")
+            # Crear video de fondo negro si no hay videos
+            clip = ColorClip(size=(1280, 720), color=(0, 0, 0), duration=duracion_total)
+            clips = [clip]
         
-        # 6. Combinar videos con transiciones suaves
+        # 6. Combinar videos
         final_clip = concatenate_videoclips(clips, method="compose")
         final_clip = final_clip.set_audio(audio)
         
@@ -197,8 +203,7 @@ def crear_video(prompt, custom_script, voz_seleccionada, musica=None):
             codec="libx264",
             audio_codec="aac",
             threads=4,
-            preset='fast',
-            ffmpeg_params=['-crf', '23']  # Calidad balanceada
+            preset='fast'
         )
         
         logger.info(f"Video generado exitosamente: {output_path}")
@@ -216,7 +221,7 @@ def crear_video(prompt, custom_script, voz_seleccionada, musica=None):
             if os.path.exists(temp_file):
                 os.remove(temp_file)
 
-# Interfaz mejorada sin enable_queue
+# Interfaz mejorada
 with gr.Blocks(title="Generador de Videos", theme=gr.themes.Soft()) as app:
     gr.Markdown("# 🎬 GENERADOR AUTOMÁTICO DE VIDEOS")