code update in app.py

app.py

@@ -73,7 +73,28 @@ def parse_banking_sms(raw_text: str) -> dict:
 # --------------------------------------------------
 # 3. Chatbot response handler
 # --------------------------------------------------
-def chatbot_response(raw_text):
+def chatbot_response(raw_text, chat_history):
+    result = parse_banking_sms(raw_text)
+    if result["is_transaction"]:
+        response = f"""✅ Transaction Detected!
+
+📅 Date: {result['date']}
+💳 Type: {result['type'].title() if result['type'] else 'N/A'}
+💰 Amount: {result['amount']}
+🏪 Category: {result['category']}
+🔢 Last 4 Digits: {result['last4']}
+"""
+    else:
+        response = """ℹ️ Non-Transaction Message
+
+This appears to be a promotional or informational message.
+"""
+    chat_history = chat_history or []
+    chat_history.append({"role": "user", "content": raw_text})
+    chat_history.append({"role": "assistant", "content": response})
+
+    return chat_history, ""
+
     result = parse_banking_sms(raw_text)
     if result["is_transaction"]:
         response = f"""✅ **Transaction Detected!**
@@ -128,7 +149,7 @@ with gr.Blocks() as demo:
 )
 
 
-    msg.submit(chatbot_response, inputs=msg, outputs=chatbot)
+    msg.submit(chatbot_response, inputs=[msg, chatbot], outputs=[chatbot, msg])
     msg.submit(lambda: "", None, msg)
 
     gr.Markdown("---\n**Model:** `rawsun00001/banking-sms-json-parser-v6-merged`")