Update app.py

app.py

@@ -63,8 +63,7 @@ format_map = {
     "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."
 }
 
-# --- Refinement Instruction Map (New) ---
-# Maps the button label to the strict instruction given to the model
+# --- Refinement Instruction Map ---
 REFINEMENT_INSTRUCTIONS = {
     "Make it Shorter": "Rewrite the poem to be significantly shorter, reducing the line count by at least 3 lines, while keeping the core meaning.",
     "Use Flowery Language": "Rewrite the poem using highly elaborate, decorative, and verbose language with many adjectives and metaphors.",
@@ -130,7 +129,7 @@ def analyze_poem(poem_text, progress=gr.Progress()):
     output_poet = llm_analyst(
         prompt=poet_prompt,
         max_tokens=200, 
-        temperature=0.4, # Lowered temperature for more stable analysis
+        temperature=0.4, 
         echo=False
     )
         
@@ -138,7 +137,8 @@ def analyze_poem(poem_text, progress=gr.Progress()):
     progress(1.0, desc="Analysis complete!")
     return poet_suggestion
 
-# --- LLM Function: Step 3 (Refine Poem - NEW) ---
+# --- LLM Function: Step 3 (Refine Poem - MODIFIED SIGNATURE FOR GRADIO) ---
+# NOTE: This function now accepts an *explicit* instruction string, which Gradio will bind.
 def refine_poem(poem_text, refinement_instruction, progress=gr.Progress()):
     if not llm_generator:
         return "Error: Generator model not loaded."
@@ -233,15 +233,17 @@ with gr.Blocks(title="The Poetry Workshop", theme=gr.themes.Soft()) as demo:
 
     gr.Markdown("---")
     
-    ## ⚙️ Poem Refinement Section (New)
+    ## ⚙️ Poem Refinement Section
     with gr.Group():
         gr.Markdown("### Refine Poem")
         gr.Markdown("Click a button below to automatically revise the current poem in the **'Your Poem'** box.")
+        
+        # NOTE: Buttons are created but we will reference them explicitly below
+        buttons_refinement = {}
         with gr.Row():
-            # Dynamically create buttons based on the map
-            buttons = []
             for label in REFINEMENT_INSTRUCTIONS.keys():
-                buttons.append(gr.Button(label, scale=1, variant="light"))
+                # Store the button object in the dictionary for later reference
+                buttons_refinement[label] = gr.Button(label, scale=1, variant="light")
     
     # --- Event Handling ---
     
@@ -259,14 +261,44 @@ with gr.Blocks(title="The Poetry Workshop", theme=gr.themes.Soft()) as demo:
         outputs=poet_suggestion_display
     )
 
-    # 3. Refinement Events (Wired to the refinement buttons)
-    # Iterate through the buttons and assign a click event to each one
-    for i, (label, instruction) in enumerate(REFINEMENT_INSTRUCTIONS.items()):
-        buttons[i].click(
-            fn=lambda p, i=instruction: refine_poem(p, i), # Use a lambda to capture the specific instruction
-            inputs=[output_display],
-            outputs=[output_display]
-        )
+    # 3. Refinement Events (Rewritten to remove lambda)
+    
+    # Create explicit functions for each button click, passing the desired instruction
+    
+    # Make it Shorter
+    buttons_refinement["Make it Shorter"].click(
+        fn=refine_poem,
+        inputs=[output_display, gr.State(REFINEMENT_INSTRUCTIONS["Make it Shorter"])],
+        outputs=[output_display]
+    )
+
+    # Use Flowery Language
+    buttons_refinement["Use Flowery Language"].click(
+        fn=refine_poem,
+        inputs=[output_display, gr.State(REFINEMENT_INSTRUCTIONS["Use Flowery Language"])],
+        outputs=[output_display]
+    )
+    
+    # Make it More Intense
+    buttons_refinement["Make it More Intense"].click(
+        fn=refine_poem,
+        inputs=[output_display, gr.State(REFINEMENT_INSTRUCTIONS["Make it More Intense"])],
+        outputs=[output_display]
+    )
+    
+    # Add a Title
+    buttons_refinement["Add a Title"].click(
+        fn=refine_poem,
+        inputs=[output_display, gr.State(REFINEMENT_INSTRUCTIONS["Add a Title"])],
+        outputs=[output_display]
+    )
+    
+    # Be More Humorous
+    buttons_refinement["Be More Humorous"].click(
+        fn=refine_poem,
+        inputs=[output_display, gr.State(REFINEMENT_INSTRUCTIONS["Be More Humorous"])],
+        outputs=[output_display]
+    )
 
 if __name__ == "__main__":
     demo.launch()