Update app.py

app.py

@@ -1,24 +1,38 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+from langdetect import detect
 
-# تحميل النموذج
-chatbot = pipeline("text-generation", model="aubmindlab/aragpt2-base")
+# تحميل نموذج متعدد اللغات (مثل Mistral)
+model_name = "mistralai/Mistral-7B-Instruct-v0.2"
 
-# الدالة التي تستقبل وترد
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+chatbot = pipeline("text-generation", model=model, tokenizer=tokenizer)
+
+# دالة الرد
 def reply(prompt):
     if prompt.strip() == "":
-        return "يرجى إدخال نص."
-    response = chatbot(prompt, max_length=100, do_sample=True)[0]["generated_text"]
-    return response
+        return "Please enter some text."
+    
+    # اكتشاف اللغة
+    try:
+        lang = detect(prompt)
+    except:
+        lang = "en"
+
+    # توليد الرد
+    result = chatbot(prompt, max_length=150, do_sample=True, top_k=50, temperature=0.7)[0]["generated_text"]
+    return result.strip()
 
-# بناء الواجهة
+# الواجهة
 interface = gr.Interface(
     fn=reply,
-    inputs=gr.Textbox(label="prompt"),
-    outputs=gr.Textbox(label="output"),
-    title="TunisAI-Agent",
-    description="وكيل ذكاء صناعي مجاني مطور في تونس 🇹🇳"
+    inputs=gr.Textbox(label="📝 Input Prompt (Any Language)"),
+    outputs=gr.Textbox(label="💬 AI Response"),
+    title="🌍 TunisAI-Agent — Multilingual Assistant",
+    description="Free multilingual AI agent built in Tunisia 🇹🇳. Supports 100+ languages!"
 )
 
-# إطلاق التطبيق مع رابط عام
+# إطلاق التطبيق
 interface.launch(share=True)
+