Update app.py

app.py

@@ -1,62 +1,67 @@
-import streamlit as st
-from utils.video_processor import compare_videos
+import gradio as gr
 import os
+from utils.video_processor import compare_videos
 
-# Set page configuration
-st.set_page_config(page_title="Video Comparison App", layout="wide")
-
-# Title and description
-st.title("Video Comparison App")
-st.write("Upload two videos of the same duration to compare frame-by-frame differences.")
-
-# File uploaders
-col1, col2 = st.columns(2)
-with col1:
-    video1_file = st.file_uploader("Upload First Video", type=["mp4", "avi", "mov"])
-with col2:
-    video2_file = st.file_uploader("Upload Second Video", type=["mp4", "avi", "mov"])
-
-# Submit button
-if st.button("Compare Videos"):
-    if video1_file is None or video2_file is None:
-        st.error("Please upload both videos.")
-    else:
-        # Save uploaded videos temporarily
-        os.makedirs("temp", exist_ok=True)
-        video1_path = os.path.join("temp", video1_file.name)
-        video2_path = os.path.join("temp", video2_file.name)
-        
-        with open(video1_path, "wb") as f:
-            f.write(video1_file.getbuffer())
-        with open(video2_path, "wb") as f:
-            f.write(video2_file.getbuffer())
+def compare_videos_interface(video1, video2):
+    if video1 is None or video2 is None:
+        return "Please upload both videos.", 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))
+    
+    # Copy uploaded files to temp directory
+    with open(video1_path, "wb") as f:
+        with open(video1, "rb") as v1:
+            f.write(v1.read())
+    with open(video2_path, "wb") as f:
+        with open(video2, "rb") as v2:
+            f.write(v2.read())
+    
+    try:
+        # Compare videos
+        plot_path, diff_video_path = compare_videos(video1_path, video2_path)
         
-        try:
-            # Compare videos
-            plot_path, diff_video_path = compare_videos(video1_path, video2_path)
-            
-            # Display results
-            st.success("Comparison complete!")
-            st.subheader("Difference Plot")
-            st.image(plot_path, caption="Frame-by-frame SSIM Difference (%)")
-            
-            st.subheader("Difference Video")
-            with open(diff_video_path, "rb") as f:
-                st.video(f.read())
-            
-            # Clean up temporary files
+        # Return results
+        return (
+            "Comparison complete!",
+            plot_path,
+            diff_video_path
+        )
+    
+    except Exception as e:
+        return f"Error during comparison: {str(e)}", None, None
+    
+    finally:
+        # Clean up temporary files
+        if os.path.exists(video1_path):
             os.remove(video1_path)
+        if os.path.exists(video2_path):
             os.remove(video2_path)
-            os.remove(plot_path)
-            os.remove(diff_video_path)
-            
-        except Exception as e:
-            st.error(f"Error during comparison: {str(e)}")
-            # Clean up temporary files if they exist
-            if os.path.exists(video1_path):
-                os.remove(video1_path)
-            if os.path.exists(video2_path):
-                os.remove(video2_path)
 
+# 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.")
+    
+    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"])
+    
+    submit_button = gr.Button("Compare Videos")
+    
+    output_text = gr.Textbox(label="Status")
+    output_plot = gr.Image(label="Difference Plot")
+    output_video = gr.Video(label="Difference Video")
+    
+    submit_button.click(
+        fn=compare_videos_interface,
+        inputs=[video1_input, video2_input],
+        outputs=[output_text, output_plot, output_video]
+    )
+
+# Launch the app
 if __name__ == "__main__":
-    os.makedirs("utils/output", exist_ok=True)
+    os.makedirs("utils/output", exist_ok=True)
+    demo.launch()