Update demo.py
Browse files
demo.py
CHANGED
|
@@ -134,13 +134,13 @@ def load_initial():
|
|
| 134 |
new_experiment = generate_new_experiment()
|
| 135 |
return [
|
| 136 |
new_experiment,
|
| 137 |
-
gr.
|
| 138 |
-
gr.
|
| 139 |
"" # Empty message
|
| 140 |
]
|
| 141 |
except Exception as e:
|
| 142 |
print(f"Error loading initial experiment: {e}")
|
| 143 |
-
return [None, gr.
|
| 144 |
|
| 145 |
|
| 146 |
def select_and_load(selected_label, experiment, message_component):
|
|
@@ -179,8 +179,8 @@ def select_and_load(selected_label, experiment, message_component):
|
|
| 179 |
# Update the video components with new video paths
|
| 180 |
return [
|
| 181 |
new_experiment,
|
| 182 |
-
gr.
|
| 183 |
-
gr.
|
| 184 |
message_component.value
|
| 185 |
]
|
| 186 |
except Exception as e:
|
|
@@ -190,8 +190,8 @@ def select_and_load(selected_label, experiment, message_component):
|
|
| 190 |
# Keep the current experiment and videos
|
| 191 |
return [
|
| 192 |
experiment,
|
| 193 |
-
gr.
|
| 194 |
-
gr.
|
| 195 |
message_component.value
|
| 196 |
]
|
| 197 |
|
|
@@ -202,7 +202,7 @@ def skip_experiment(mode, experiment, message_component):
|
|
| 202 |
Saves the skip and loads the next experiment.
|
| 203 |
"""
|
| 204 |
if mode != "skip":
|
| 205 |
-
return [experiment, gr.
|
| 206 |
|
| 207 |
# Set selected_video to "None" to indicate skip
|
| 208 |
experiment["selected_video"] = "None"
|
|
@@ -230,8 +230,8 @@ def skip_experiment(mode, experiment, message_component):
|
|
| 230 |
# Update the video components with new video paths
|
| 231 |
return [
|
| 232 |
new_experiment,
|
| 233 |
-
gr.
|
| 234 |
-
gr.
|
| 235 |
message_component.value
|
| 236 |
]
|
| 237 |
except Exception as e:
|
|
@@ -241,8 +241,8 @@ def skip_experiment(mode, experiment, message_component):
|
|
| 241 |
# Keep the current experiment and videos
|
| 242 |
return [
|
| 243 |
experiment,
|
| 244 |
-
gr.
|
| 245 |
-
gr.
|
| 246 |
message_component.value
|
| 247 |
]
|
| 248 |
|
|
@@ -358,4 +358,5 @@ with gr.Blocks(title="Unsupervised Video Editing", css=css) as demo:
|
|
| 358 |
outputs=[experiment, video_left_component, video_right_component, message_component]
|
| 359 |
)
|
| 360 |
|
| 361 |
-
|
|
|
|
|
|
| 134 |
new_experiment = generate_new_experiment()
|
| 135 |
return [
|
| 136 |
new_experiment,
|
| 137 |
+
gr.update(value=new_experiment["video_left"]),
|
| 138 |
+
gr.update(value=new_experiment["video_right"]),
|
| 139 |
"" # Empty message
|
| 140 |
]
|
| 141 |
except Exception as e:
|
| 142 |
print(f"Error loading initial experiment: {e}")
|
| 143 |
+
return [None, gr.update(value=""), gr.update(value=""), "❌ Failed to load initial videos ❌"]
|
| 144 |
|
| 145 |
|
| 146 |
def select_and_load(selected_label, experiment, message_component):
|
|
|
|
| 179 |
# Update the video components with new video paths
|
| 180 |
return [
|
| 181 |
new_experiment,
|
| 182 |
+
gr.update(value=new_experiment["video_left"]),
|
| 183 |
+
gr.update(value=new_experiment["video_right"]),
|
| 184 |
message_component.value
|
| 185 |
]
|
| 186 |
except Exception as e:
|
|
|
|
| 190 |
# Keep the current experiment and videos
|
| 191 |
return [
|
| 192 |
experiment,
|
| 193 |
+
gr.update(value=experiment["video_left"]),
|
| 194 |
+
gr.update(value=experiment["video_right"]),
|
| 195 |
message_component.value
|
| 196 |
]
|
| 197 |
|
|
|
|
| 202 |
Saves the skip and loads the next experiment.
|
| 203 |
"""
|
| 204 |
if mode != "skip":
|
| 205 |
+
return [experiment, gr.update(), gr.update(), ""]
|
| 206 |
|
| 207 |
# Set selected_video to "None" to indicate skip
|
| 208 |
experiment["selected_video"] = "None"
|
|
|
|
| 230 |
# Update the video components with new video paths
|
| 231 |
return [
|
| 232 |
new_experiment,
|
| 233 |
+
gr.update(value=new_experiment["video_left"]),
|
| 234 |
+
gr.update(value=new_experiment["video_right"]),
|
| 235 |
message_component.value
|
| 236 |
]
|
| 237 |
except Exception as e:
|
|
|
|
| 241 |
# Keep the current experiment and videos
|
| 242 |
return [
|
| 243 |
experiment,
|
| 244 |
+
gr.update(value=experiment["video_left"]),
|
| 245 |
+
gr.update(value=experiment["video_right"]),
|
| 246 |
message_component.value
|
| 247 |
]
|
| 248 |
|
|
|
|
| 358 |
outputs=[experiment, video_left_component, video_right_component, message_component]
|
| 359 |
)
|
| 360 |
|
| 361 |
+
# Launch the app with share=True to create a public link
|
| 362 |
+
demo.launch(share=True)
|