Update app.py

app.py

@@ -10,15 +10,22 @@ import gradio as gr
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 import torch
 
-from pexels_api import API 
+# --- NO MÁS 'from pexels_api import API' ---
+# Ahora Pexels se gestiona directamente con requests
+# from pexels_api import API 
+
+# --- Importar KeyBERT para extracción de palabras clave ---
 from keybert import KeyBERT 
 
+# --- Configuración de Logging ---
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
+# --- Clave API de Pexels ---
 PEXELS_API_KEY = os.environ.get("PEXELS_API_KEY") 
-pexels_api = API(PEXELS_API_KEY) 
+# Ya no se inicializa pexels_api = API(PEXELS_API_KEY)
 
+# --- Inicialización de Tokenizer y Modelo GPT-2 ---
 MODEL_NAME = "gpt2-small" 
 try:
     tokenizer = GPT2Tokenizer.from_pretrained(MODEL_NAME)
@@ -31,6 +38,7 @@ except Exception as e:
     tokenizer = None
     model = None
 
+# --- Inicialización del modelo KeyBERT ---
 try:
     kw_model = KeyBERT('multi-qa-MiniLM-L6-cos-v1') 
     logger.info("Modelo KeyBERT cargado exitosamente.")
@@ -38,6 +46,8 @@ except Exception as e:
     logger.error(f"Error al cargar el modelo KeyBERT: {e}. La búsqueda de videos será menos precisa.", exc_info=True)
     kw_model = None
 
+# --- Funciones de Utilidad ---
+
 def generate_script(prompt, max_length=250): 
     if not tokenizer or not model:
         logger.error("Modelo GPT-2 no disponible para generar guion.")
@@ -151,6 +161,7 @@ def extract_visual_keywords_from_script(script_text, max_keywords_per_segment=2)
     logger.info(f"Palabras clave visuales extraídas del guion: {list(all_keywords)}")
     return list(all_keywords) 
 
+# --- Búsqueda de videos en Pexels (AHORA CON REQUESTS DIRECTOS) ---
 def search_pexels_videos(query_list, num_videos_per_query=5, min_duration_sec=7):
     if not PEXELS_API_KEY:
         logger.error("ERROR: PEXELS_API_KEY no está configurada. No se pueden buscar videos en Pexels.")
@@ -165,32 +176,46 @@ def search_pexels_videos(query_list, num_videos_per_query=5, min_duration_sec=7)
     for query in query_list:
         logger.info(f"Buscando {num_videos_per_query} videos en Pexels para la query: '{query}'")
         try:
-            pexels_api.search_videos(query, per_page=num_videos_per_query * 2) 
-            videos = pexels_api.get_videos()
+            # URL de la API de búsqueda de Pexels
+            # https://www.pexels.com/api/documentation/#videos-search
+            api_url = "https://api.pexels.com/videos/search"
+            headers = {"Authorization": PEXELS_API_KEY}
+            params = {
+                "query": query,
+                "per_page": num_videos_per_query * 2, # Pedimos el doble por si algunos no cumplen duración
+                "orientation": "landscape", # Opcional: landscape, portrait, square
+                "min_duration": min_duration_sec
+            }
+            
+            response = requests.get(api_url, headers=headers, params=params, timeout=15)
+            response.raise_for_status() # Lanza un error si la solicitud HTTP no fue exitosa (código 4xx o 5xx)
+            
+            data = response.json()
+            videos = data.get('videos', [])
 
             if not videos:
                 logger.info(f"No se encontraron videos en Pexels para la query: '{query}'")
                 continue
 
             for video in videos:
-                if video.get('duration') and video['duration'] < min_duration_sec:
-                    continue
-
                 best_quality_url = None
-                for file in video.get('video_files', []):
-                    if file.get('quality') in ['hd', 'sd', 'medium', 'large'] and file.get('link'):
+                # Ordenar por calidad para obtener la mejor disponible
+                video_files = sorted(video.get('video_files', []), key=lambda x: x.get('width', 0) * x.get('height', 0), reverse=True)
+                
+                for file in video_files:
+                    if file.get('link'):
                         best_quality_url = file['link']
                         break 
-                    elif file.get('link'):
-                        best_quality_url = file['link']
 
                 if best_quality_url:
                     all_video_urls.add(best_quality_url)
                 else:
                     logger.warning(f"No se encontró URL de descarga válida para el video Pexels ID: {video.get('id')}")
 
+        except requests.exceptions.RequestException as e:
+            logger.error(f"Error de HTTP/red al buscar videos en Pexels para '{query}': {e}", exc_info=True)
         except Exception as e:
-            logger.error(f"Error al buscar videos en Pexels para '{query}': {e}", exc_info=True)
+            logger.error(f"Error inesperado al buscar videos en Pexels para '{query}': {e}", exc_info=True)
             
     final_urls = list(all_video_urls)
     logger.info(f"Total de {len(final_urls)} URLs de video únicas obtenidas de Pexels.")