Update app.py

app.py

@@ -1,59 +1,47 @@
 import gradio as gr
-from transformers import pipeline
 import random
+from transformers import pipeline
 
-# المرحلة 1: توليد جملة بسيطة بالعربية (منطقية وملهمة)
-base_generator = pipeline(
-    "text2text-generation",
-    model="google/flan-t5-small"   # يمكن تغييره لاحقًا لنموذج عربي
+# المرحلة 1: نموذج توليد الفكرة (عربي فعلي)
+idea_generator = pipeline(
+    "text-generation",
+    model="akhooli/gpt2-small-arabic",
 )
 
-# المرحلة 2: تحويل الجملة إلى أسلوب شعري
+# المرحلة 2: نموذج تحويل الفكرة إلى شعر
 poetry_generator = pipeline(
     "text-generation",
-    model="akhooli/gpt2-small-arabic-poetry"
+    model="akhooli/gpt2-small-arabic-poetry",
 )
 
 def dual_inspire(future, value, skill):
     if not future or not value or not skill:
-        return "⚠️ رجاءً أجيبي على جميع الأسئلة أولاً."
+        return "⚠️ رجاءً أجيبي على جميع الحقول."
 
-    # المرحلة الأولى: توليد فكرة أساسية
+    # 1️⃣ توليد الفكرة الأساسية
     idea_prompt = (
         f"اكتب جملة قصيرة بالعربية تصف مستقبل شخص يريد {future}، "
         f"ويؤمن بـ {value}، ويمتلك مهارة {skill}. "
-        "لتكن بسيطة، مفهومة، وتلهم القارئ."
+        "لتكن جملة بسيطة، واضحة، وملهمة."
     )
+    idea = idea_generator(idea_prompt, max_new_tokens=40, temperature=0.9, top_p=0.9)[0]["generated_text"]
+    # تنظيف النص
+    idea = idea.replace("\n", " ").strip().split(".")[0]
+    if len(idea) < 15:
+        idea = "يسعى نحو غدٍ يصنع فيه أثرًا من ضوءٍ وإصرار."
+
+    # 2️⃣ توليد النسخة الشعرية بناءً على الفكرة
+    poetry_prompt = (
+        f"حوّل الجملة التالية إلى شعر عربي جميل:\n"
+        f"\"{idea}\"\n"
+        "لتكن شاعرية بإحساس إنساني ولغة عاطفية."
+    )
+    poem = poetry_generator(poetry_prompt, max_new_tokens=60, temperature=1.2, top_p=0.95)[0]["generated_text"]
+    # تنظيف الشعر
+    poem = poem.replace(poetry_prompt, "").strip().split(".")[0]
 
-    try:
-        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 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}"
+    # 3️⃣ دمج النتيجة
+    return f"🌱 الفكرة الأساسية:\n{idea}\n\n🎨 النسخة الشاعرية:\n{poem}"
 
 # واجهة Gradio
 demo = gr.Interface(
@@ -63,9 +51,9 @@ demo = gr.Interface(
         gr.Textbox(label="💡 الأثر الذي تريدين تركه"),
         gr.Textbox(label="⚙️ مهارتك المميزة"),
     ],
-    outputs=gr.Textbox(label="🌟 الجملة النهائية (عقل + شعر)", lines=6),
+    outputs=gr.Textbox(label="🌟 الجملة النهائية (فكرة + شعر)", lines=6),
     title="ثنائية الإلهام – من الفكرة إلى الشعر 💫",
-    description="الذكاء الاصطناعي يصيغ لك فكرة تلهم المستقبل، ثم يحولها إلى شعر نابض بالحياة. ✨",
+    description="الذكاء الاصطناعي يولّد فكرة عربية ثم يحولها إلى شعر جميل يعبّر عنك.",
     theme="soft"
 )