chore: handle empty video

app.py

@@ -22,11 +22,11 @@ def trim_video(video_path, start, end, output_file='result.mp4'):
     return output_file
 
 def display_prediction(result):
-    return f'Moment({result[0]} ~ {result[1]}), Score: {result[2]}'
+    return f'### Moment  Start time:  {result[0]},   End time:  {result[1]},  Score:  {result[2]}'
 
 with gr.Blocks(theme=gr.themes.Default()) as demo:
-    output_videos = gr.State([])
-    moment_prediction = gr.State([])
+    output_videos = gr.State(None)
+    moment_prediction = gr.State(None)
     gr.HTML("""<h2 align="center"> 🎞️  Highlight Detection with MomentDETR </h2>""")
     gr.Markdown(DESCRIPTION)
     with gr.Column():
@@ -46,16 +46,18 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
                     placeholder="What should be highlighted?",
                     visible=True
                 )
-                submit =gr.Button("Submit")
+                submit = gr.Button("Submit")
             with gr.Column():
-                display_score = gr.Markdown("### Moment Score: ")
                 radio_button = gr.Radio(
                     choices=[i+1 for i in range(10)], 
                     label="Moments", 
                     value=1
                 )
+                display_score = gr.Markdown("### Moment Score: ")
 
         def update_video_player(radio_value, output_videos, moment_prediction):
+            if output_videos is None or moment_prediction is None:
+                return [None, None]
             return {
                 playable_video: output_videos[radio_value-1], 
                 display_score: display_prediction(moment_prediction[radio_value-1])
@@ -64,6 +66,8 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
     def submit_video(input_video, retrieval_text):
         print(f'== video path: {input_video}')
         print(f'== retrieval_text: {retrieval_text}')
+        if input_video is None:
+            return [None, None, None, None, 1]
         if retrieval_text is None:
             retrieval_text = ''
         predictions, video_frames = moment_detr_predictor.localize_moment(
@@ -83,7 +87,8 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
             output_videos: output_files, 
             moment_prediction: predictions,
             playable_video:  output_files[0],
-            display_score: display_prediction(predictions[0])
+            display_score: display_prediction(predictions[0]),
+            radio_button: 1
         }
 
     radio_button.change(
@@ -95,7 +100,7 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
     submit.click(
         fn=submit_video, 
         inputs=[input_video, retrieval_text], 
-        outputs=[output_videos, moment_prediction, playable_video, display_score]
+        outputs=[output_videos, moment_prediction, playable_video, display_score, radio_button]
     )
 
 demo.launch()