Update app.py

app.py

@@ -2,15 +2,12 @@ import gradio as gr
 import moviepy.editor as mp
 import numpy as np
 from PIL import Image
+import tempfile
 import os
 import traceback
-import sys
-
-# Redirect logs for Hugging Face
-sys.stdout = sys.stderr
 
 # Resize image while maintaining aspect ratio
-def resize_and_fit_image(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
+def resize_image_with_aspect_ratio(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
     width, height = img.size
     target_width, target_height = target_size
 
@@ -26,91 +23,78 @@ def resize_and_fit_image(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
 
     img_resized = img.resize((new_width, new_height))
     final_img = Image.new('RGB', target_size, padding_color)
-    final_img.paste(img_resized, ((target_width - new_width) // 2, (target_height - new_height) // 2))
+    padding_left = (target_width - new_width) // 2
+    padding_top = (target_height - new_height) // 2
+    final_img.paste(img_resized, (padding_left, padding_top))
 
     return final_img
 
-# Apply zoom effect to image clip
-def apply_zoom_effect(image_clip):
-    return image_clip.resize(lambda t: 1 + 0.05 * t)  # Gradual zoom effect
-
 # Video generation function
 def process_and_generate_video(audio_file, images):
+    debug_log = []  # Log for debugging
+
     try:
-        print("Starting video generation...")
-
-        # Validate input files
-        if not audio_file:
-            raise ValueError("No audio file provided.")
-        if not images or len(images) == 0:
-            raise ValueError("No images provided.")
-
-        # Handle audio file path
-        if isinstance(audio_file, str):
-            audio_path = audio_file
-        else:
-            raise ValueError("Audio file is not a valid file path.")
-
-        # Load audio
-        print(f"Audio file: {audio_path}")
-        audio = mp.AudioFileClip(audio_path)
+        # Validate inputs
+        if not audio_file or not images:
+            raise ValueError("Both audio and images are required for video generation.")
+
+        # 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
+        audio = mp.AudioFileClip(audio_file)
         audio_duration = audio.duration
-        print(f"Audio duration: {audio_duration} seconds")
+        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:
-            print(f"Processing image: {img_path}")
-            if isinstance(img_path, str):
-                img = Image.open(img_path)
-            else:
-                raise ValueError("Image file is not a valid file path.")
-            
-            img = resize_and_fit_image(img, target_size=(1280, 720))
-            img_clip = mp.ImageClip(np.array(img)).set_duration(image_duration).set_fps(30)
-            img_clip = apply_zoom_effect(img_clip)
+            debug_log.append(f"Processing image: {img_path}")
+            img = Image.open(img_path)
+            resized_img = resize_image_with_aspect_ratio(img, target_size=(1280, 720))
+            img_clip = mp.ImageClip(np.array(resized_img)).set_duration(image_duration).set_fps(24)
             image_clips.append(img_clip)
 
-        print(f"Generated {len(image_clips)} image clips.")
+        debug_log.append(f"Created {len(image_clips)} image clips")
 
         # Concatenate clips
         video = mp.concatenate_videoclips(image_clips, method="compose")
         video = video.set_audio(audio)
 
-        # Save video
+        # Save video to a temporary path
         output_path = "/tmp/generated_video.mp4"
-        print(f"Saving video to {output_path}...")
         video.write_videofile(output_path, codec="libx264", audio_codec="aac")
+        debug_log.append(f"Video successfully saved to {output_path}")
 
-        print("Video generation completed successfully.")
-        return output_path
+        return output_path, "\n".join(debug_log)
 
     except Exception as e:
-        error_message = f"Error during video generation: {e}\nTraceback:\n{traceback.format_exc()}"
-        print(error_message)  # Log the detailed error
-        return error_message  # Return error as feedback
+        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)
 
 # 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")
-                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")
-            error_output = gr.Textbox(label="Error Log", interactive=False)
+                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
+
+            output_video = gr.Video(label="Generated Video")  # Display the output video
+            debug_logs = gr.Textbox(label="Debug Logs", interactive=False)  # Display debug logs
 
         generate_button.click(
             fn=process_and_generate_video,
             inputs=[mp3_input, image_input],
-            outputs=[output_video, error_output]
+            outputs=[output_video, debug_logs]
         )
-    
+
     demo.launch()
 
-# Run the Gradio interface
+# Run the Gradio app
 gradio_interface()