Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
-import logging  # For analytics
+import logging
+from transformers import pipeline
 
 # Omdurman National Bank-specific guidelines
 ONB_GUIDELINES = {
@@ -40,28 +41,66 @@ OPTION_TO_KEY = {
     "الاتصال بالبنك": "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": "الاتصال بالبنك"
+}
+
 # Set up logging for analytics
 logging.basicConfig(filename='chatbot_queries.log', level=logging.INFO)
 
-def respond(option: str):
-    # Log the selected option
-    logging.info(f"Selected Option: {option}")
+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, "عذرًا، لم يتم التعرف على الخيار المحدد.")
     
-    # Map the Arabic option to the English key
-    key = OPTION_TO_KEY.get(option)
+    # If the user types a message
+    elif message:
+        option = classify_intent(message)
+        key = OPTION_TO_KEY.get(option)
+        response = ONB_GUIDELINES.get(key, "عذرًا، لم أفهم سؤالك. الرجاء اختيار أحد الخيارات التالية.")
     
-    # Return the corresponding response
-    if key:
-        return ONB_GUIDELINES.get(key, "عذرًا، لم يتم التعرف على الخيار المحدد.")
     else:
-        return "عذرًا، لم يتم التعرف على الخيار المحدد."
+        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
 
 # Omdurman National Bank-specific interface
 with gr.Blocks(css=".gradio-container {direction: rtl;}") as demo:
     gr.Markdown("# <center>بنك أم درمان الوطني - المساعد المصرفي</center>")
     
     with gr.Tab("المحادثة"):
-        gr.Markdown("## اختر أحد الخيارات التالية:")
+        gr.Markdown("## اختر أحد الخيارات التالية أو اكتب سؤالك:")
         
         # Dropdown menu for questions
         dropdown = gr.Dropdown(
@@ -70,23 +109,27 @@ with gr.Blocks(css=".gradio-container {direction: rtl;}") as demo:
             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)
         
-        # Link dropdown and button to response function
+        # 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,
-            outputs=output
+            inputs=[dropdown, text_input],
+            outputs=[output, feedback_buttons]
         )
-    
-    with gr.Tab("الإرشادات المصرفية"):
-        gr.Markdown("## إرشادات بنك أم درمان الوطني")
-        for key, value in ONB_GUIDELINES.items():
-            gr.Markdown(f"**{key.capitalize()}**: {value}")
 
 if __name__ == "__main__":
     demo.launch(