alexnasa commited on
Commit
cfc9d46
·
verified ·
1 Parent(s): 5a24be4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -5
app.py CHANGED
@@ -223,7 +223,7 @@ def update_gallery_on_video_upload(input_video, session_id):
223
  return None, None, None
224
 
225
  target_dir, image_paths = extract_frames(input_video, session_id)
226
- return None, target_dir, image_paths, image_paths
227
 
228
  def update_gallery_on_images_upload(input_images, session_id):
229
 
@@ -231,7 +231,7 @@ def update_gallery_on_images_upload(input_images, session_id):
231
  return None, None, None
232
 
233
  target_dir, image_paths = extract_images(input_images, session_id)
234
- return None, target_dir, image_paths, image_paths
235
 
236
  @spaces.GPU()
237
  def generate_splats_from_video(video_path, session_id=None):
@@ -247,6 +247,14 @@ def generate_splats_from_video(video_path, session_id=None):
247
  @spaces.GPU()
248
  def generate_splats_from_images(image_paths, session_id=None):
249
 
 
 
 
 
 
 
 
 
250
  print(image_paths)
251
 
252
  if session_id is None:
@@ -408,19 +416,19 @@ if __name__ == "__main__":
408
 
409
  submit_btn.click(
410
  fn=generate_splats_from_images,
411
- inputs=[input_images, session_state],
412
  outputs=[reconstruction_output, rgb_video, depth_video])
413
 
414
  input_video.upload(
415
  fn=update_gallery_on_video_upload,
416
  inputs=[input_video, session_state],
417
- outputs=[reconstruction_output, target_dir_output, image_gallery, input_images],
418
  )
419
 
420
  input_images.upload(
421
  fn=update_gallery_on_images_upload,
422
  inputs=[input_images, session_state],
423
- outputs=[reconstruction_output, target_dir_output, image_gallery, input_images],
424
  )
425
 
426
  demo.unload(cleanup)
 
223
  return None, None, None
224
 
225
  target_dir, image_paths = extract_frames(input_video, session_id)
226
+ return None, target_dir, image_paths
227
 
228
  def update_gallery_on_images_upload(input_images, session_id):
229
 
 
231
  return None, None, None
232
 
233
  target_dir, image_paths = extract_images(input_images, session_id)
234
+ return None, target_dir, image_paths
235
 
236
  @spaces.GPU()
237
  def generate_splats_from_video(video_path, session_id=None):
 
247
  @spaces.GPU()
248
  def generate_splats_from_images(image_paths, session_id=None):
249
 
250
+ processed_image_paths = []
251
+
252
+ for file_data in image_paths:
253
+ if isinstance(file_data, tuple):
254
+ file_path, _ = file_data
255
+ processed_image_paths.append(file_path)
256
+
257
+ image_paths = processed_image_paths
258
  print(image_paths)
259
 
260
  if session_id is None:
 
416
 
417
  submit_btn.click(
418
  fn=generate_splats_from_images,
419
+ inputs=[image_gallery, session_state],
420
  outputs=[reconstruction_output, rgb_video, depth_video])
421
 
422
  input_video.upload(
423
  fn=update_gallery_on_video_upload,
424
  inputs=[input_video, session_state],
425
+ outputs=[reconstruction_output, target_dir_output, image_gallery],
426
  )
427
 
428
  input_images.upload(
429
  fn=update_gallery_on_images_upload,
430
  inputs=[input_images, session_state],
431
+ outputs=[reconstruction_output, target_dir_output, image_gallery],
432
  )
433
 
434
  demo.unload(cleanup)