bhacquin commited on
Commit
937f233
·
verified ·
1 Parent(s): 2940ad4

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +14 -13
demo.py CHANGED
@@ -134,13 +134,13 @@ def load_initial():
134
  new_experiment = generate_new_experiment()
135
  return [
136
  new_experiment,
137
- gr.Video.update(value=new_experiment["video_left"]),
138
- gr.Video.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.Video.update(value=""), gr.Video.update(value=""), "❌ Failed to load initial videos ❌"]
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.Video.update(value=new_experiment["video_left"]),
183
- gr.Video.update(value=new_experiment["video_right"]),
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.Video.update(value=experiment["video_left"]),
194
- gr.Video.update(value=experiment["video_right"]),
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.Video.update(), gr.Video.update(), ""]
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.Video.update(value=new_experiment["video_left"]),
234
- gr.Video.update(value=new_experiment["video_right"]),
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.Video.update(value=experiment["video_left"]),
245
- gr.Video.update(value=experiment["video_right"]),
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
- demo.launch()
 
 
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)