Update app.py

app.py

@@ -94,17 +94,22 @@ def create_video_from_images_ffmpeg_sequence(processed_image_folder, num_images,
         video_filename_temp
     ]
     
+    print(f"DEBUG: FFmpeg image sequence command: {' '.join(command)}")
     try:
         result = subprocess.run(command, check=True, capture_output=True, text=True)
-        print("FFmpeg Image Sequence stdout:", result.stdout)
-        print("FFmpeg Image Sequence stderr:", result.stderr)
+        print("DEBUG: FFmpeg Image Sequence stdout:", result.stdout)
+        print("DEBUG: FFmpeg Image Sequence stderr:", result.stderr)
     except subprocess.CalledProcessError as e:
+        print(f"DEBUG: FFmpeg Image Sequence CalledProcessError: {e.stderr}")
         raise Exception(f"Error al crear video desde imágenes con FFmpeg. Salida de error: {e.stderr}")
     except FileNotFoundError:
+        print("DEBUG: FFmpeg not found during image sequence creation.")
         raise Exception("Error: FFmpeg no encontrado. Asegúrate de que esté instalado en el entorno.")
 
     if not os.path.exists(video_filename_temp):
+        print(f"DEBUG: {video_filename_temp} does not exist after FFmpeg run.")
         raise IOError(f"Error: FFmpeg no creó el archivo de video {video_filename_temp} desde la secuencia de imágenes.")
+    print(f"DEBUG: {video_filename_temp} successfully created by FFmpeg.")
     return video_filename_temp
 
 
@@ -122,29 +127,22 @@ def combine_video_and_audio_ffmpeg(video_path, audio_path, output_path):
         output_path
     ]
     
+    print(f"DEBUG: FFmpeg combine command: {' '.join(command)}")
     try:
         result = subprocess.run(command, check=True, capture_output=True, text=True)
-        print("FFmpeg combine stdout:", result.stdout)
-        print("FFmpeg combine stderr:", result.stderr)
+        print("DEBUG: FFmpeg combine stdout:", result.stdout)
+        print("DEBUG: FFmpeg combine stderr:", result.stderr)
     except subprocess.CalledProcessError as e:
+        print(f"DEBUG: FFmpeg combine CalledProcessError: {e.stderr}")
         raise Exception(f"Error al combinar video y audio con FFmpeg. Salida de error: {e.stderr}")
     except FileNotFoundError:
+        print("DEBUG: FFmpeg not found during combine.")
         raise Exception("Error: FFmpeg no encontrado. Asegúrate de que esté instalado en el entorno.")
 
-def create_news_video_app(news_text_input, gemini_prompt, image_files, video_ratio):
+def generate_tts_only(news_text_input, gemini_prompt):
     news_content = ""
-    output_video_path = None
-    error_message = ""
-
-    temp_audio_file = "audio_temp.mp3"
-    temp_video_no_audio = "video_no_audio_temp.mp4"
-    processed_image_folder = "temp_processed_images_for_video"
-    final_output_video_path = "output_news_video_final.mp4"
+    temp_audio_file = "audio_temp_preview.mp3"
 
-    if os.path.exists(processed_image_folder):
-        shutil.rmtree(processed_image_folder)
-    os.makedirs(processed_image_folder, exist_ok=True)
-    
     if news_text_input:
         news_content = news_text_input
     elif gemini_prompt:
@@ -155,10 +153,44 @@ def create_news_video_app(news_text_input, gemini_prompt, image_files, video_rat
         return "Por favor, escribe una noticia o proporciona un prompt para Gemini.", None
 
     if not news_content:
-        return "No se pudo obtener el contenido de la noticia. Intenta con un prompt diferente o escribe directamente.", None
+        return "No se pudo obtener el contenido de la noticia.", None
 
     try:
         text_to_speech(news_content, output_filename=temp_audio_file)
+        return "Audio generado con éxito.", temp_audio_file
+    except Exception as e:
+        return f"Ocurrió un error al generar solo el audio: {e}", None
+
+def create_news_video_app(news_text_input, gemini_prompt, image_files, video_ratio, input_audio_file=None):
+    news_content = ""
+    output_video_path = None
+    error_message = ""
+
+    temp_audio_file = input_audio_file if input_audio_file and os.path.exists(input_audio_file) else "audio_temp_for_video.mp3"
+    temp_video_no_audio = "video_no_audio_temp.mp4"
+    processed_image_folder = "temp_processed_images_for_video"
+    final_output_video_path = "output_news_video_final.mp4"
+
+    if os.path.exists(processed_image_folder):
+        shutil.rmtree(processed_image_folder)
+    os.makedirs(processed_image_folder, exist_ok=True)
+    
+    try:
+        if not input_audio_file or not os.path.exists(input_audio_file):
+            if news_text_input:
+                news_content = news_text_input
+            elif gemini_prompt:
+                news_content = generate_news_with_gemini(gemini_prompt)
+                if "Error" in news_content:
+                    return news_content, None
+            else:
+                return "Por favor, escribe una noticia, proporciona un prompt para Gemini, o genera el audio primero.", None
+
+            if not news_content:
+                return "No se pudo obtener el contenido de la noticia. Intenta con un prompt diferente o escribe directamente.", None
+            
+            text_to_speech(news_content, output_filename=temp_audio_file)
+
         audio_duration = get_audio_duration(temp_audio_file)
 
         if not image_files:
@@ -189,10 +221,14 @@ def create_news_video_app(news_text_input, gemini_prompt, image_files, video_rat
 
     except Exception as e:
         error_message = f"Ocurrió un error: {e}"
+        print(f"DEBUG: Error caught in main app function: {e}")
         return error_message, None
     finally:
-        if os.path.exists(temp_audio_file):
+        # Clean up all temporary files
+        if os.path.exists(temp_audio_file) and temp_audio_file != input_audio_file:
             os.remove(temp_audio_file)
+        if os.path.exists("audio_temp_preview.mp3"):
+             os.remove("audio_temp_preview.mp3") # Clean up preview audio if it exists
         if os.path.exists(temp_video_no_audio):
             os.remove(temp_video_no_audio)
         if os.path.exists(processed_image_folder):
@@ -222,27 +258,39 @@ with gr.Blocks() as demo:
                 placeholder="Ej: Escribe una noticia sobre un nuevo récord mundial de ajedrez.",
                 value=""
             )
+            
+            # Botón intermedio para generar solo audio
+            generate_audio_button = gr.Button("1. Generar Solo Audio")
+            audio_output_preview = gr.Audio(label="Audio de Noticia (Vista Previa)", interactive=False)
+            audio_status_message = gr.Textbox(label="Estado del Audio", interactive=False)
+
             image_upload = gr.File(
-                label="Sube de 1 a 60 imágenes (JPG, PNG)",
+                label="2. Sube de 1 a 60 imágenes (JPG, PNG)",
                 file_count="multiple",
                 type="filepath",
                 file_types=[".jpg", ".jpeg", ".png"]
             )
             video_ratio_dropdown = gr.Dropdown(
-                label="Relación de Aspecto del Video",
+                label="3. Relación de Aspecto del Video",
                 choices=["16:9", "9:16"],
                 value="16:9",
                 interactive=True
             )
-            generate_button = gr.Button("Generar Video de Noticia")
+            generate_video_button = gr.Button("4. Generar Video de Noticia")
         
         with gr.Column():
-            output_message = gr.Textbox(label="Estado", interactive=False)
+            output_message = gr.Textbox(label="Estado del Video", interactive=False)
             video_output = gr.Video(label="Video de la Noticia")
 
-    generate_button.click(
+    generate_audio_button.click(
+        fn=generate_tts_only,
+        inputs=[news_input, gemini_prompt_input],
+        outputs=[audio_status_message, audio_output_preview]
+    )
+
+    generate_video_button.click(
         fn=create_news_video_app,
-        inputs=[news_input, gemini_prompt_input, image_upload, video_ratio_dropdown],
+        inputs=[news_input, gemini_prompt_input, image_upload, video_ratio_dropdown, audio_output_preview],
         outputs=[output_message, video_output]
     )