app.py

import os
import gradio as gr
from transformers import pipeline

# نحدد كاش للموديل عشان يقلل زمن التحميل
os.environ["TRANSFORMERS_CACHE"] = "/app/cache"

# اسم الموديل بتاعك
MODEL_REPO = "Reem1/hakayti-merged"

# مثال few-shot زي ما هو بالظبط
few_shot_example = """<s>[User]: اسمي كريم وأحب الفضاء. 
اكتب لي قصة طويلة للأطفال (40 جملة على الأقل) مكتوبة بأسلوب قصصي ممتع.
[Assistant]:
كريم كان يحب النظر إلى السماء كل ليلة. كان يجلس في شرفة بيته، يراقب النجوم ويعدها واحدة تلو الأخرى. أحيانًا كان يتخيل أن النجوم مثل مصابيح صغيرة تضيء له الطريق نحو أحلامه الكبيرة.  
في المدرسة، لاحظ المعلم شغف كريم بالفضاء. أعطاه كتابًا مليئًا بالصور الملونة للكواكب والأقمار. انبهر كريم عندما رأى صور كوكب زحل وحلقاته اللامعة، وأخذ يتخيل نفسه يطير حوله.  
قرر كريم أن يصنع نموذجًا صغيرًا للمجموعة الشمسية. جمع كرات صغيرة وطلّاها بالألوان: الأزرق للأرض، الأحمر للمريخ، والأصفر للشمس. علّقها بخيوط في غرفته، فصارت كأنها سماء صغيرة يعيش فيها.  
في أحد الأيام، أعلنت المدرسة عن مسابقة للرسم بعنوان "رحلة إلى الفضاء". رسم كريم لوحة كبيرة يظهر فيها صاروخ ينطلق من الأرض، وحوله أطفال يلوّحون بأيديهم. فاز كريم بالجائزة الأولى، وكان سعيدًا جدًا لأن حلمه بالفضاء أصبح أقرب مما يتخيل.  
في المساء، جلس كريم مع والده على سطح المنزل. أشار والده إلى القمر وقال: "تخيّل يا كريم أن العلماء يدرسون الآن كيف يعيش الناس يومًا ما في الفضاء". ابتسم كريم وأجاب: "سأكون أنا واحدًا منهم".  
وهكذا تعلم كريم أن الحلم يبدأ بخطوة صغيرة، وأن حب المعرفة يمكن أن يفتح له أبوابًا واسعة نحو المستقبل.
</s>"""

# تحميل الموديل
pipe = pipeline(
    "text-generation",
    model=MODEL_REPO,
    tokenizer=MODEL_REPO,
    max_new_tokens=900,
    temperature=0.9,
    top_p=0.95
)

# نفس دالة توليد القصة
def generate_story(name, hobby):
    prompt = f"""{few_shot_example}
[User]: اسمي {name} وأحب {hobby}. 
اكتب لي قصة طويلة للأطفال (40 جملة على الأقل) مكتوبة بنفس الأسلوب السابق،
وتأكد أن القصة كلها تدور حول {hobby} فقط.
[Assistant]:"""

    out = pipe(
        prompt,
        max_new_tokens=900,
        temperature=0.9,
        top_p=0.95,
        do_sample=True
    )

    full_text = out[0]["generated_text"]
    story = full_text.split("[Assistant]:")[-1].replace("</s>", "").strip()
    return story

# واجهة Gradio
iface = gr.Interface(
    fn=generate_story,
    inputs=[
        gr.Textbox(label="✍️ ما اسمك؟"),
        gr.Textbox(label="🎨 ما هو اهتمامك أو الشيء الذي تحبه؟")
    ],
    outputs=gr.Textbox(label="📖 قصتك"),
    title="📖 حكايتي - صانع القصص للأطفال",
    description="أدخل اسمك واهتماماتك، ودعني أكتب لك قصة طويلة للأطفال بأسلوب ممتع ❤️"
)

iface.launch()