Update app.py

app.py

@@ -1,10 +1,6 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
 import logging  # For analytics
 
-# Use Arabic-optimized model
-client = InferenceClient("aubmindlab/aragpt2-base")
-
 # Omdurman National Bank-specific guidelines
 ONB_GUIDELINES = {
     "balance": "يمكنك التحقق من رصيدك عبر الإنترنت أو عبر تطبيق الهاتف الخاص ببنك أم درمان الوطني.",
@@ -31,46 +27,21 @@ logging.basicConfig(filename='chatbot_queries.log', level=logging.INFO)
 def respond(
     message: str,
     history: list[list[str]],
-    system_message="أنت مساعد مصرفي لبنك أم درمان الوطني. قدم معلومات عامة فقط.",
-    max_tokens=256,
-    temperature=0.5,
-    top_p=0.9,
 ):
     # Log the query
     logging.info(f"Query: {message}")
     
     # Check for Sudanese dialect phrases
     if message in SUDANESE_EXAMPLES:
-        yield SUDANESE_EXAMPLES[message]
-        return
-    
-    # Force Arabic responses
-    prompt = f"باللغة العربية: {message}"
+        return SUDANESE_EXAMPLES[message]
     
-    # Format history in messages format
-    messages = [{"role": "system", "content": system_message}]
+    # Check for specific keywords in the message
+    for keyword, response in ONB_GUIDELINES.items():
+        if keyword in message:
+            return response
     
-    for user_msg, bot_msg in history:
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
-        if bot_msg:
-            messages.append({"role": "assistant", "content": bot_msg})
-
-    messages.append({"role": "user", "content": prompt})
-
-    # Generate Arabic response
-    response = ""
-    for chunk in client.text_generation(
-        prompt=prompt,
-        max_new_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-        repetition_penalty=1.2,
-        stop_sequences=["\n", "؟", "!"],  # Stop at natural sentence endings
-    ):
-        response += chunk
-        yield response
+    # Default response if no keyword is matched
+    return "عذرًا، لم أفهم سؤالك. الرجاء إعادة الصياغة أو اختيار أحد الخيارات التالية: " + ", ".join(ONB_GUIDELINES.keys())
 
 # Omdurman National Bank-specific interface
 with gr.Blocks(css=".gradio-container {direction: rtl;}") as demo: