Update app.py

app.py

@@ -4,14 +4,11 @@ import os
 import requests
 
 def process_video(audio_file, video_file):
-    # Unpack the audio and video file tuples
+    # Unpack the audio and video file paths
     audio_path = audio_file[1] if isinstance(audio_file, tuple) else audio_file
-    video_path = video_file[1] if isinstance(video_file, tuple) else video_file
+    video_path = video_file if isinstance(video_file, str) else video_file.name
     out_path = "output_video.mp4"
     
-    # No need to save files as Gradio now handles this automatically
-    # and provides the path directly
-    
     # Define command flags
     sample_mode = "cross"  # or "reconstruction"
     generate_from_filelist = 0
@@ -41,8 +38,7 @@ def process_video(audio_file, video_file):
     except subprocess.CalledProcessError as e:
         return f"Error processing video: {e}"
     finally:
-        # Clean up the files after processing
-        # Don't remove input files as they're managed by Gradio
+        # Clean up output file if it exists
         if os.path.exists(out_path):
             os.remove(out_path)
 
@@ -52,18 +48,30 @@ with gr.Blocks() as iface:
     gr.Markdown("Upload an audio file and a video file to process the video based on the audio input.")
     
     with gr.Row():
-        audio_input = gr.Audio(label="Input Audio", type="filepath")
-        video_input = gr.Video(label="Input Video", type="filepath")
+        audio_input = gr.Audio(label="Input Audio", source="upload")
+        video_input = gr.Video(label="Input Video", source="upload")
     
     with gr.Row():
         process_button = gr.Button("Process Video")
-        
+        status_msg = gr.Textbox(label="Status", interactive=False)
+    
     video_output = gr.Video(label="Processed Video")
     
+    def process_with_status(audio, video):
+        try:
+            status_msg.update(value="Processing... Please wait.")
+            result = process_video(audio, video)
+            status_msg.update(value="Done!")
+            return [result, "Processing completed successfully!"]
+        except Exception as e:
+            error_msg = f"Error during processing: {str(e)}"
+            status_msg.update(value=error_msg)
+            return [None, error_msg]
+    
     process_button.click(
-        fn=process_video,
+        fn=process_with_status,
         inputs=[audio_input, video_input],
-        outputs=video_output
+        outputs=[video_output, status_msg]
     )
 
 # Launch the interface