Update app.py

app.py

@@ -4,60 +4,54 @@ import ffmpeg
 import gradio as gr
 import traceback
 
-# Global variable to store logs
+# Debug log list
 debug_logs = []
 
-# Function to log messages
 def log_message(message):
-    global debug_logs
+    """Log messages for debugging."""
     debug_logs.append(message)
-    print(message)  # Print to console as well for backend debugging
+    print(message)
 
 # Function to add music to video
 def add_music_to_video(video_path, chunks_folder="chunks", output_path="output_with_music.mp4"):
     try:
-        log_message("Starting the process to add music to video...")
+        log_message("Starting add_music_to_video process...")
         
-        # Check if chunks folder exists
-        if not os.path.exists(chunks_folder):
-            error_message = "Error: Chunks folder does not exist."
-            log_message(error_message)
-            return error_message
-
         # List all MP3 files in the chunks folder
         music_files = [os.path.join(chunks_folder, f) for f in os.listdir(chunks_folder) if f.endswith('.mp3')]
+        
         if not music_files:
             error_message = "Error: No audio files found in the chunks folder."
             log_message(error_message)
             return error_message
-
+        
         # Randomly select one MP3 file
         music_file_path = random.choice(music_files)
         log_message(f"Selected music file: {music_file_path}")
-
+        
         # Get the duration of the video
         video_info = ffmpeg.probe(video_path)
         video_duration = float(video_info['streams'][0]['duration'])
         log_message(f"Video duration: {video_duration} seconds")
-
+        
         # Trim the music file to match the video duration
         trimmed_music_path = "trimmed_music.mp3"
         ffmpeg.input(music_file_path).output(
             trimmed_music_path, ss=0, t=video_duration
         ).run(overwrite_output=True)
-        log_message(f"Trimmed music saved at: {trimmed_music_path}")
-
+        log_message("Music trimmed successfully.")
+        
         # Combine video and audio
         video_input = ffmpeg.input(video_path)
         audio_input = ffmpeg.input(trimmed_music_path)
         ffmpeg.concat(video_input, audio_input, v=1, a=1).output(
             output_path, vcodec="libx264", acodec="aac", strict="experimental"
         ).run(overwrite_output=True)
-        log_message(f"Output video saved at: {output_path}")
-
+        log_message(f"Video with music saved to: {output_path}")
+        
         return output_path
     except Exception as e:
-        error_message = f"Error during processing: {str(e)}"
+        error_message = f"Error in add_music_to_video: {str(e)}"
         log_message(error_message)
         traceback.print_exc()
         return error_message
@@ -66,7 +60,13 @@ def add_music_to_video(video_path, chunks_folder="chunks", output_path="output_w
 def process_video(uploaded_video):
     try:
         log_message("Video received for processing.")
-        video_path = uploaded_video.name
+
+        # Ensure the correct handling of uploaded_video as a string or file object
+        if isinstance(uploaded_video, str):
+            video_path = uploaded_video  # If it's already a string path
+        else:
+            video_path = uploaded_video.name  # File-like object with a .name attribute
+
         log_message(f"Uploaded video path: {video_path}")
 
         output_video = add_music_to_video(video_path, chunks_folder="/content/chunks")
@@ -81,23 +81,16 @@ def process_video(uploaded_video):
         return None, "\n".join(debug_logs)
 
 # Gradio Interface
-with gr.Blocks() as interface:
-    gr.Markdown("# Add Background Music to Video")
-    gr.Markdown("Upload a video, and this app will add a background music track that matches the video duration.")
-    
-    with gr.Row():
-        with gr.Column():
-            video_input = gr.Video(label="Upload Video")
-            submit_button = gr.Button("Generate Video")
-        with gr.Column():
-            video_output = gr.Video(label="Video with Background Music")
-            debug_output = gr.Textbox(label="Debug Logs", lines=10, interactive=False)
-    
-    submit_button.click(
-        fn=process_video,
-        inputs=[video_input],
-        outputs=[video_output, debug_output]
-    )
+interface = gr.Interface(
+    fn=process_video,
+    inputs=gr.Video(label="Upload Video"),
+    outputs=[
+        gr.Video(label="Video with Background Music"),
+        gr.Textbox(label="Debug Logs")
+    ],
+    title="Add Background Music to Video",
+    description="Upload a video, and this app will add a 10-second background music track that matches the video duration."
+)
 
 # Launch Gradio app
-interface.launch(share=True, debug=True)
+interface.launch(share=True)