Update app.py

app.py

@@ -1,38 +1,27 @@
 import gradio as gr
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+from transformers import pipeline
 from langdetect import detect
 
-# تحميل نموذج متعدد اللغات (مثل Mistral)
-model_name = "mistralai/Mistral-7B-Instruct-v0.2"
+model_name = "aubmindlab/aragpt2-base"
+chatbot = pipeline("text-generation", model=model_name)
 
-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 "Please enter some text."
-    
-    # اكتشاف اللغة
+        return "يرجى إدخال نص."
     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()
+        lang = "ar"
+    response = chatbot(prompt, max_length=100, do_sample=True)[0]["generated_text"]
+    return response.strip()
 
-# الواجهة
 interface = gr.Interface(
     fn=reply,
-    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!"
+    inputs=gr.Textbox(label="📝 أدخل نصًا بأي لغة"),
+    outputs=gr.Textbox(label="💬 رد الوكيل"),
+    title="🌍 TunisAI-Agent — ذكاء صناعي متعدد اللغات",
+    description="وكيل ذكاء صناعي مجاني يعمل بأي لغة 🧠🌐"
 )
 
-# إطلاق التطبيق
 interface.launch(share=True)