Update app.py

app.py

@@ -1,57 +1,71 @@
 import gradio as gr
-import random
 from transformers import pipeline
+import random
+
+# المرحلة 1: توليد جملة بسيطة بالعربية (منطقية وملهمة)
+base_generator = pipeline(
+    "text2text-generation",
+    model="google/flan-t5-small"   # يمكن تغييره لاحقًا لنموذج عربي
+)
 
-# نموذج عربي فعلي (GPT2)
-generator = pipeline(
+# المرحلة 2: تحويل الجملة إلى أسلوب شعري
+poetry_generator = pipeline(
     "text-generation",
     model="akhooli/gpt2-small-arabic-poetry"
 )
 
-def inspire(future, value, skill):
+def dual_inspire(future, value, skill):
     if not future or not value or not skill:
         return "⚠️ رجاءً أجيبي على جميع الأسئلة أولاً."
 
-    prompt = (
-        f"اكتب جملة قصيرة وملهمة بالعربية تصف مستقبل شخص يريد {future}، "
+    # المرحلة الأولى: توليد فكرة أساسية
+    idea_prompt = (
+        f"اكتب جملة قصيرة بالعربية تصف مستقبل شخص يريد {future}، "
         f"ويؤمن بـ {value}، ويمتلك مهارة {skill}. "
-        "لتكن الجملة شاعرية، مفعمة بالأمل، وفيها لمسة إنسانية وجمال."
+        "لتكن بسيطة، مفهومة، وتلهم القارئ."
     )
 
     try:
-        result = generator(prompt, max_new_tokens=60, temperature=1.1, top_p=0.95)
-        text = result[0]["generated_text"].strip()
+        idea_result = base_generator(idea_prompt, max_new_tokens=40)
+        idea_text = idea_result[0]["generated_text"].strip()
+        if not idea_text or len(idea_text) < 10:
+            idea_text = "تمضي نحو غدٍ تصنع فيه أثرًا من ضوءٍ وإصرار."
+
+        # المرحلة الثانية: تحويل الجملة إلى شعر
+        poetry_prompt = (
+            f"حوّل الجملة التالية إلى بيت أو سطر شعري جميل بالعربية: {idea_text}"
+        )
+
+        poetic_result = poetry_generator(
+            poetry_prompt,
+            max_new_tokens=60,
+            temperature=1.2,
+            top_p=0.95
+        )
+        poetic_text = poetic_result[0]["generated_text"].strip()
 
         # تنظيف المخرجات
-        if text.startswith(prompt):
-            text = text[len(prompt):].strip()
-        text = text.replace("\n", " ").split(".")[0].strip()
-
-        # fallback ذكي
-        fallbacks = [
-            "تمضي بخطواتٍ من ضوء، تصنع أثرها بين الحلم والإصرار.",
-            "تزهرُ رؤيتها كفجرٍ يعلن ولادة أملٍ جديد.",
-            "تسافر بثقةٍ إلى الغد، تحمل في قلبها نبض الطموح.",
-            "تكتب حكايتها بين الإبداع والعزيمة، لتضيء الطريق للآخرين."
-        ]
-        if not text or len(text) < 10:
-            text = random.choice(fallbacks)
-
-        return text
+        if poetic_text.startswith(poetry_prompt):
+            poetic_text = poetic_text[len(poetry_prompt):].strip()
+
+        poetic_text = poetic_text.replace("\n", " ").split(".")[0].strip()
+
+        return f"🌱 الجملة الأساسية:\n{idea_text}\n\n🎨 النسخة الشاعرية:\n{poetic_text}"
 
     except Exception as e:
         return f"⚠️ حدث خطأ أثناء التوليد: {e}"
 
+# واجهة Gradio
 demo = gr.Interface(
-    fn=inspire,
+    fn=dual_inspire,
     inputs=[
         gr.Textbox(label="🎯 حلمك المستقبلي"),
         gr.Textbox(label="💡 الأثر الذي تريدين تركه"),
         gr.Textbox(label="⚙️ مهارتك المميزة"),
     ],
-    outputs=gr.Textbox(label="🌟 رؤيتك المستقبلية الشاعرية", lines=3),
-    title="ذكاء يصمم مستقبلك بإحساس إنساني وشاعري 💫",
-    description="أجيبي على الأسئلة الثلاثة ودعي الذكاء الاصطناعي يكتب لك جملة تلهمك وتعبّر عنك. 🌷",
+    outputs=gr.Textbox(label="🌟 الجملة النهائية (عقل + شعر)", lines=6),
+    title="ثنائية الإلهام – من الفكرة إلى الشعر 💫",
+    description="الذكاء الاصطناعي يصيغ لك فكرة تلهم المستقبل، ثم يحولها إلى شعر نابض بالحياة. ✨",
     theme="soft"
 )