Update app.py

app.py

@@ -1,18 +1,10 @@
 import gradio as gr
-from transformers import pipeline
 import re
 
-# Load Arabic NLP model for intent classification
-intent_classifier_ar = pipeline("text-classification", model="aubmindlab/bert-base-arabertv02")
-
-# Load English NLP model for zero-shot classification
-intent_classifier_en = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
-
 # Load language detection model only (smaller model)
 from transformers import pipeline
 language_detector = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
 
-
 # Omdurman National Bank-specific guidelines in Arabic
 ONB_GUIDELINES_AR = {
     "balance": "يمكنك التحقق من رصيدك عبر الإنترنت أو عبر تطبيق الهاتف الخاص ببنك الوطني.",
@@ -39,17 +31,44 @@ ONB_GUIDELINES_EN = {
     "contact": "Contact us at 249-123-456-789 or via email at [email protected]."
 }
 
+# Menu options in both languages
+MENU_AR = """
+قائمة الخدمات المصرفية:
+1. رصيد - استعلام عن رصيد حسابك
+2. بطاقة - الإبلاغ عن بطاقة مفقودة
+3. قرض - معلومات عن القروض
+4. تحويل - تحويل الأموال
+5. حساب - فتح حساب جديد
+6. فائدة - أسعار الفائدة
+7. فرع - مواقع الفروع
+8. ساعات - ساعات العمل
+9. اتصال - معلومات الاتصال
+"""
+
+MENU_EN = """
+Banking Services Menu:
+1. balance - Check your account balance
+2. card - Report a lost card
+3. loan - Information about loans
+4. transfer - Transfer funds
+5. account - Open a new account
+6. interest - Interest rates
+7. branch - Branch locations
+8. hours - Working hours
+9. contact - Contact information
+"""
+
 # Map intents to keywords (enhanced)
 INTENT_KEYWORDS = {
-    "balance": ["balance", "check balance", "account balance", "how much", "رصيد", "حساب", "كم المبلغ"],
-    "lost_card": ["lost", "card", "stolen", "missing", "فقدت", "بطاقة", "مسروقة", "ضائعة"],
-    "loan": ["loan", "borrow", "borrowing", "credit", "قرض", "استدانة", "إئتمان"],
-    "transfer": ["transfer", "send money", "payment", "تحويل", "ارسال", "دفع"],
-    "new_account": ["account", "open", "create", "new", "حساب", "فتح", "جديد", "إنشاء"],
-    "interest_rates": ["interest", "rate", "rates", "return", "فائدة", "نسبة", "عائد"],
-    "branches": ["branch", "location", "where", "office", "فرع", "موقع", "أين", "مكتب"],
-    "working_hours": ["hours", "time", "open", "close", "ساعات", "وقت", "مفتوح", "مغلق"],
-    "contact": ["contact", "phone", "email", "call", "اتصال", "هاتف", "بريد", "اتصل"]
+    "balance": ["balance", "check balance", "account balance", "how much", "رصيد", "حساب", "كم المبلغ", "1"],
+    "lost_card": ["lost", "card", "stolen", "missing", "فقدت", "بطاقة", "مسروقة", "ضائعة", "2"],
+    "loan": ["loan", "borrow", "borrowing", "credit", "قرض", "استدانة", "إئتمان", "3"],
+    "transfer": ["transfer", "send money", "payment", "تحويل", "ارسال", "دفع", "4"],
+    "new_account": ["account", "open", "create", "new", "حساب", "فتح", "جديد", "إنشاء", "5"],
+    "interest_rates": ["interest", "rate", "rates", "return", "فائدة", "نسبة", "عائد", "6"],
+    "branches": ["branch", "location", "where", "office", "فرع", "موقع", "أين", "مكتب", "7"],
+    "working_hours": ["hours", "time", "open", "close", "ساعات", "وقت", "مفتوح", "مغلق", "8"],
+    "contact": ["contact", "phone", "email", "call", "اتصال", "هاتف", "بريد", "اتصل", "9"]
 }
 
 def detect_language(text):
@@ -59,12 +78,20 @@ def detect_language(text):
     return language
 
 def classify_intent(message: str):
-    # Use keyword matching for all languages
-    message = message.lower()
+    # Check for menu request
+    menu_keywords = ["menu", "options", "help", "قائمة", "خيارات", "مساعدة"]
+    message_lower = message.lower()
+    
+    for keyword in menu_keywords:
+        if keyword in message_lower:
+            return "menu"
+    
+    # Use keyword matching for intent classification
     for intent_key, keywords in INTENT_KEYWORDS.items():
         for keyword in keywords:
-            if keyword.lower() in message:
+            if keyword.lower() in message_lower:
                 return intent_key
+                
     return "unknown"
 
 def respond(message: str):
@@ -83,24 +110,27 @@ def respond(message: str):
     
     # Classify the user's intent using keyword matching
     intent = classify_intent(message)
-       
+    
     # Prepare responses in both languages
     responses = {
         "ar": "",
         "en": ""
     }
     
+    # Special handling for menu request
+    if intent == "menu":
+        responses["ar"] = MENU_AR
+        responses["en"] = MENU_EN
+        return responses
+    
     # If intent is recognized, return the corresponding response
     if intent != "unknown":
         responses["ar"] = ONB_GUIDELINES_AR.get(intent, "عذرًا، لم يتم التعرف على الخيار المحدد.")
         responses["en"] = ONB_GUIDELINES_EN.get(intent, "Sorry, the selected option was not recognized.")
     else:
-        # Default response if no intent is matched
-        ar_options = ", ".join(list(ONB_GUIDELINES_AR.keys()))
-        en_options = ", ".join(list(ONB_GUIDELINES_EN.keys()))
-        
-        responses["ar"] = f"عذرًا، لم أفهم سؤالك. الرجاء إعادة الصياغة أو اختيار أحد الخيارات التالية: {ar_options}"
-        responses["en"] = f"Sorry, I didn't understand your question. Please rephrase or choose one of the following options: {en_options}"
+        # Default response if no intent is matched - show menu
+        responses["ar"] = "عذرًا، لم أفهم سؤالك. إليك قائمة بالخدمات المتاحة:" + MENU_AR
+        responses["en"] = "Sorry, I didn't understand your question. Here's a menu of available services:" + MENU_EN
     
     return responses
 
@@ -137,7 +167,7 @@ custom_css = """
 
 .header-section {
     background-color: #1a5276;
-    font-color: white;
+    color: white;
     padding: 1rem;
     border-radius: 10px;
     margin-bottom: 1rem;
@@ -155,6 +185,10 @@ custom_css = """
     text-align: right;
     margin-bottom: 1rem;
 }
