Update app.py

app.py

@@ -1,55 +1,45 @@
-import gradio as gr
+iimport gradio as gr
 from transformers import pipeline
 
-# نموذج خفيف ومتوافق مع البيئات المجانية
-MODEL_NAME = "google/flan-t5-base"
+# نموذج عربي مفتوح المصدر من منصة معرفة
+MODEL_NAME = "marefa-nlp/marefa-mt5-small"
 
-# تهيئة الـ pipeline
 generator = pipeline(
     task="text2text-generation",
     model=MODEL_NAME,
     tokenizer=MODEL_NAME
 )
 
-def inspire_free(future, value, skill):
+def inspire_arabic(future, value, skill):
     if not future or not value or not skill:
-        return "⚠️ رجاءً أجب على جميع الأسئلة الثلاثة أولاً."
+        return "⚠️ رجاءً أجيبي على جميع الأسئلة الثلاثة أولاً."
 
     prompt = (
-        f"اكتب عبارة عربية قصيرة وملهمة تصف مستقبل شخص يريد {future}، "
+        f"صِغْ عبارة عربية قصيرة وملهمة تصف مستقبل شخص يريد {future}، "
         f"ويؤمن بـ {value}، ويمتلك مهارة {skill}. "
-        "لتكن شاعرية، إيجابية، وتبعث الأمل دون أي ترجمة للإنجليزية."
+        "لتكن شاعرية وإيجابية وتمنح الأمل."
     )
 
     try:
-        outputs = generator(
-            prompt,
-            max_length=120,
-            num_return_sequences=1,
-            clean_up_tokenization_spaces=True
-        )
-
-        if not outputs or "generated_text" not in outputs[0]:
-            return "✨ النموذج ما رجّع نص هذه المرة، جرّبي مرة أخرى."
-
+        outputs = generator(prompt, max_length=120, num_return_sequences=1)
         text = outputs[0]["generated_text"].strip()
         if not text:
-            return "✨ النموذج صامت هذه المرة، أعطيه مدخلات مختلفة وجربي من جديد."
+            return "✨ النموذج صامت هذه المرة، جرّبي كلمات مختلفة."
         return text
 
     except Exception as e:
         return f"⚠️ حدث خطأ أثناء التوليد: {e}"
 
 demo = gr.Interface(
-    fn=inspire_free,
+    fn=inspire_arabic,
     inputs=[
         gr.Textbox(label="🎯 حلمك المستقبلي"),
         gr.Textbox(label="💡 الأثر الذي تريدين تركه"),
         gr.Textbox(label="⚙️ مهارتك المميزة"),
     ],
     outputs=gr.Textbox(label="✨ رؤيتك المستقبلية", lines=4),
-    title="ذكاء يصمم مستقبلك (مجاني)",
-    description="أجيبي عن الأسئلة الثلاثة ودعي النموذج المفتوح المصدر يصيغ لك رؤية ملهمة لمستقبلك.",
+    title="ذكاء يصمم مستقبلك (نموذج عربي)",
+    description="أجيبي على الأسئلة الثلاثة ودعي الذكاء العربي يصيغ لك رؤية شاعرية لمستقبلك.",
     theme="soft"
 )