Update app.py

app.py

@@ -12,6 +12,18 @@ def simple_detect_language(text):
         return "ar"
     return "en"
 
+# Try to import transformers for Arabic NLP if available
+try:
+    from transformers import pipeline
+    arabic_nlp = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
+    ARABIC_NLP_AVAILABLE = True
+    
+    def detect_language_with_nlp(text):
+        result = arabic_nlp(text)
+        return result[0]['label']
+except ImportError:
+    ARABIC_NLP_AVAILABLE = False
+
 # Import customer service enhancements
 try:
     from customer_service_enhancements import (
@@ -200,8 +212,14 @@ def respond(message: str):
             "en": "Please type your question."
         }
     
-    # Detect language using simple function
-    language = simple_detect_language(message)
+    # Detect language using NLP if available, otherwise use simple detection
+    if ARABIC_NLP_AVAILABLE:
+        language = detect_language_with_nlp(message)
+        # If the language is neither Arabic nor English, default to English
+        if language != "ar" and language != "en":
+            language = "en"
+    else:
+        language = simple_detect_language(message)
     
     # Classify the user's intent using keyword matching
     intent = classify_intent(message)
@@ -374,39 +392,6 @@ custom_css = """
     background-color: #f9f9f9;
 }
 
-.typing-indicator {
-    display: inline-block;
-    width: 50px;
-    text-align: left;
-}
-
-.typing-indicator span {
-    display: inline-block;
-    width: 8px;
-    height: 8px;
-    background-color: #1a5276;
-    border-radius: 50%;
-    margin-right: 5px;
-    animation: typing 1s infinite;
-}
-
-.typing-indicator span:nth-child(2) {
-    animation-delay: 0.2s;
-}
-
-.typing-indicator span:nth-child(3) {
-    animation-delay: 0.4s;
-}
-
-@keyframes typing {
-    0%, 100% {
-        transform: translateY(0);
-    }
-    50% {
-        transform: translateY(-5px);
-    }
-}
-
 .live-agent-button {
     background-color: #27ae60;
     color: white;
@@ -487,8 +472,6 @@ a:hover {
 
 # Chat interface with simplified UI for Hugging Face
 with gr.Blocks(css=custom_css) as demo:
-    # Store conversation history
-    state = gr.State(value=[])
     # Store selected language
     selected_lang = gr.State(value="ar")
     # Store user name for personalization
@@ -510,12 +493,21 @@ with gr.Blocks(css=custom_css) as demo:
     with gr.Row(elem_classes="chat-container"):
         chatbot = gr.Chatbot(height=400)
     
-    # Quick action buttons (will be populated based on language)
+    # Quick action buttons container
     with gr.Row(elem_classes="quick-actions", visible=True) as quick_actions_container:
-        quick_action_buttons = []
-        for i in range(9):  # Create 9 buttons (one for each intent)
-            button = gr.Button("", visible=False, elem_classes="quick-action-button")
-            quick_action_buttons.append(button)
+        balance_btn = gr.Button("تحقق من الرصيد", elem_classes="quick-action-button")
+        lost_card_btn = gr.Button("الإبلاغ عن بطاقة مفقودة", elem_classes="quick-action-button")
+        loan_btn = gr.Button("معلومات القرض", elem_classes="quick-action-button")
+    
+    with gr.Row(elem_classes="quick-actions", visible=True):
+        transfer_btn = gr.Button("تحويل الأموال", elem_classes="quick-action-button")
+        new_account_btn = gr.Button("فتح حساب جديد", elem_classes="quick-action-button")
+        interest_btn = gr.Button("أسعار الفائدة", elem_classes="quick-action-button")
+    
+    with gr.Row(elem_classes="quick-actions", visible=True):
+        branches_btn = gr.Button("مواقع الفروع", elem_classes="quick-action-button")
+        hours_btn = gr.Button("ساعات العمل", elem_classes="quick-action-button")
+        contact_btn = gr.Button("اتصل بنا", elem_classes="quick-action-button")
     
     with gr.Row():
         with gr.Column(scale=8):
@@ -539,28 +531,42 @@ with gr.Blocks(css=custom_css) as demo:
         gr.Markdown("© 2025 Omdurman National Bank. All Rights Reserved. | جميع الحقوق محفوظة لبنك أم درمان الوطني ٢٠٢٥ ©")
     
     # Update language state and quick action buttons when language is changed
-    def update_language_and_buttons(lang):
+    def update_language(lang):
         language_code = "ar" if lang == "العربية" else "en"
         
         # Get the appropriate quick actions based on language
         quick_actions = QUICK_ACTIONS_AR if language_code == "ar" else QUICK_ACTIONS_EN
         
-        # Update button visibility and text
-        button_updates = []
-        for i, button in enumerate(quick_action_buttons):
-            if i < len(quick_actions):
-                button_updates.append(gr.Button.update(visible=True, value=quick_actions[i]["text"]))
-            else:
-                button_updates.append(gr.Button.update(visible=False))
-        
-        return [language_code] + button_updates
+        # Update button text based on language
+        return [
+            language_code,
+            quick_actions[0]["text"],  # balance
+            quick_actions[1]["text"],  # lost_card
+            quick_actions[2]["text"],  # loan
+            quick_actions[3]["text"],  # transfer
+            quick_actions[4]["text"],  # new_account
+            quick_actions[5]["text"],  # interest
+            quick_actions[6]["text"],  # branches
+            quick_actions[7]["text"],  # working_hours
+            quick_actions[8]["text"]   # contact
+        ]
     
     # Connect language button to update function
-    outputs = [selected_lang] + quick_action_buttons
     language_btn.change(
-        fn=update_language_and_buttons,
+        fn=update_language,
         inputs=language_btn,
-        outputs=outputs
+        outputs=[
+            selected_lang,
+            balance_btn,
+            lost_card_btn,
+            loan_btn,
+            transfer_btn,
+            new_account_btn,
+            interest_btn,
+            branches_btn,
+            hours_btn,
+            contact_btn
+        ]
     )
     
     # Handle message submission
@@ -584,7 +590,7 @@ with gr.Blocks(css=custom_css) as demo:
                 break
         
         # Add user message to chat history
-        chat_history.append([message, None])
+        chat_history = chat_history + [[message, None]]
         
         # Get response
         responses = respond(message)
@@ -616,85 +622,87 @@ with gr.Blocks(css=custom_css) as demo:
         menu_text = menu_responses[lang]
         
         # Add system message showing the menu
-        chat_history.append([None, menu_text.replace("\n", "<br>")])
+        chat_history = chat_history + [[None, menu_text.replace("\n", "<br>")]]
         
         return chat_history
     
     # Handle quick action button clicks
-    def handle_quick_action(button_index, chat_history, lang, name):
+    def handle_quick_action(intent, chat_history, lang, name):
         # Get the appropriate quick actions based on language
         quick_actions = QUICK_ACTIONS_AR if lang == "ar" else QUICK_ACTIONS_EN
         
-        if button_index < len(quick_actions):
-            # Get the intent for this button
-            intent = quick_actions[button_index]["intent"]
-            button_text = quick_actions[button_index]["text"]
-            
-            # Add button text as user message
-            chat_history.append([button_text, None])
-            
-            # Get the response for this intent
-            if CUSTOMER_SERVICE_ENHANCEMENTS_AVAILABLE:
-                # Use enhanced response if available
-                response_ar = get_enhanced_response(intent, "ar", name)
-                response_en = get_enhanced_response(intent, "en", name)
-            else:
-                # Add a greeting phrase at the beginning
-                greeting_ar = get_random_phrase("greeting", "ar")
-                greeting_en = get_random_phrase("greeting", "en")
-                
-                # Add the main response
-                main_response_ar = ONB_GUIDELINES_AR.get(intent, "")
-                main_response_en = ONB_GUIDELINES_EN.get(intent, "")
-                
-                # Add a follow-up phrase at the end
-                follow_up_ar = get_random_phrase("follow_up", "ar")
-                follow_up_en = get_random_phrase("follow_up", "en")
-                
-                # Combine all parts
-                response_ar = f"{greeting_ar}<br><br>{main_response_ar}<br><br>{follow_up_ar}"
-                response_en = f"{greeting_en}<br><br>{main_response_en}<br><br>{follow_up_en}"
-            
-            responses = {
-                "ar": response_ar,
-                "en": response_en
-            }
-            
-            # Select response based on language
-            response = responses[lang]
-            
-            # Personalize response if name is available
-            if name:
-                if lang == "ar":
-                    response = response.replace("مرحبًا", f"مرحبًا {name}")
-                else:
-                    response = response.replace("Welcome", f"Welcome {name}")
-                    response = response.replace("Hello", f"Hello {name}")
+        # Find the button text for this intent
+        button_text = ""
+        for action in quick_actions:
+            if action["intent"] == intent:
+                button_text = action["text"]
+                break
+        
+        # Add button text as user message
+        chat_history = chat_history + [[button_text, None]]
+        
+        # Get the response for this intent
+        if CUSTOMER_SERVICE_ENHANCEMENTS_AVAILABLE:
+            # Use enhanced response if available
+            response_ar = get_enhanced_response(intent, "ar", name)
+            response_en = get_enhanced_response(intent, "en", name)
+        else:
+            # Add a greeting phrase at the beginning
+            greeting_ar = get_random_phrase("greeting", "ar")
+            greeting_en = get_random_phrase("greeting", "en")
             
-            # Update bot response in chat history
-            chat_history[-1][1] = response
+            # Add the main response
+            main_response_ar = ONB_GUIDELINES_AR.get(intent, "")
+            main_response_en = ONB_GUIDELINES_EN.get(intent, "")
             
-            # Log the interaction
-            log_interaction(button_text, response, intent, lang)
+            # Add a follow-up phrase at the end
+            follow_up_ar = get_random_phrase("follow_up", "ar")
+            follow_up_en = get_random_phrase("follow_up", "en")
             
+            # Combine all parts
+            response_ar = f"{greeting_ar}<br><br>{main_response_ar}<br><br>{follow_up_ar}"
+            response_en = f"{greeting_en}<br><br>{main_response_en}<br><br>{follow_up_en}"
+        
+        responses = {
+            "ar": response_ar,
+            "en": response_en
+        }
+        
+        # Select response based on language
+        response = responses[lang]
+        
+        # Personalize response if name is available
+        if name:
+            if lang == "ar":
+                response = response.replace("مرحبًا", f"مرحبًا {name}")
+            else:
+                response = response.replace("Welcome", f"Welcome {name}")
+                response = response.replace("Hello", f"Hello {name}")
+        
+        # Update bot response in chat history
+        chat_history[-1][1] = response
+        
+        # Log the interaction
+        log_interaction(button_text, response, intent, lang)
+        
         return chat_history
     
     # Handle live agent button click
     def connect_to_live_agent(chat_history, lang):
         message = "Connecting to a live customer service agent. Please wait a moment..." if lang == "en" else "جاري الاتصال بوكيل خدمة العملاء. يرجى الانتظار لحظة..."
-        chat_history.append([None, message])
+        chat_history = chat_history + [[None, message]]
         return chat_history
     
     # Handle satisfaction survey button click
     def show_satisfaction_survey(chat_history, lang):
         if CUSTOMER_SERVICE_ENHANCEMENTS_AVAILABLE:
             survey_html = offer_satisfaction_survey(lang)
-            chat_history.append([None, survey_html])
+            chat_history = chat_history + [[None, survey_html]]
         else:
             # Simple survey message if enhancements not available
             title = "استطلاع رضا العملاء" if lang == "ar" else "Customer Satisfaction Survey"
             message = f"<div class='satisfaction-survey'><h3>{title}</h3><p>{'نشكرك على استخدام المساعد المصرفي الافتراضي!' if lang == 'ar' else 'Thank you for using our virtual banking assistant!'}</p></div>"
-            chat_history.append([None, message])
+            chat_history = chat_history + [[None, message]]
         
         return chat_history
     
@@ -727,12 +735,59 @@ with gr.Blocks(css=custom_css) as demo:
     )
     
     # Link quick action buttons to handler function
-    for i, button in enumerate(quick_action_buttons):
-        button.click(
-            fn=lambda idx=i, chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action(idx, chat, lang, name),
-            inputs=[chatbot, selected_lang, user_name],
-            outputs=[chatbot]
-        )
+    balance_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("balance", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
+    
+    lost_card_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("lost_card", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
+    
+    loan_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("loan", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
+    
+    transfer_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("transfer", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
+    
+    new_account_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("new_account", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
+    
+    interest_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("interest_rates", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
+    
+    branches_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("branches", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
+    
+    hours_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("working_hours", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
+    
+    contact_btn.click(
+        fn=lambda chat=chatbot, lang=selected_lang, name=user_name: handle_quick_action("contact", chat, lang, name),
+        inputs=[chatbot, selected_lang, user_name],
+        outputs=[chatbot]
+    )
     
     # Also trigger on Enter key
     text_input.submit(
@@ -752,24 +807,13 @@ with gr.Blocks(css=custom_css) as demo:
         chat_history.append([None, welcome_ar])
         chat_history.append([None, welcome_en])
         
-        # Get the appropriate quick actions for Arabic (default)
-        quick_actions = QUICK_ACTIONS_AR
-        
-        # Update button visibility and text
-        button_updates = []
-        for i, button in enumerate(quick_action_buttons):
-            if i < len(quick_actions):
-                button_updates.append(gr.Button.update(visible=True, value=quick_actions[i]["text"]))
-            else:
-                button_updates.append(gr.Button.update(visible=False))
-        
-        return [chat_history] + button_updates
+        return chat_history
     
     # Initialize the chat when the app starts
     demo.load(
         fn=init_chat,
         inputs=[],
-        outputs=[chatbot] + quick_action_buttons
+        outputs=[chatbot]
     )
 
 if __name__ == "__main__":