Upload app.py with huggingface_hub

app.py

@@ -97,7 +97,7 @@ def _prune_downloads(max_age_seconds: int = 3600) -> None:
         pass
 
 
-def _download_path(markdown: str | None) -> str | None:
+def _write_download(markdown: str | None) -> str | None:
     if not markdown or not markdown.strip():
         return None
     try:
@@ -177,7 +177,7 @@ def _run_harness(text: str | None, num_candidates: int | None, structured_fallba
 
 
 @spaces.GPU(duration=_estimate_duration)
-def convert(text: str | None = None, num_candidates: int | None = None, structured_fallback: bool | None = None, max_output_tokens: int | None = None) -> tuple[str, str, str, str]:
+def convert(text: str | None = None, num_candidates: int | None = None, structured_fallback: bool | None = None, max_output_tokens: int | None = None) -> tuple[str, str, str, str, str | None]:
     """Convert a plain-text document into faithful GitHub-Flavored Markdown.
 
     PASITA samples the document several times, every candidate is post-processed and
@@ -191,11 +191,12 @@ def convert(text: str | None = None, num_candidates: int | None = None, structur
         structured_fallback: Use the deterministic converter for HTML and delimited data.
         max_output_tokens: Optional cap on generated tokens. When omitted, the length is chosen automatically.
     Returns:
-        A tuple of the rendered Markdown, the raw Markdown, a status line and run details.
+        A tuple of the rendered Markdown, the raw Markdown, a status line, run details
+        and a download path for the Markdown file (or None).
     """
     text = (text or "").strip()
     if not text:
-        return EMPTY_MD, "", _status("READY / PASTE A DOCUMENT TO CONVERT"), DETAILS_EMPTY
+        return EMPTY_MD, "", _status("READY / PASTE A DOCUMENT TO CONVERT"), DETAILS_EMPTY, None
     if len(text) > MAX_INPUT_CHARS:
         message = f"Input too long ({len(text):,} chars). The limit is {MAX_INPUT_CHARS:,} characters."
         return (
@@ -203,6 +204,7 @@ def convert(text: str | None = None, num_candidates: int | None = None, structur
             "",
             _status("ERROR / INPUT TOO LONG", "error"),
             _details_md(None, 0.0, error=message),
+            None,
         )
 
     try:
@@ -219,6 +221,7 @@ def convert(text: str | None = None, num_candidates: int | None = None, structur
             "",
             _status(f"ERROR / {err_type}", "error"),
             _details_md(None, 0.0, error=f"{plain} Technical detail: {err_type}: {exc}"),
+            None,
         )
 
     metrics = result["metrics"]
@@ -234,7 +237,7 @@ def convert(text: str | None = None, num_candidates: int | None = None, structur
     detail = f"{source_label} / {numbers} / {overlap} / {taken:.1f}S"
     if result.get("strategy", {}).get("truncated_input"):
         detail += " / INPUT TRUNCATED TO 2048 TOKENS"
-    return result["markdown"], result["markdown"], _status("DONE", "ready", detail), _details_md(result, taken)
+    return result["markdown"], result["markdown"], _status("DONE", "ready", detail), _details_md(result, taken), _write_download(result["markdown"])
 
 
 @spaces.GPU(duration=_estimate_duration)
@@ -458,8 +461,8 @@ def _extract_user_text(messages: list[ChatMessage]) -> str | None:
     return None
 
 
-def reset() -> tuple[str, str, str, str, str]:
-    return EMPTY_MD, "", _status("READY"), DETAILS_EMPTY, _counter("")
+def reset() -> tuple[str, str, str, str, str | None, str]:
+    return EMPTY_MD, "", _status("READY"), DETAILS_EMPTY, None, _counter("")
 
 
 @server.post("/v1/chat/completions", response_model=None)
@@ -528,12 +531,12 @@ def chat_completions(body: ChatCompletionRequest, request: fastapi.Request) -> d
     }
 
 
-BUSY_JS = """(text, num_candidates, structured_fallback, md, source, status, details) => {
+BUSY_JS = """(text, num_candidates, structured_fallback, md, source, status, details, dl) => {
   const n = Number(num_candidates) || 1;
   const label = n > 1 ? 'PASITA ' + n + ' SAMPLES / RERANK' : 'PASITA SAMPLE / RERANK';
   return ['_Converting…_', '',
     '<div class="p-status busy"><span class="p-dot"></span><span class="p-text">CONVERTING</span><span class="p-detail">' + label + '</span></div>',
-    '_Checking drafts…_'];
+    '_Checking drafts…_', null];
 }"""
 
 COUNTER_JS = """(text) => {
@@ -893,13 +896,13 @@ with gr.Blocks(title="PASITA: plain text to Markdown") as demo:
                     details = gr.Markdown(DETAILS_EMPTY, elem_id="p-details")
             with gr.Row(elem_classes=["p-out-head"]):
                 out_label = gr.HTML('<span class="p-label">MARKDOWN</span>')
-                dl = gr.DownloadButton("DOWNLOAD .MD", value=_download_path, inputs=[source], size="sm", elem_id="p-dl")
+                dl = gr.DownloadButton("DOWNLOAD .MD", size="sm", elem_id="p-dl")
 
     gr.Examples(
         examples=EXAMPLES,
         example_labels=["ES REPORT", "EN MEETING", "HTML", "CSV"],
         inputs=[text_in],
-        outputs=[md_out, source, status, details],
+        outputs=[md_out, source, status, details, dl],
         fn=convert,
         cache_examples=True,
         cache_mode="lazy",
@@ -912,7 +915,7 @@ with gr.Blocks(title="PASITA: plain text to Markdown") as demo:
     run.click(
         convert,
         inputs=[text_in, candidates, structured],
-        outputs=[md_out, source, status, details],
+        outputs=[md_out, source, status, details, dl],
         api_name="convert",
         js=BUSY_JS,
         show_progress="minimal",
@@ -920,13 +923,13 @@ with gr.Blocks(title="PASITA: plain text to Markdown") as demo:
     text_in.submit(
         convert,
         inputs=[text_in, candidates, structured],
-        outputs=[md_out, source, status, details],
+        outputs=[md_out, source, status, details, dl],
         api_visibility="undocumented",
         js=BUSY_JS,
         show_progress="minimal",
     )
     text_in.change(None, inputs=[text_in], outputs=[label_in], js=COUNTER_JS, api_visibility="undocumented")
-    clear.click(reset, inputs=None, outputs=[md_out, source, status, details, label_in], api_visibility="undocumented")
+    clear.click(reset, inputs=None, outputs=[md_out, source, status, details, dl, label_in], api_visibility="undocumented")
 
 demo.queue()
 demo.launch(_app=server, theme=theme, css=CSS, mcp_server=True)