Fix local refinement mode for HF Space

app.py

@@ -1109,7 +1109,7 @@ footer {
 
 /* Custom placeholder */
 #pdf_upload .wrap::after {
-  content: "☁️\A Drag & drop your PDF file here\A or click to browse";
+  content: "☁️\\A Drag & drop your PDF file here\\A or click to browse";
   white-space: pre-line;
   color: #334155 !important;
   -webkit-text-fill-color: #334155 !important;

src/paper2lab/inference/pipeline.py

@@ -27,7 +27,7 @@ from paper2lab.inference.visual_explainer import explain_figures_and_tables
 from paper2lab.inference.auto_select import build_auto_best_card
 
 
-RefinementMode = Literal["none", "nemotron"]
+RefinementMode = Literal["none", "local", "nemotron"]
 
 
 class PaperPipeline:
@@ -50,7 +50,13 @@ class PaperPipeline:
         pdf_path: str | Path,
         refinement_mode: RefinementMode | None = None,
     ) -> Dict[str, Any]:
-        active_refinement_mode = refinement_mode or self.refinement_mode
+        selected_refinement_mode = (
+            refinement_mode or self.refinement_mode or "local"
+        ).lower().strip()
+
+        active_refinement_mode = (
+            "none" if selected_refinement_mode == "local" else selected_refinement_mode
+        )
 
         extracted = extract_pdf(pdf_path, engine=self.pdf_engine)
         paper_card = build_paper_card(extracted)
@@ -78,8 +84,8 @@ class PaperPipeline:
             return_comparison=True,
         )
         auto_selection = build_auto_best_card(
-        local_card=paper_card,
-        refinement=refinement,
+            local_card=paper_card,
+            refinement=refinement,
             )
 
         final_paper_card = auto_selection["final_paper_card"]
@@ -112,7 +118,7 @@ class PaperPipeline:
 
         result: Dict[str, Any] = {
             "status": "ok",
-            "refinement_mode": active_refinement_mode,
+            "refinement_mode": selected_refinement_mode,
             "paper_card": paper_card,
             "paper_card_refined": refinement.get("after_refinement", paper_card),
             "paper_card_final": final_paper_card,

src/paper2lab/inference/refinement.py

@@ -5,7 +5,7 @@ from typing import Any, Dict, Literal
 from paper2lab.inference.nemotron_refiner import refine_with_nemotron
 
 
-RefinementMode = Literal["none", "nemotron"]
+RefinementMode = Literal["none", "local", "nemotron"]
 
 
 def refine_optional(
@@ -14,14 +14,16 @@ def refine_optional(
     return_comparison: bool = True,
 ) -> Dict[str, Any]:
     """
-    Default: keep local hardcoded/rule-based extraction.
-    Optional: refine the candidate with Nemotron.
+    none/local: keep local rule-based extraction.
+    nemotron: refine the candidate with Nemotron.
     """
 
-    if mode == "none":
+    mode = (mode or "none").lower().strip()
+
+    if mode in {"none", "local"}:
         return {
             "status": "skipped",
-            "mode": "none",
+            "mode": mode,
             "before_refinement": paper_card,
             "after_refinement": paper_card,
             "diff_summary": {
@@ -67,4 +69,15 @@ def refine_optional(
                 },
             }
 
-    raise ValueError(f"Unsupported refinement mode: {mode}")
+    return {
+        "status": "error",
+        "mode": mode,
+        "error": f"Unsupported refinement mode: {mode}",
+        "before_refinement": paper_card,
+        "after_refinement": paper_card,
+        "diff_summary": {
+            "changed_fields": [],
+            "added_fields": [],
+            "removed_fields": [],
+        },
+    }