code update in app.py

app.py

@@ -59,8 +59,14 @@ def parse_banking_sms(raw_text: str) -> dict:
         except json.JSONDecodeError:
             pass
 
-    return {"date": None, "type": None, "amount": None,
-            "category": None, "last4": None, "is_transaction": False}
+    return {
+        "date": None,
+        "type": None,
+        "amount": None,
+        "category": None,
+        "last4": None,
+        "is_transaction": False
+    }
 
 # --------------------------------------------------
 # 3. Chatbot response handler
@@ -92,24 +98,43 @@ This appears to be a promotional or informational message.
     return [(raw_text, response)]
 
 # --------------------------------------------------
-# 4. Gradio Interface
+# 4. Gradio Blocks UI
 # --------------------------------------------------
-iface = gr.Interface(
-    fn=chatbot_response,
-    inputs=gr.Textbox(lines=3, placeholder="Paste your banking SMS/email here…"),
-    outputs=gr.Chatbot(label="Banking SMS Parser", height=400),
-    title="🏦 Banking SMS JSON Parser Chatbot",
-    description="Paste any banking SMS or email to extract structured JSON.",
-    examples=[
-        ["Your A/c XX1234 debited for 5000 on 15-Jan-2024 at AMAZON"],
-        ["2500 credited to A/c 9876 on 20-Dec-2023 from PAYROLL"],
-        ["Card 4321 used for 120 at STARBUCKS on 10-Nov-2023"],
-    ],
-    show_api=False
-)
+with gr.Blocks() as demo:
+    gr.Markdown("""
+    # 🏦 Banking SMS JSON Parser Chatbot
+
+    Paste any banking SMS or email below – no special formatting needed!
+
+    **Features:**
+    - ✅ Detects real transactions vs promotional messages
+    - ✅ Extracts date, amount, merchant, category, account info
+    - ✅ Works with all Indian & global banking formats
+    """)
+
+    chatbot = gr.Chatbot(label="Banking SMS Parser", height=400)
+    msg = gr.Textbox(
+        lines=3,
+        placeholder="Paste your banking SMS/email here…",
+        label="Input Message"
+    )
+
+    gr.Examples(
+        examples=[
+            ["Your A/c XX1234 debited for 5000 on 15-Jan-2024 at AMAZON"],
+            ["2500 credited to A/c 9876 on 20-Dec-2023 from PAYROLL"],
+            ["Card 4321 used for 120 at STARBUCKS on 10-Nov-2023"],
+        ],
+        inputs=msg
+    )
+
+    msg.submit(chatbot_response, inputs=msg, outputs=chatbot)
+    msg.submit(lambda: "", None, msg)
+
+    gr.Markdown("---\n**Model:** `rawsun00001/banking-sms-json-parser-v6-merged`")
 
 # --------------------------------------------------
-# 5. App launcher (Spaces-compatible)
+# 5. App launcher
 # --------------------------------------------------
 if __name__ == "__main__":
-    iface.launch(share=True)
+    demo.launch(share=True)