Update app.py

app.py

@@ -29,27 +29,23 @@ def load_llm():
 
 load_llm()
 
-# --- The "A-Grade" Logic: Hardcoded Prompt Engineering ---
-# UPDATE: Added 'progress=gr.Progress()' here to enable the loading bar
+# --- Generation Function ---
 def generate_poem(format_type, persona, topic, progress=gr.Progress()):
-    """
-    Takes user variables and wraps them in a sophisticated prompt 
-    that the user never sees.
-    """
     # 1. VISUAL FEEDBACK: Immediately show the bar
+    # This now works because the Textbox below has a fixed height!
     progress(0, desc="Consulting the Muse...")
-    
+    time.sleep(0.2) # Force a tiny pause so the eye catches the bar
+
     if not llm:
         return "Error: Model not loaded."
     
     if not topic:
         return "Please enter a topic!"
 
-    # 2. VISUAL FEEDBACK: Update status while we build the prompt
+    # 2. Progress Update
     progress(0.2, desc=f"Summoning {persona}...")
-    time.sleep(0.3) # Small pause so the user actually sees the message
+    time.sleep(0.3)
 
-    # 1. Define Persona Instructions (The "Voice")
     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.",
@@ -57,7 +53,6 @@ def generate_poem(format_type, persona, topic, progress=gr.Progress()):
         "Shakespearean Actor": "You are a dramatic Shakespearean actor. Use Early Modern English (thee, thou), dramatic flair, and iambic rhythm."
     }
     
-    # 2. Define Format Instructions (The "Constraint")
     format_map = {
         "Limerick": "Write a Limerick. It must have exactly 5 lines. The rhyme scheme must be A-A-B-B-A. It should be humorous.",
         "Sonnet": "Write a Sonnet. It must have exactly 14 lines and express a complete thought or feeling.",
@@ -77,10 +72,9 @@ def generate_poem(format_type, persona, topic, progress=gr.Progress()):
         f"### Response:\n"
     )
 
-    # 3. VISUAL FEEDBACK: Indicate the heavy lifting is starting
-    progress(0.4, desc="Writing your masterpiece (this may take a moment)...")
+    # 3. Progress Update
+    progress(0.4, desc="Drafting Masterpiece...")
 
-    # 4. Generate
     output = llm(
         prompt=full_prompt,
         max_tokens=MAX_NEW_TOKENS,
@@ -89,20 +83,18 @@ def generate_poem(format_type, persona, topic, progress=gr.Progress()):
         echo=False
     )
     
-    # 4. VISUAL FEEDBACK: Done
-    progress(0.9, desc="Final polish...")
+    # 4. Progress Update
+    progress(0.9, desc="Polishing rhymes...")
     time.sleep(0.2)
     
     return output['choices'][0]['text'].strip()
 
-# --- The Innovative UI Layout ---
+# --- UI Layout ---
 with gr.Blocks(title="The Poetry Workshop", theme=gr.themes.Soft()) as demo:
     
-    # Header
     gr.Markdown("# 🖋️ The Poetry Workshop")
     gr.Markdown("Your own personal muse to help you get started with poetry.")
     
-    # The "Sentence" Layout (Mad Libs Style)
     with gr.Group():
         with gr.Row(equal_height=True):
             gr.Markdown("### I want to write a ")
@@ -124,7 +116,6 @@ with gr.Blocks(title="The Poetry Workshop", theme=gr.themes.Soft()) as demo:
                 scale=3
             )
     
-    # Topic Input
     with gr.Row(equal_height=True):
         gr.Markdown("### about this topic: ")
         topic_input = gr.Textbox(
@@ -135,11 +126,18 @@ with gr.Blocks(title="The Poetry Workshop", theme=gr.themes.Soft()) as demo:
         )
         generate_btn = gr.Button("✨ Create Masterpiece", variant="primary", scale=1)
 
-    # Output Area
     gr.Markdown("---")
-    output_display = gr.Markdown(label="Your Poem")
+    
+    # --- CRITICAL CHANGE HERE ---
+    # Swapped gr.Markdown for gr.Textbox.
+    # 'lines=10' forces the box to be visible immediately, so the loading bar has a place to live.
+    output_display = gr.Textbox(
+        label="Your Poem", 
+        lines=12,            # Ensures a big empty box exists on load
+        interactive=False,   # Read-only
+        placeholder="Your masterpiece will appear here..."
+    )
 
-    # Event Wiring
     generate_btn.click(
         fn=generate_poem,
         inputs=[format_dropdown, persona_dropdown, topic_input],