.

app.py

@@ -95,91 +95,91 @@ def create_ui(examples_path: str):
         @spaces.GPU
         def on_process(video):
             if not video:
-                yield {
-                    "status": "Please upload a video",  # Changed to string key
-                    "video_description": gr.update(value=""),  # Added gr.update
-                    "highlight_types": gr.update(value=""),  # Added gr.update
-                    "output_video": gr.update(visible=False),
-                    "analysis_accordion": gr.update(visible=False)
-                }
+                yield [
+                    "Please upload a video",  # status
+                    "",  # video_description
+                    "",  # highlight_types
+                    gr.update(visible=False),  # output_video
+                    gr.update(visible=False)  # analysis_accordion
+                ]
                 return
             
             try:
                 duration = get_video_duration_seconds(video)
                 if duration > 1200:  # 20 minutes
-                    yield {
-                        "status": "Video must be shorter than 20 minutes",
-                        "video_description": gr.update(value=""),
-                        "highlight_types": gr.update(value=""),
-                        "output_video": gr.update(visible=False),
-                        "analysis_accordion": gr.update(visible=False)
-                    }
+                    yield [
+                        "Video must be shorter than 20 minutes",
+                        "",
+                        "",
+                        gr.update(visible=False),
+                        gr.update(visible=False)
+                    ]
                     return
 
                 # Make accordion visible as soon as processing starts
-                yield {
-                    "status": "Loading model...",
-                    "video_description": gr.update(value=""),
-                    "highlight_types": gr.update(value=""),
-                    "output_video": gr.update(visible=False),
-                    "analysis_accordion": gr.update(visible=True)
-                }
+                yield [
+                    "Loading model...",
+                    "",
+                    "",
+                    gr.update(visible=False),
+                    gr.update(visible=True)
+                ]
 
                 model, processor = load_model()
                 detector = BatchedVideoHighlightDetector(model, processor, batch_size=8)
 
-                yield {
-                    "status": "Analyzing video content...",
-                    "video_description": gr.update(value=""),
-                    "highlight_types": gr.update(value=""),
-                    "output_video": gr.update(visible=False),
-                    "analysis_accordion": gr.update(visible=True)
-                }
+                yield [
+                    "Analyzing video content...",
+                    "",
+                    "",
+                    gr.update(visible=False),
+                    gr.update(visible=True)
+                ]
                 
                 video_desc = detector.analyze_video_content(video)
                 formatted_desc = f"#Summary: {video_desc[:500] + '...' if len(video_desc) > 500 else video_desc}"
                 
                 # Update description as soon as it's available
-                yield {
-                    "status": "Determining highlight types...",
-                    "video_description": gr.update(value=formatted_desc),
-                    "highlight_types": gr.update(value=""),
-                    "output_video": gr.update(visible=False),
-                    "analysis_accordion": gr.update(visible=True)
-                }
+                yield [
+                    "Determining highlight types...",
+                    formatted_desc,
+                    "",
+                    gr.update(visible=False),
+                    gr.update(visible=True)
+                ]
                 
                 highlights = detector.determine_highlights(video_desc)
                 formatted_highlights = f"#Highlights to search for: {highlights[:500] + '...' if len(highlights) > 500 else highlights}"
                 
                 # Update highlights as soon as they're available
-                yield {
-                    "status": "Detecting and extracting highlights...",
-                    "video_description": gr.update(value=formatted_desc),
-                    "highlight_types": gr.update(value=formatted_highlights),
-                    "output_video": gr.update(visible=False),
-                    "analysis_accordion": gr.update(visible=True)
-                }
+                yield [
+                    "Detecting and extracting highlights...",
+                    formatted_desc,
+                    formatted_highlights,
+                    gr.update(visible=False),
+                    gr.update(visible=True)
+                ]
 
                 with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as tmp_file:
                     temp_output = tmp_file.name
                 detector.create_highlight_video(video, temp_output)
                 
-                yield {
-                    "status": "Processing complete!",
-                    "video_description": gr.update(value=formatted_desc),
-                    "highlight_types": gr.update(value=formatted_highlights),
-                    "output_video": gr.update(value=temp_output, visible=True),
-                    "analysis_accordion": gr.update(visible=True)
-                }
+                yield [
+                    "Processing complete!",
+                    formatted_desc,
+                    formatted_highlights,
+                    gr.update(value=temp_output, visible=True),
+                    gr.update(visible=True)
+                ]
 
             except Exception as e:
-                yield {
-                    "status": f"Error processing video: {str(e)}",
-                    "video_description": gr.update(value=""),
-                    "highlight_types": gr.update(value=""),
-                    "output_video": gr.update(visible=False),
-                    "analysis_accordion": gr.update(visible=False)
-                }
+                yield [
+                    f"Error processing video: {str(e)}",
+                    "",
+                    "",
+                    gr.update(visible=False),
+                    gr.update(visible=False)
+                ]
 
         process_btn.click(
             on_process,
@@ -191,7 +191,7 @@ def create_ui(examples_path: str):
                 output_video,
                 analysis_accordion
             ],
-            queue=True,  # Added queue=True
+            queue=True,
         )
 
     return app