Update app.py

app.py

@@ -1,40 +1,62 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-# Load your exported model from the Hub
-model_id = "akhooli/gpt2-small-arabic"
+# Arabic-capable generative model
+MODEL_ID = "akhooli/gpt2-small-arabic"
 
-tokenizer = AutoTokenizer.from_pretrained(model_id)
-model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
+# Load tokenizer & model
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+model = AutoModelForCausalLM.from_pretrained(MODEL_ID, device_map="auto")
 
+# Create generation pipeline with tuned parameters
 pipe = pipeline(
     "text-generation",
     model=model,
     tokenizer=tokenizer,
     max_new_tokens=200,
-    temperature=0.9,
-    top_p=0.95,
-    repetition_penalty=1.05
+    temperature=0.85,       # Slightly creative but not chaotic
+    top_p=0.92,             # Nucleus sampling
+    repetition_penalty=1.1, # Avoids echoing the same line
+    do_sample=True
 )
 
-def generate_lyrics(style, mood, prompt):
-    full_prompt = f"أنت كاتب أغاني لبناني. النمط: {style}، المزاج: {mood}.\n{prompt}"
-    out = pipe(full_prompt)[0]["generated_text"]
-    return out[len(full_prompt):].strip()
+# Persistent role instruction
+ROLE = (
+    "أنت شاعر غنائي لبناني محترف، تكتب كلمات أغاني جذابة "
+    "بصيغة مقاطع [المقطع الأول]، [اللازمة]، [المقطع الثاني]، "
+    "تمزج الصور الشعرية باللهجة اللبنانية وتضيف لمسة موسيقية واضحة."
+)
+
+def generate_lyrics(style, mood, user_prompt):
+    scaffold = (
+        f"{ROLE}\n"
+        f"النمط: {style}\n"
+        f"المزاج: {mood}\n"
+        f"أكتب الأغنية بالتنسيق التالي:\n"
+        "[المقطع الأول]:\n"
+        "[اللازمة]:\n"
+        "[المقطع الثاني]:\n\n"
+        f"الموضوع: {user_prompt}\n\n"
+    )
+
+    result = pipe(scaffold)[0]["generated_text"]
+    # Trim the role/metadata from output
+    clean = result[len(scaffold):].strip()
+    return clean
 
-styles = ["بوب", "طرب", "دبكة", "راب"]
-moods = ["رومانسي", "حنين", "احتفالي", "حزين"]
+styles = ["بوب", "طرب", "دبكة", "راب", "مزج"]
+moods = ["رومانسي", "ساخر", "احتفالي", "حزين", "حنين"]
 
 demo = gr.Interface(
     fn=generate_lyrics,
     inputs=[
-        gr.Dropdown(styles, label="النمط"),
-        gr.Dropdown(moods, label="المزاج"),
-        gr.Textbox(lines=3, placeholder="أدخل فكرة أو جملة بداية...")
+        gr.Dropdown(styles, label="🎼 النمط"),
+        gr.Dropdown(moods, label="🎭 المزاج"),
+        gr.Textbox(lines=3, placeholder="💡 الفكرة أو الجملة الافتتاحية للأغنية")
     ],
-    outputs=gr.Textbox(label="الكلمات الناتجة"),
-    title="مولد الأغاني اللبنانية",
-    description="أدخل النمط والمزاج وفكرة البداية، واحصل على كلمات قابلة للغناء."
+    outputs=gr.Textbox(label="🎤 كلمات الأغنية الناتجة", lines=10),
+    title="مولّد الأغاني اللبنانية",
+    description="اختر النمط والمزاج، ثم أدخل فكرة الأغنية. سيقوم المولّد بإنشاء كلمات بمقاطع ولازمات."
 )
 
 if __name__ == "__main__":