Update demo.py

demo.py

@@ -1,6 +1,4 @@
 import datetime
-from functools import partial
-import json
 from pathlib import Path
 import random
 import gradio as gr
@@ -42,7 +40,6 @@ except Exception as e:
 # Data Layer
 ##################################################################
 
-
 class Experiment(dict):
     """
     Represents an experiment consisting of two videos and the user's selection.
@@ -55,7 +52,6 @@ class Experiment(dict):
             selected_video=selected_video,
         )
 
-
 def experiment_to_dict(experiment, skip=False):
     """
     Converts the Experiment object to a dictionary for Firebase storage.
@@ -96,7 +92,6 @@ def experiment_to_dict(experiment, skip=False):
 
     return info
 
-
 def generate_new_experiment() -> Experiment:
     """
     Generates a new experiment by randomly selecting a sample number and fetching corresponding videos from both folders.
@@ -121,14 +116,12 @@ def generate_new_experiment() -> Experiment:
         video_left, video_right = video_omni_path, video_ours_path
 
     print(f"Generated new experiment with left video: {video_left}, right video: {video_right}")
-
     return Experiment(
         DATASET,
         str(video_left),
         str(video_right),
     )
 
-
 def load_initial():
     """
     Initializes the first experiment on app load.
@@ -146,14 +139,11 @@ def load_initial():
         print(f"Error loading initial experiment: {e}")
         return [None, gr.update(value=""), gr.update(value=""), "❌ Failed to load initial videos ❌"]
 
-
 def select_and_load(selected_label, experiment, message_component):
     """
     Handles the selection of a video by the user.
     Saves the selection and loads the next experiment.
     """
-    print(f"User selected: {selected_label}")
-
     # Update the experiment's selected video
     if selected_label.lower() == "left":
         experiment["selected_video"] = "left"
@@ -166,7 +156,10 @@ def select_and_load(selected_label, experiment, message_component):
     try:
         dict_to_save = experiment_to_dict(experiment, skip=False)
         firebase_data_ref.push(dict_to_save)
-        print("Saved selection to Firebase:", dict_to_save)
+
+        print("=====================")
+        print(dict_to_save)
+        print("=====================")
 
         # Update the message component with a success message
         message_component.value = "✅ Your choice has been saved to Firebase ✅"
@@ -198,18 +191,11 @@ def select_and_load(selected_label, experiment, message_component):
             message_component.value
         ]
 
-
-def skip_experiment(mode, experiment, message_component):
+def skip_experiment(experiment, message_component):
     """
     Handles the skipping of an experiment.
     Saves the skip and loads the next experiment.
     """
-    if mode != "skip":
-        print("Skip mode not activated.")
-        return [experiment, gr.update(), gr.update(), ""]
-
-    print("User chose to skip the current experiment.")
-
     # Set selected_video to "None" to indicate skip
     experiment["selected_video"] = "None"
 
@@ -217,7 +203,10 @@ def skip_experiment(mode, experiment, message_component):
     try:
         dict_to_save = experiment_to_dict(experiment, skip=True)
         firebase_data_ref.push(dict_to_save)
-        print("Saved skip to Firebase:", dict_to_save)
+
+        print("=====================")
+        print(dict_to_save)
+        print("=====================")
 
         # Update the message component with a success message
         message_component.value = "✅ Your skip has been recorded ✅"
@@ -253,18 +242,7 @@ def skip_experiment(mode, experiment, message_component):
 # UI Layer
 ##################################################################
 
-
 css = """
-#unsel {
-    border: solid 5px transparent !important;
-    border-radius: 15px !important;
-    position: relative;
-}
-#sel {
-    border: solid 5px #00c0ff !important;
-    border-radius: 15px !important;
-    position: relative;
-}
 #padded {
     padding-left: 2%;
     padding-right: 2%;
@@ -296,7 +274,6 @@ with gr.Blocks(title="Unsupervised Video Editing", css=css) as demo:
         with gr.Column(scale=3, elem_id="padded"):
             gr.Markdown("<div style='width: 100%'><h1 style='text-align: center;'>Choose Your Preferred Video</h1></div>")
             gr.Markdown("<div style='width: 100%'><h3 style='text-align: center;'>Select the video you prefer.<br/>⚠️Consider fidelity and quality⚠️</h3></div>")
-            # Removed the submit button as per requirement
             btn_skip = gr.Button("I have no preference")
             message_component = gr.Markdown("")  # For displaying messages
 
@@ -331,27 +308,27 @@ with gr.Blocks(title="Unsupervised Video Editing", css=css) as demo:
     with gr.Row():
         # Configure the skip button
         btn_skip.click(
-            fn=partial(skip_experiment, mode="skip"),
+            fn=skip_experiment,
             inputs=[experiment, message_component],
             outputs=[experiment, video_left_component, video_right_component, message_component],
-            queue=False,
-            show_progress=False
+            queue=True,
+            show_progress=True
         )
 
     # Configure the select buttons to handle selection and load next experiment
     select_left.click(
-        fn=partial(select_and_load, selected_label="left"),
+        fn=lambda exp, msg: select_and_load("left", exp, msg),
         inputs=[experiment, message_component],
         outputs=[experiment, video_left_component, video_right_component, message_component],
-        queue=False,
-        show_progress=False
+        queue=True,
+        show_progress=True
     )
     select_right.click(
-        fn=partial(select_and_load, selected_label="right"),
+        fn=lambda exp, msg: select_and_load("right", exp, msg),
         inputs=[experiment, message_component],
         outputs=[experiment, video_left_component, video_right_component, message_component],
-        queue=False,
-        show_progress=False
+        queue=True,
+        show_progress=True
     )
 
     # Load the first experiment on app startup