Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 from PIL import Image
 import os
 
+# Resize image while maintaining aspect ratio
 def resize_and_fit_image(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
     width, height = img.size
     target_width, target_height = target_size
@@ -29,47 +30,59 @@ def resize_and_fit_image(img, target_size=(1280, 720), padding_color=(0, 0, 0)):
 
     return final_img
 
+# Apply zoom effect to image clip
 def apply_zoom_effect(image_clip):
     zoomed_clip = image_clip.resize(lambda t: 1 + 0.05 * t)  # Zoom in gradually
     return zoomed_clip
 
 def process_and_generate_video(audio_file, images):
     try:
-        audio = mp.AudioFileClip(audio_file)
+        # Check if the file paths are strings, and convert to actual file-like objects
+        if isinstance(audio_file, str):
+            audio = mp.AudioFileClip(audio_file)
+        else:
+            raise ValueError("Expected a valid file path for audio.")
+
         audio_duration = audio.duration
         image_clips = []
         image_count = len(images)
         image_duration = audio_duration / image_count
-        
+
         print(f"Audio duration: {audio_duration} seconds, Image count: {image_count}")
-        
+
+        # Iterate over images, resize them, and create video clips
         for img_path in images:
-            img = Image.open(img_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)
-            
-            # Add transition effect - Crossfade In
-            if len(image_clips) > 0:
-                img_clip = img_clip.crossfadein(1)
-            
-            image_clips.append(img_clip)
-        
+            if isinstance(img_path, str):  # Ensure img_path is a string (file path)
+                img = Image.open(img_path)
+                img = resize_and_fit_image(img, target_size=(1280, 720))
+
+                # Create image clip and apply zoom effect
+                img_clip = mp.ImageClip(np.array(img)).set_duration(image_duration).set_fps(30)
+                img_clip = apply_zoom_effect(img_clip)
+
+                # Add transition effect - Crossfade In
+                if len(image_clips) > 0:
+                    img_clip = img_clip.crossfadein(1)
+
+                image_clips.append(img_clip)
+
         print(f"Image clips: {len(image_clips)} clips created.")
-        
+
+        # Concatenate image clips with audio
         video = mp.concatenate_videoclips(image_clips, method="compose")
         video = video.set_audio(audio)
-        
+
+        # Set output path
         output_path = '/content/generated_video.mp4'
         video.write_videofile(output_path, codec='libx264', audio_codec='aac', threads=4, fps=30, preset='ultrafast')
-        
+
         return output_path  # Return the file path for Gradio output
-    
+
     except Exception as e:
         print(f"Error during video generation: {str(e)}")
         return f"Error generating video: {str(e)}"
 
+# Gradio interface setup
 def gradio_interface():
     with gr.Blocks() as demo:
         with gr.Row():
@@ -80,6 +93,7 @@ def gradio_interface():
                 
             output_video = gr.Video(label="Generated Video")
         
+        # Generate button triggers video creation process
         generate_button.click(fn=process_and_generate_video, inputs=[mp3_input, image_input], outputs=output_video)
     
     demo.launch()