Fix state initialization for catalog

frame_extraction/src/frame_extraction/app.py

@@ -28,12 +28,11 @@ def ensure_output_dirs() -> None:
     (OUTPUT_DIR / "frames").mkdir(parents=True, exist_ok=True)
 
 
-def build_catalog_from_video(files: list[gr.FileData]) -> tuple[str | None, str]:
-    if not files:
+def build_catalog_from_video(file: gr.FileData) -> tuple[str | None, str]:
+    if file is None:
         raise gr.Error("Please upload a source video first.")
 
     ensure_output_dirs()
-    file = files[0]
     run_id = uuid.uuid4().hex[:8]
     video_dir = OUTPUT_DIR / "videos"
     video_path = video_dir / f"{run_id}_{Path(file.name).name}"
@@ -83,15 +82,10 @@ def predict_from_arrays(arrays: list[np.ndarray], catalog_path: str | None) -> t
 def build_interface() -> gr.Blocks:
     with gr.Blocks() as demo:
         gr.Markdown("# Character Reference Matcher")
-        catalog_state = gr.State[str | None](str(CATALOG_PATH) if CATALOG_PATH.exists() else None)
+        catalog_state = gr.State(str(CATALOG_PATH) if CATALOG_PATH.exists() else None)
         status_box = gr.Textbox(label="Status", value="Upload a video to generate a catalog.", interactive=False)
 
-        video_upload = gr.UploadButton(
-            label="Upload source video",
-            file_types=["video"],
-            file_count="single",
-        )
-
+        video_upload = gr.File(label="Source video", file_types=["video"], height="auto")
         frame_upload = gr.UploadButton(
             label="Upload frames",
             file_types=["image"],
@@ -101,7 +95,7 @@ def build_interface() -> gr.Blocks:
         matches_json = gr.JSON(label="Matches")
         gallery = gr.Gallery(label="Reference Thumbnails", columns=2)
 
-        video_upload.upload(build_catalog_from_video, inputs=video_upload, outputs=[catalog_state, status_box])
+        video_upload.change(build_catalog_from_video, inputs=video_upload, outputs=[catalog_state, status_box])
 
         def handle_frames(files: list[gr.FileData], catalog_path: str | None) -> tuple[list[dict[str, Any]], list[list[str]]]:
             arrays = [np.array(Image.open(file.name).convert("RGB")) for file in files]