Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import logging
-from transformers import pipeline
 
 # Omdurman National Bank-specific guidelines
 ONB_GUIDELINES = {
@@ -15,85 +14,41 @@ ONB_GUIDELINES = {
     "contact": "الاتصال بنا على الرقم 249-123-456-789 أو عبر البريد الإلكتروني [email protected]."
 }
 
-# Arabic menu options
-menu_options = [
-    "التحقق من الرصيد",
-    "الإبلاغ عن فقدان البطاقة",
-    "شروط الحصول على قرض",
-    "تحويل الأموال",
-    "فتح حساب جديد",
-    "أسعار الفائدة",
-    "فروع البنك",
-    "ساعات العمل",
-    "الاتصال بالبنك"
-]
-
-# Map Arabic menu options to English keys
-OPTION_TO_KEY = {
-    "التحقق من الرصيد": "balance",
-    "الإبلاغ عن فقدان البطاقة": "lost_card",
-    "شروط الحصول على قرض": "loan",
-    "تحويل الأموال": "transfer",
-    "فتح حساب جديد": "new_account",
-    "أسعار الفائدة": "interest_rates",
-    "فروع البنك": "branches",
-    "ساعات العمل": "working_hours",
-    "الاتصال بالبنك": "contact"
-}
-
-# Load Arabic intent classification model
-intent_classifier = pipeline("text-classification", model="aubmindlab/bert-base-arabertv02")
-
-# Map intents to menu options
-INTENT_TO_OPTION = {
-    "balance": "التحقق من الرصيد",
-    "lost_card": "الإبلاغ عن فقدان البطاقة",
-    "loan": "شروط الحصول على قرض",
-    "transfer": "تحويل الأموال",
-    "new_account": "فتح حساب جديد",
-    "interest_rates": "أسعار الفائدة",
-    "branches": "فروع البنك",
-    "working_hours": "ساعات العمل",
-    "contact": "الاتصال بالبنك"
+# Map keywords to responses
+KEYWORD_TO_RESPONSE = {
+    "رصيد": "balance",
+    "بطاقة": "lost_card",
+    "قرض": "loan",
+    "تحويل": "transfer",
+    "حساب": "new_account",
+    "فائدة": "interest_rates",
+    "فرع": "branches",
+    "ساعات": "working_hours",
+    "اتصال": "contact"
 }
 
 # Set up logging for analytics
 logging.basicConfig(filename='chatbot_queries.log', level=logging.INFO)
 
-def classify_intent(message: str):
-    # Classify the user's intent
-    result = intent_classifier(message)
-    intent = result[0]['label']
-    return INTENT_TO_OPTION.get(intent, "عذرًا، لم أفهم سؤالك. الرجاء اختيار أحد الخيارات التالية.")
-
-def respond(option: str, message: str, history: list):
-    # If the user selects from the dropdown
-    if option:
-        key = OPTION_TO_KEY.get(option)
-        response = ONB_GUIDELINES.get(key, "عذرًا، لم يتم التعرف على الخيار المحدد.")
+def respond(message: str, history: list):
+    # Log the query
+    logging.info(f"Query: {message}")
     
-    # If the user types a message
-    elif message:
-        option = classify_intent(message)
-        key = OPTION_TO_KEY.get(option)
-        response = ONB_GUIDELINES.get(key, "عذرًا، لم أفهم سؤالك. الرجاء اختيار أحد الخيارات التالية.")
+    # Check for keywords in the message
+    for keyword, key in KEYWORD_TO_RESPONSE.items():
+        if keyword in message:
+            return ONB_GUIDELINES.get(key, "عذرًا، لم يتم التعرف على الخيار المحدد.")
     
-    else:
-        response = "عذرًا، لم يتم تحديد سؤال."
-    
-    # Add follow-up questions based on intent
-    if key == "loan":
-        response += "\n\nما هو نوع القرض الذي تبحث عنه؟"
-    elif key == "new_account":
-        response += "\n\nهل تريد فتح حساب شخصي أو حساب تجاري؟"
-    
-    # Add feedback buttons
-    feedback_buttons = gr.Row([
-        gr.Button("👍", variant="secondary"),
-        gr.Button("👎", variant="secondary")
-    ])
-    
-    return response, feedback_buttons
+    # Default response if no keyword is matched
+    return "عذرًا، لم أفهم سؤالك. الرجاء إعادة الصياغة أو اختيار أحد الخيارات التالية: " + ", ".join(KEYWORD_TO_RESPONSE.keys())
+
+# Main menu with submenus
+main_menu = {
+    "الحسابات": ["التحقق من الرصيد", "فتح حساب جديد"],
+    "القروض": ["شروط الحصول على قرض", "أسعار الفائدة"],
+    "الفروع": ["فروع البنك", "ساعات العمل"],
+    "الدعم": ["الإبلاغ عن فقدان البطاقة", "الاتصال بالبنك"]
+}
 
 # Omdurman National Bank-specific interface
 with gr.Blocks(css=".gradio-container {direction: rtl;}") as demo:
@@ -102,34 +57,27 @@ with gr.Blocks(css=".gradio-container {direction: rtl;}") as demo:
     with gr.Tab("المحادثة"):
         gr.Markdown("## اختر أحد الخيارات التالية أو اكتب سؤالك:")
         
-        # Dropdown menu for questions
-        dropdown = gr.Dropdown(
-            choices=menu_options,
-            label="اختر سؤالك",
-            interactive=True
-        )
-        
-        # Free-text input
-        text_input = gr.Textbox(label="أو اكتب سؤالك هنا")
-        
-        # Submit button
-        submit_btn = gr.Button("إرسال")
-        
-        # Output textbox for responses
-        output = gr.Textbox(label="الرد", interactive=False)
-        
-        # Feedback buttons
-        feedback_buttons = gr.Row([
-            gr.Button("👍", variant="secondary"),
-            gr.Button("👎", variant="secondary")
-        ])
-        
-        # Link inputs and button to response function
-        submit_btn.click(
-            fn=respond,
-            inputs=[dropdown, text_input],
-            outputs=[output, feedback_buttons]
+        # Chat interface
+        chatbot = gr.ChatInterface(
+            respond,
+            examples=[
+                "كيف يمكنني التحقق من رصيدي؟",  # Check balance
+                "أريد الإبلاغ عن فقدان بطاقتي",  # Report lost card
+                "ما هي شروط الحصول على قرض؟",  # Loan eligibility
+                "ما هي ساعات العمل؟",  # Working hours
+                "أين يوجد أقرب فرع؟"  # Branch locations
+            ]
         )
+    
+    with gr.Tab("القائمة الرئيسية"):
+        gr.Markdown("## القائمة الرئيسية")
+        for category, options in main_menu.items():
+            with gr.Accordion(category):
+                for option in options:
+                    gr.Button(option).click(
+                        fn=lambda opt=option: respond(opt, []),
+                        outputs=chatbot.chatbot
+                    )
 
 if __name__ == "__main__":
     demo.launch(