Update app.py

app.py

@@ -1,4 +1,8 @@
 import gradio as gr
+from transformers import pipeline  # For NLP
+
+# Load Arabic NLP model for intent classification
+intent_classifier = pipeline("text-classification", model="aubmindlab/bert-base-arabertv02")
 
 # Omdurman National Bank-specific guidelines
 ONB_GUIDELINES = {
@@ -13,27 +17,40 @@ ONB_GUIDELINES = {
     "contact": "الاتصال بنا على الرقم 249-123-456-789 أو عبر البريد الإلكتروني [email protected]."
 }
 
-# Map keywords to responses
-KEYWORD_TO_RESPONSE = {
-    "رصيد": "balance",
-    "بطاقة": "lost_card",
-    "قرض": "loan",
-    "تحويل": "transfer",
-    "حساب": "new_account",
-    "فائدة": "interest_rates",
-    "فرع": "branches",
-    "ساعات": "working_hours",
-    "اتصال": "contact"
+# Map intents to responses
+INTENT_TO_RESPONSE = {
+    "balance": "balance",
+    "lost_card": "lost_card",
+    "loan": "loan",
+    "transfer": "transfer",
+    "new_account": "new_account",
+    "interest_rates": "interest_rates",
+    "branches": "branches",
+    "working_hours": "working_hours",
+    "contact": "contact"
 }
 
+def classify_intent(message: str):
+    # Use NLP model to classify the user's intent
+    result = intent_classifier(message)
+    intent = result[0]['label']
+    return INTENT_TO_RESPONSE.get(intent, "unknown")
+
 def respond(message: str):
-    # Check for keywords in the message
-    for keyword, key in KEYWORD_TO_RESPONSE.items():
+    # Classify the user's intent using NLP
+    intent = classify_intent(message)
+    
+    # If intent is recognized, return the corresponding response
+    if intent != "unknown":
+        return ONB_GUIDELINES.get(intent, "عذرًا، لم يتم التعرف على الخيار المحدد.")
+    
+    # Fallback to keyword matching if NLP doesn't recognize the intent
+    for keyword, key in INTENT_TO_RESPONSE.items():
         if keyword in message:
             return ONB_GUIDELINES.get(key, "عذرًا، لم يتم التعرف على الخيار المحدد.")
     
-    # Default response if no keyword is matched
-    return "عذرًا، لم أفهم سؤالك. الرجاء إعادة الصياغة أو اختيار أحد الخيارات التالية: " + ", ".join(KEYWORD_TO_RESPONSE.keys())
+    # Default response if no intent or keyword is matched
+    return "عذرًا، لم أفهم سؤالك. الرجاء إعادة الصياغة أو اختيار أحد الخيارات التالية: " + ", ".join(INTENT_TO_RESPONSE.keys())
 
 # Chat interface
 with gr.Blocks(css=".gradio-container {direction: rtl;}") as demo: