Update app.py

app.py

@@ -49,18 +49,21 @@ load_llm()
 
 # --- Persona and Format Maps (Hardened Constraints) ---
 persona_map = {
-    "A Grumpy Pirate": "You are a salty, grumpy pirate captain. Use nautical slang, complain about the sea, and say 'Arrr'.",
-    "A Melancholy Philosopher": "You are a deep, existential philosopher. Use complex vocabulary, metaphor, and a somber tone.",
-    "A Hyperactive 5-Year-Old": "You are a very excited 5-year-old. Use simple words, lots of exclamation marks, and get distracted easily.",
-    "Shakespearean Actor": "You are a dramatic Shakespearean actor. Use Early Modern English (thee, thou), dramatic flair, and iambic rhythm."
+        "Grumpy Pirate": "You are a grumpy and annoyed pirate captain. Use salty nautical slang, complain about the sea and use 'Arrr' and 'matey'.",
+        "Philosopher": "You are a philosopher. Use metaphors and reflective musings about the human condition.",
+        "Ancient Wizard": "You are an ancient wizard. Speak in mystical tones. Use references to magic.",
+        "Shakespearean Actor": "You are a Shakespearean actor. Use Early Modern English (thee, thou) and dramatic flair.",
+        "Sarcastic Teenager": "You are a sarcastic teenager. Use dry humor."
 }
 
 format_map = {
-    "Limerick": "The poem MUST have exactly 5 lines (A-A-B-B-A rhyme scheme) and be HUMOROUS.",
-    "Sonnet": "The poem MUST have exactly 14 lines and express a complete thought or feeling.",
-    "Haiku": "The poem ABSOLUTELY MUST have exactly 3 lines. The syllable structure should be 5-7-5 (approximate). Focus on nature and brevity.",
-    "Free Verse": "The poem should be written in Free Verse. It should have 6-10 lines and focus on imagery and emotion without strict rhyme or rhythm.",
-    "Ballad": "The poem MUST tell a narrative story using four-line stanzas (quatrains) and a simple rhyme scheme like A-B-C-B. It should be 4-6 stanzas long."
+
+        "Free Verse": "Write in Free Verse about the topic. Focus on vivid imagery and emotion.",
+        "Ballad": "Write a Ballad about the topic. It should tell a narrative story with a clear beginning, middle, and end. Use four-line stanzas (quatrains) and a simple rhyme scheme like A-B-C-B. Keep the language simple and musical.",
+        "Ode": "Write an Ode about the topic. Use elevated, expressive language to praise or celebrate the subject. Focus on strong emotion, rich imagery, and admiration.",
+        "Elegy": "Write an Elegy about the topic. Use a somber, reflective tone to mourn a loss or contemplate death. Focus on sorrow, remembrance, and emotional depth.",
+        "Hymn": "Write a Hymn about the topic. Use a reverent, uplifting tone. Focus on praise, devotion, or spiritual reflection, with a rhythmic, chant-like flow.",
+        "Epic": "Write an Epic about the topic. Use grand, dramatic language to tell a heroic story."
 }
 
 # --- Refinement Instruction Map ---
@@ -185,8 +188,8 @@ with gr.Blocks(title="The Poetry Workshop", theme=gr.themes.Soft()) as demo:
         with gr.Row(equal_height=True):
             gr.Markdown("### I want to write a ")
             format_dropdown = gr.Dropdown(
-                choices=["Limerick", "Sonnet", "Haiku", "Ballad", "Free Verse"],
-                value="Limerick",
+                choices=["Free Verse", "Ballad", "Ode", "Elegy", "Hymn", "Epic"],
+                value="Free Verse",
                 label="Poem Type",
                 show_label=False,
                 container=False,
@@ -194,8 +197,8 @@ with gr.Blocks(title="The Poetry Workshop", theme=gr.themes.Soft()) as demo:
             )
             gr.Markdown("### in the style of ")
             persona_dropdown = gr.Dropdown(
-                choices=["A Grumpy Pirate", "A Melancholy Philosopher", "A Hyperactive 5-Year-Old", "Shakespearean Actor"],
-                value="A Grumpy Pirate",
+                choices=["Grumpy Pirate", "Philosopher", "Ancient Wizard", "Shakespearean Actor", "Sarcastic Teenager"],
+                value="Grumpy Pirate",
                 label="Persona",
                 show_label=False,
                 container=False,