Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
-from transformers import pipeline  # For NLP
+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")
@@ -30,6 +31,12 @@ INTENT_TO_RESPONSE = {
     "contact": "contact"
 }
 
+def detect_language(text):
+    try:
+        return detect(text)
+    except:
+        return "unknown"
+
 def classify_intent(message: str):
     # Use NLP model to classify the user's intent
     result = intent_classifier(message)
@@ -37,6 +44,13 @@ def classify_intent(message: str):
     return INTENT_TO_RESPONSE.get(intent, "unknown")
 
 def respond(message: str):
+    # Detect language
+    language = detect_language(message)
+    
+    # If the language is not Arabic, return a default response
+    if language != "ar":
+        return "عذرًا، هذا المساعد يدعم اللغة العربية فقط. الرجاء إعادة الصياغة بالعربية."
+    
     # Classify the user's intent using NLP
     intent = classify_intent(message)