Update app.py

app.py

@@ -6,18 +6,22 @@ def compare_videos_interface(video1, video2):
     if video1 is None or video2 is None:
         return "Please upload both videos.", None, None
     
+    # Validate file extensions
+    valid_extensions = [".mp4", ".avi", ".mov"]
+    if not any(video1.name.lower().endswith(ext) for ext in valid_extensions) or \
+       not any(video2.name.lower().endswith(ext) for ext in valid_extensions):
+        return "Invalid video format. Please upload MP4, AVI, or MOV files.", None, None
+    
     # Save uploaded videos temporarily
     os.makedirs("temp", exist_ok=True)
-    video1_path = os.path.join("temp", os.path.basename(video1))
-    video2_path = os.path.join("temp", os.path.basename(video2))
+    video1_path = os.path.join("temp", os.path.basename(video1.name))
+    video2_path = os.path.join("temp", os.path.basename(video2.name))
     
     # Copy uploaded files to temp directory
     with open(video1_path, "wb") as f:
-        with open(video1, "rb") as v1:
-            f.write(v1.read())
+        f.write(video1.read())
     with open(video2_path, "wb") as f:
-        with open(video2, "rb") as v2:
-            f.write(v2.read())
+        f.write(video2.read())
     
     try:
         # Compare videos
@@ -43,11 +47,11 @@ def compare_videos_interface(video1, video2):
 # Define Gradio interface
 with gr.Blocks(title="Video Comparison App") as demo:
     gr.Markdown("# Video Comparison App")
-    gr.Markdown("Upload two videos of the same duration to compare frame-by-frame differences.")
+    gr.Markdown("Upload two videos of the same duration (MP4, AVI, MOV) to compare frame-by-frame differences.")
     
     with gr.Row():
-        video1_input = gr.Video(label="Upload First Video", format=["mp4", "avi", "mov"])
-        video2_input = gr.Video(label="Upload Second Video", format=["mp4", "avi", "mov"])
+        video1_input = gr.File(label="Upload First Video", file_types=[".mp4", ".avi", ".mov"])
+        video2_input = gr.File(label="Upload Second Video", file_types=[".mp4", ".avi", ".mov"])
     
     submit_button = gr.Button("Compare Videos")