+
+.menu-button {
+    margin-top: 0.5rem;
+}
 """
 
 # Chat interface with enhanced UI
@@ -190,6 +224,10 @@ with gr.Blocks(css=custom_css) as demo:
         with gr.Column(scale=1):
             submit_btn = gr.Button("Send | إرسال", variant="primary")
     
+    with gr.Row():
+        with gr.Column(elem_classes="menu-button"):
+            menu_btn = gr.Button("Show Menu | إظهار القائمة")
+    
     with gr.Row(elem_classes="footer-section"):
         gr.Markdown("© 2025 Omdurman National Bank. All Rights Reserved. | جميع الحقوق محفوظة لبنك أم درمان الوطني ٢٠٢٥ ©")
     
@@ -225,6 +263,19 @@ with gr.Blocks(css=custom_css) as demo:
         
         return "", chat_history
     
+    # Handle menu button click
+    def show_menu(chat_history, lang):
+        menu_responses = {
+            "ar": MENU_AR,
+            "en": MENU_EN
+        }
+        
+        # Add system message showing the menu
+        menu_text = menu_responses[lang]
+        chat_history.append([None, menu_text])
+        
+        return chat_history
+    
     # Link inputs and button to response function
     submit_btn.click(
         fn=on_submit,
@@ -232,6 +283,13 @@ with gr.Blocks(css=custom_css) as demo:
         outputs=[text_input, chat_box]
     )
     
+    # Link menu button to show menu function
+    menu_btn.click(
+        fn=show_menu,
+        inputs=[chat_box, selected_lang],
+        outputs=[chat_box]
+    )
+    
     # Also trigger on Enter key
     text_input.submit(
         fn=on_submit,