Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 from transformers import pipeline
-from langdetect import detect
 
 # Load Arabic NLP model for intent classification
 intent_classifier = pipeline("text-classification", model="aubmindlab/bert-base-arabertv02")
 
+# Load language detection model
+language_detector = pipeline("text-classification", model="papluca/xlm-roberta-base-language-detection")
+
 # Omdurman National Bank-specific guidelines
 ONB_GUIDELINES = {
     "balance": "يمكنك التحقق من رصيدك عبر الإنترنت أو عبر تطبيق الهاتف الخاص ببنك أم درمان الوطني.",
@@ -32,10 +34,10 @@ INTENT_TO_RESPONSE = {
 }
 
 def detect_language(text):
-    try:
-        return detect(text)
-    except:
-        return "unknown"
+    # Use Hugging Face language detection model
+    result = language_detector(text)
+    language = result[0]['label']
+    return language
 
 def classify_intent(message: str):
     # Use NLP model to classify the user's intent