Update app.py

app.py

@@ -2,9 +2,7 @@ import gradio as gr
 import moviepy.editor as mp
 import numpy as np
 from PIL import Image
-import tempfile
 import os
-import traceback
 
 # Resize image while maintaining aspect ratio
 def resize_image_with_aspect_ratio(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
@@ -29,27 +27,20 @@ def resize_image_with_aspect_ratio(img, target_size=(1280, 720), padding_color=(
 
     return final_img
 
-# Video generation function
+# Video generation function with debug logging
 def process_and_generate_video(audio_file, images):
-    debug_log = []  # Log for debugging
+    debug_log = []
 
     try:
-        # Validate inputs
-        if not audio_file or not images:
-            raise ValueError("Both audio and images are required for video generation.")
+        debug_log.append(f"Audio file received: {audio_file}")
+        debug_log.append(f"Images received: {images}")
 
-        # Log received inputs
-        debug_log.append(f"Received audio file: {audio_file}")
-        debug_log.append(f"Received images: {[img for img in images]}")
-
-        # Load audio file
+        # Process audio file
         audio = mp.AudioFileClip(audio_file)
         audio_duration = audio.duration
-        debug_log.append(f"Audio duration: {audio_duration:.2f} seconds")
-
-        # Process images
         image_clips = []
         image_duration = audio_duration / len(images)
+
         for img_path in images:
             debug_log.append(f"Processing image: {img_path}")
             img = Image.open(img_path)
@@ -57,44 +48,39 @@ def process_and_generate_video(audio_file, images):
             img_clip = mp.ImageClip(np.array(resized_img)).set_duration(image_duration).set_fps(24)
             image_clips.append(img_clip)
 
-        debug_log.append(f"Created {len(image_clips)} image clips")
+        debug_log.append(f"Created {len(image_clips)} image clips.")
 
-        # Concatenate clips
+        # Concatenate image clips
         video = mp.concatenate_videoclips(image_clips, method="compose")
         video = video.set_audio(audio)
 
-        # Save video to a temporary path
-        output_path = "/tmp/generated_video.mp4"
-        video.write_videofile(output_path, codec="libx264", audio_codec="aac")
-        debug_log.append(f"Video successfully saved to {output_path}")
+        # Output video path
+        output_path = '/tmp/generated_video.mp4'
+        video.write_videofile(output_path, codec='libx264', audio_codec='aac')
 
-        return output_path, "\n".join(debug_log)
+        debug_log.append(f"Video successfully generated: {output_path}")
+        print("\n".join(debug_log))  # Print logs to the console for debugging
+        return output_path
 
     except Exception as e:
-        error_message = f"Error generating video: {str(e)}"
-        debug_log.append(error_message)
-        debug_log.append(traceback.format_exc())
-        return None, "\n".join(debug_log)
+        debug_log.append(f"Error: {str(e)}")
+        print("\n".join(debug_log))  # Print logs to the console for debugging
+        return f"Error: {str(e)}"
 
 # Gradio interface setup
 def gradio_interface():
     with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column():
-                mp3_input = gr.Audio(type="filepath", label="Upload MP3")  # MP3 input
-                image_input = gr.File(type="filepath", file_types=[".jpg", ".png"], label="Upload Images", file_count="multiple")  # Image inputs
-                generate_button = gr.Button("Generate Video")  # Button to generate video
+                mp3_input = gr.Audio(type="filepath", label="Upload MP3")
+                image_input = gr.File(type="filepath", file_types=[".jpg", ".png"], label="Upload Images", file_count="multiple")
+                generate_button = gr.Button("Generate Video")
 
-            output_video = gr.Video(label="Generated Video")  # Display the output video
-            debug_logs = gr.Textbox(label="Debug Logs", interactive=False)  # Display debug logs
+            output_video = gr.Video(label="Generated Video")
 
-        generate_button.click(
-            fn=process_and_generate_video,
-            inputs=[mp3_input, image_input],
-            outputs=[output_video, debug_logs]
-        )
+        generate_button.click(fn=process_and_generate_video, inputs=[mp3_input, image_input], outputs=output_video)
 
     demo.launch()
 
-# Run the Gradio app
+# Run the interface
 gradio_interface()