Update result_ui.py

gradio_scripts/result_ui.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import numpy as np
 from gradio_scripts.pdf_handler import make_pdf
+import os
 
 js_update_tab_labels = """
     async () => {
@@ -42,12 +43,17 @@ def update_result(i, state, result, inference_handler):
 
     make_pdf(state['index']-1, state, result, table_headers)
 
+    # Check if files exist
+    video_path = result["path_video"][i]
+    if not os.path.exists(video_path): video_path = ""
+
     # Send update to UI, and to inference_handler to start next file inference
+
     print("loaded result")
     return {
         zip_out: gr.update(value=result["path_zip"]),
         tabs[i]['tab']: gr.update(),
-        tabs[i]['video']: gr.update(value=result["path_video"][i], visible=True),
+        tabs[i]['video']: gr.update(value=video_path, visible=True),
         tabs[i]['metadata']: gr.update(value=result["fish_info"][i], visible=True),
         tabs[i]['table']: gr.update(value=result["fish_table"][i], visible=True),
         tabs[i]['annotation_btn']: gr.update(visible=annotation_avaliable),