Update app.py

app.py

@@ -28,12 +28,14 @@ def resize_image_with_aspect_ratio(img, target_size=(1280, 720), padding_color=(
 
     return final_img
 
-# Video generation function with transition
+# Video generation function with transition and debug logging
 def process_and_generate_video(audio_file, images):
+    debug_log = []
+
     try:
         # Log the files received
-        print(f"Audio file received: {audio_file.name}")
-        print(f"Images received: {[img.name for img in images]}")
+        debug_log.append(f"Audio file received: {audio_file.name}")
+        debug_log.append(f"Images received: {[img.name for img in images]}")
 
         # Create temporary directory for images
         with tempfile.TemporaryDirectory() as temp_dir:
@@ -46,7 +48,7 @@ def process_and_generate_video(audio_file, images):
                 image_paths.append(temp_image_path)
 
             # Log image paths for debugging
-            print(f"Saved images in temporary directory: {image_paths}")
+            debug_log.append(f"Saved images in temporary directory: {image_paths}")
 
             # Process audio file
             audio = mp.AudioFileClip(audio_file.name)  # Use .name for filepath
@@ -55,11 +57,11 @@ def process_and_generate_video(audio_file, images):
             image_count = len(image_paths)
             image_duration = audio_duration / image_count
 
-            print(f"Audio duration: {audio_duration} seconds, Image count: {image_count}")
+            debug_log.append(f"Audio duration: {audio_duration} seconds, Image count: {image_count}")
             
             # Process each image and create video clip
             for img_path in image_paths:
-                print(f"Processing image: {img_path}")  # Debug print
+                debug_log.append(f"Processing image: {img_path}")  # Debug print
                 img = Image.open(img_path)
                 img = resize_image_with_aspect_ratio(img, target_size=(1280, 720))
                 
@@ -72,8 +74,7 @@ def process_and_generate_video(audio_file, images):
                 
                 image_clips.append(img_clip)
 
-            # Log the number of image clips created
-            print(f"Created {len(image_clips)} image clips.")
+            debug_log.append(f"Created {len(image_clips)} image clips.")
 
             # Concatenate image clips with transitions
             video = mp.concatenate_videoclips(image_clips, method="compose")
@@ -82,23 +83,22 @@ def process_and_generate_video(audio_file, images):
             # Set output file path in a temporary location
             output_path = '/tmp/generated_video.mp4'  # Temporary path for output
 
-            # Check if the video is generated before writing it to file
-            print(f"Writing video to {output_path}...")
+            debug_log.append(f"Writing video to {output_path}...")
 
             # Write video to file
             video.write_videofile(output_path, codec='libx264', audio_codec='aac')
 
             # Check if the video file exists
             if os.path.exists(output_path):
-                print(f"Video generated successfully at {output_path}")
-                return output_path  # Return the file path for Gradio output
+                debug_log.append(f"Video generated successfully at {output_path}")
+                return output_path, "\n".join(debug_log)  # Return the video path and debug log
             else:
-                print(f"Error: Video not generated at {output_path}")
-                return "Error: Video not generated."
+                debug_log.append(f"Error: Video not generated at {output_path}")
+                return "Error: Video not generated.", "\n".join(debug_log)
 
     except Exception as e:
-        print(f"Error during video generation: {str(e)}")
-        return f"Error generating video: {str(e)}"
+        debug_log.append(f"Error during video generation: {str(e)}")
+        return f"Error generating video: {str(e)}", "\n".join(debug_log)
 
 # Gradio interface setup
 def gradio_interface():
@@ -110,10 +110,11 @@ def gradio_interface():
                 generate_button = gr.Button("Generate Video")  # Button to generate video
                 
             output_video = gr.Video(label="Generated Video")  # Video output display
+            output_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)
+        generate_button.click(fn=process_and_generate_video, inputs=[mp3_input, image_input], outputs=[output_video, output_logs])
 
     demo.launch()  # Launch the Gradio interface
 
 # Run the interface
-gradio_interface()
+gradio_interface()