fix download

app.py

@@ -112,10 +112,14 @@ def _validate_upload_file(path: Path, allowed_suffixes: set[str], max_mb: int) -
 
 def parse_files(reference_files: list[str] | None):
     if not reference_files:
-        return "Upload at least one reference file.", None, None
+        return "Upload at least one reference file.", None, gr.update(value=None, visible=False, interactive=False)
 
     if len(reference_files) > MAX_UPLOAD_FILES:
-        return f"Too many files. Maximum allowed is {MAX_UPLOAD_FILES}.", None, None
+        return (
+            f"Too many files. Maximum allowed is {MAX_UPLOAD_FILES}.",
+            None,
+            gr.update(value=None, visible=False, interactive=False),
+        )
 
     input_paths = [Path(p) for p in reference_files]
     for p in input_paths:
@@ -125,17 +129,21 @@ def parse_files(reference_files: list[str] | None):
             max_mb=MAX_UPLOAD_FILE_MB,
         )
         if validation_error:
-            return validation_error, None, None
+            return validation_error, None, gr.update(value=None, visible=False, interactive=False)
 
     run_dir = _new_run_dir()
     parsed_output = run_dir / f"parsed_{_timestamp_slug()}.xlsx"
 
     df = reference_parser.parse_references(input_paths=input_paths, output_path=parsed_output)
     if df.empty:
-        return "No records found in the uploaded files.", None, None
+        return "No records found in the uploaded files.", None, gr.update(value=None, visible=False, interactive=False)
 
     msg = f"Parsed {len(df)} deduplicated records."
-    return msg, str(parsed_output), str(parsed_output)
+    return (
+        msg,
+        str(parsed_output),
+        gr.update(value=str(parsed_output), visible=True, interactive=True),
+    )
 
 
 def screen_excel(
@@ -151,14 +159,23 @@ def screen_excel(
 ):
     missing = _missing_secrets()
     if missing:
-        return "Missing required Space secrets: " + ", ".join(missing), None
+        yield "Missing required Space secrets: " + ", ".join(missing), gr.update(
+            value=None, visible=False, interactive=False
+        )
+        return
 
     if not parsed_excel_path:
-        return "Run parser first to generate a parsed Excel file.", None
+        yield "Run parser first to generate a parsed Excel file.", gr.update(
+            value=None, visible=False, interactive=False
+        )
+        return
 
     parsed_path = Path(parsed_excel_path)
     if not parsed_path.exists():
-        return "Parsed Excel file was not found. Please run parser again.", None
+        yield "Parsed Excel file was not found. Please run parser again.", gr.update(
+            value=None, visible=False, interactive=False
+        )
+        return
 
     screened_output = parsed_path.parent / f"screened_{_timestamp_slug()}.xlsx"
     criteria_path = None
@@ -171,7 +188,8 @@ def screen_excel(
             max_mb=MAX_CRITERIA_FILE_MB,
         )
         if validation_error:
-            return validation_error, None
+            yield validation_error, gr.update(value=None, visible=False, interactive=False)
+            return
         criteria_path = candidate
     else:
         inclusion = _parse_criteria_lines(criteria_inclusion_text)
@@ -179,11 +197,20 @@ def screen_excel(
         topic = (criteria_topic or "").strip()
 
         if not topic:
-            return "Provide a criteria topic (or upload criteria file).", None
+            yield "Provide a criteria topic (or upload criteria file).", gr.update(
+                value=None, visible=False, interactive=False
+            )
+            return
         if not inclusion:
-            return "Provide at least one inclusion criterion (or upload criteria file).", None
+            yield "Provide at least one inclusion criterion (or upload criteria file).", gr.update(
+                value=None, visible=False, interactive=False
+            )
+            return
         if not exclusion:
-            return "Provide at least one exclusion criterion (or upload criteria file).", None
+            yield "Provide at least one exclusion criterion (or upload criteria file).", gr.update(
+                value=None, visible=False, interactive=False
+            )
+            return
 
         criteria_obj = {
             "topic": topic,
@@ -234,6 +261,8 @@ def screen_excel(
     worker = threading.Thread(target=_run_screening, daemon=True)
     worker.start()
 
+    yield "Preparing screening...", gr.update(value=None, visible=False, interactive=False)
+
     while worker.is_alive():
         elapsed_s = int(time.perf_counter() - started)
         eta_s = int(progress_state["eta_seconds"])
@@ -253,15 +282,20 @@ def screen_excel(
         else:
             status = f"Initializing client... Elapsed {elapsed_min:02d}:{elapsed_sec:02d}"
 
-        yield status, None
+        yield status, gr.update(value=None, visible=False, interactive=False)
         time.sleep(1)
 
     if worker_error["exc"] is not None:
-        yield f"Screening failed: {worker_error['exc']}", None
+        yield f"Screening failed: {worker_error['exc']}", gr.update(
+            value=None, visible=False, interactive=False
+        )
         return
 
     progress(1, desc="Screening complete.")
-    yield f"Screening complete: {screened_output.name}", str(screened_output)
+    yield (
+        f"Screening complete: {screened_output.name}",
+        gr.update(value=str(screened_output), visible=True, interactive=True),
+    )
 
 
 def build_app() -> gr.Blocks:
@@ -290,10 +324,20 @@ def build_app() -> gr.Blocks:
                 )
                 parse_btn = gr.Button("1) Run Parser", variant="primary")
                 parse_status = gr.Textbox(label="Parser Status", interactive=False)
-                parsed_excel_download = gr.File(label="Parsed Excel")
+                parsed_excel_download = gr.DownloadButton(
+                    "Download Parsed Excel",
+                    value=None,
+                    visible=False,
+                    interactive=False,
+                )
                 screen_btn = gr.Button("2) Screen Excel", variant="primary")
                 screen_status = gr.Textbox(label="Screening Status", interactive=False)
-                screened_excel_download = gr.File(label="Screened Excel")
+                screened_excel_download = gr.DownloadButton(
+                    "Download Screened Excel",
+                    value=None,
+                    visible=False,
+                    interactive=False,
+                )
 
             with gr.Column():
                 criteria_file = gr.File(