Update app.py

app.py

@@ -231,7 +231,7 @@ def _today_str():
 
 
 def decide_fiscal(note, issuer_status, parse, doc_radio,
-                  clarify_vat=None, clarify_tax_id=None):
+                  clarify_vat=None, clarify_tax_id=None, clarify_name=None):
     """
     Returns (issuer_dict, final_parse, questions_dict).
     questions_dict empty → ready to generate.
@@ -247,6 +247,8 @@ def decide_fiscal(note, issuer_status, parse, doc_radio,
               "is_company": False}
 
     # ── 2. Mandatory content ──────────────────────────────────────────────
+    if clarify_name and clarify_name.strip():
+        parse["client_name"] = clarify_name.strip()
     if not parse.get("client_name", "").strip():
         questions["client_name"] = "שם הלקוח / המשלם?"
     if not parse.get("items"):
@@ -503,44 +505,45 @@ def render_recommendations(recs):
 # ─────────────────────────────────────────────────────────────────────────────
 # 6. Main pipeline
 # ─────────────────────────────────────────────────────────────────────────────
-def generate(note, issuer_radio, doc_choice, clarify_vat, clarify_tax_id):
+def generate(note, issuer_radio, doc_choice, clarify_vat, clarify_tax_id, clarify_name):
     """
     Stateless single handler.
-    Returns: (status_md, parse_html, doc_html, recs_html, vat_group_visible, tax_group_visible)
+    Returns: (status_md, parse_html, doc_html, recs_html, name_vis, vat_vis, tax_vis)
     """
     note = (note or "").strip()
     if not note:
-        return "✏️ כתוב פתק הכנסה ולחץ הפק מסמך.", "", "", "", False, False
+        return "✏️ כתוב פתק הכנסה ולחץ הפק מסמך.", "", "", "", False, False, False
 
     issuer_status = issuer_radio if issuer_radio in ("פטור", "מורשה") else None
 
-    # Issuer check BEFORE model (saves ~90s on CPU)
     if not issuer_status:
-        return "👤 בחר תחילה: **עוסק פטור** או **עוסק מורשה** למעלה.", "", "", "", False, False
+        return "👤 בחר תחילה: **עוסק פטור** או **עוסק מורשה** למעלה.", "", "", "", False, False, False
 
     _lazy_init()
     parse = model_parse(note)
 
     if parse is None or _missing_fields(parse) == ["parse_failed"]:
         return ("🤔 לא הצלחתי לחלץ פרטים. נסח מחדש — לדוגמה: 'קיבלתי 500₪ ממשה על ייעוץ'.",
-                "", "", "", False, False)
+                "", "", "", False, False, False)
 
-    cvat = clarify_vat    if clarify_vat    not in (None, "לא צוין") else None
-    ctax = clarify_tax_id.strip() if clarify_tax_id and clarify_tax_id.strip() else None
+    cvat  = clarify_vat  if clarify_vat  not in (None, "לא צוין") else None
+    ctax  = clarify_tax_id.strip() if clarify_tax_id  and clarify_tax_id.strip()  else None
+    cname = clarify_name.strip()   if clarify_name     and clarify_name.strip()    else None
 
     issuer, final, questions = decide_fiscal(
-        note, issuer_status, parse, doc_choice, cvat, ctax)
+        note, issuer_status, parse, doc_choice, cvat, ctax, cname)
 
     if questions:
-        show_vat_q   = "vat_basis"    in questions
+        show_name_q  = "client_name"   in questions
+        show_vat_q   = "vat_basis"     in questions
         show_taxid_q = "client_tax_id" in questions
         return (_build_clarification_message(questions),
-                "", "", "", show_vat_q, show_taxid_q)
+                "", "", "", show_name_q, show_vat_q, show_taxid_q)
 
     try:
         completed = core.complete(issuer, final, _STATE["rng"])
     except Exception as e:
-        return f"⚠️ שגיאה בעיבוד: {e}", "", "", "", False, False
+        return f"⚠️ שגיאה בעיבוד: {e}", "", "", "", False, False, False
 
     show_vat = (final["doc_type"] != "receipt")
 
@@ -548,7 +551,7 @@ def generate(note, issuer_radio, doc_choice, clarify_vat, clarify_tax_id):
             render_parse_panel(final, completed, show_vat),
             render_document(completed, show_vat),
             render_recommendations(recommend(note)),
-            False, False)
+            False, False, False)
 
 
 # ─────────────────────────────────────────────────────────────────────────────
@@ -608,6 +611,13 @@ with gr.Blocks(title="Text2Receipt", theme=gr.themes.Soft(
     status_output = gr.Markdown("")
 
     # Clarification fields — appear only when needed
+    with gr.Group(visible=False) as clarify_name_group:
+        gr.HTML('<div dir="rtl" style="font-size:12px;font-weight:600;color:#7c3aed;'
+                'margin-bottom:6px;">👤 שם לקוח נדרש</div>')
+        clarify_name = gr.Textbox(
+            label="שם הלקוח / המשלם", placeholder="למשל: נועה כהן", value="",
+        )
+
     with gr.Group(visible=False) as clarify_vat_group:
         gr.HTML('<div dir="rtl" style="font-size:12px;font-weight:600;color:#7c3aed;'
                 'margin-bottom:6px;">💰 בסיס מע״מ נדרש</div>')
@@ -635,29 +645,33 @@ with gr.Blocks(title="Text2Receipt", theme=gr.themes.Soft(
     recs_output  = gr.HTML()
 
     # Handlers
-    def _on_generate(note, issuer_r, doc_c, cvat, ctax):
-        status, phtml, dhtml, rhtml, show_vat_q, show_taxid_q = generate(
-            note, issuer_r, doc_c, cvat, ctax)
+    def _on_generate(note, issuer_r, doc_c, cvat, ctax, cname):
+        status, phtml, dhtml, rhtml, show_name_q, show_vat_q, show_taxid_q = generate(
+            note, issuer_r, doc_c, cvat, ctax, cname)
         return (status, phtml, dhtml, rhtml,
+                gr.update(visible=show_name_q),
                 gr.update(visible=show_vat_q),
                 gr.update(visible=show_taxid_q))
 
     submit_btn.click(
         fn=_on_generate,
-        inputs=[note_input, issuer_radio, doc_choice, clarify_vat, clarify_tax_id],
+        inputs=[note_input, issuer_radio, doc_choice,
+                clarify_vat, clarify_tax_id, clarify_name],
         outputs=[status_output, parse_output, doc_output, recs_output,
-                 clarify_vat_group, clarify_tax_group],
+                 clarify_name_group, clarify_vat_group, clarify_tax_group],
     )
 
     def _on_clear():
-        return ("", None, "זיהוי אוטומטי", "לא צוין", "", "", "", "", "",
-                gr.update(visible=False), gr.update(visible=False))
+        return ("", None, "זיהוי אוטומטי", "לא צוין", "", "",
+                "", "", "", "",
+                gr.update(visible=False), gr.update(visible=False), gr.update(visible=False))
 
     clear_btn.click(
         fn=_on_clear,
-        outputs=[note_input, issuer_radio, doc_choice, clarify_vat, clarify_tax_id,
+        outputs=[note_input, issuer_radio, doc_choice,
+                 clarify_vat, clarify_tax_id, clarify_name,
                  status_output, parse_output, doc_output, recs_output,
-                 clarify_vat_group, clarify_tax_group],
+                 clarify_name_group, clarify_vat_group, clarify_tax_group],
     )
 
 # ─────────────────────────────────────────────────────────────────────────────