Update app.py

app.py

@@ -1,15 +1,14 @@
-
-
 import gradio as gr
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
+import time  # Imported to add small pauses for effect
 
 # --- Configuration ---
 MODEL_REPO = "Kezovic/iris-q4gguf-lora-test"
 MODEL_FILE = "Llama-3.2-1B-Instruct.Q4_K_M.gguf"
-CONTEXT_WINDOW = 2048 # Reduced slightly for speed, poetry is short
-MAX_NEW_TOKENS = 512  # Poems rarely need more than this
-TEMPERATURE = 0.9     # Slightly higher for creativity
+CONTEXT_WINDOW = 2048 
+MAX_NEW_TOKENS = 400  
+TEMPERATURE = 0.8     
 
 # --- Model Loading ---
 llm = None
@@ -31,18 +30,25 @@ def load_llm():
 load_llm()
 
 # --- The "A-Grade" Logic: Hardcoded Prompt Engineering ---
-def generate_poem(format_type, persona, topic):
+# UPDATE: Added 'progress' argument to the function signature
+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. Update Progress: Start
+    progress(0, desc="Initializing Muse...")
+    
     if not llm:
         return "Error: Model not loaded."
     
     if not topic:
         return "Please enter a topic!"
 
-    # 1. Define Persona Instructions (The "Voice")
+    # 2. Update Progress: Constructing Prompt
+    progress(0.2, desc=f"Summoning {persona}...")
+    time.sleep(0.5) # Artificial pause so user sees the message (optional)
+
     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.",
@@ -50,9 +56,6 @@ def generate_poem(format_type, persona, topic):
         "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")
-    # We use structural formats (Rhyme/Lines) rather than syllable counting (Hexameter)
-    # to prevent the model from failing at math.
     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.",
@@ -61,8 +64,6 @@ def generate_poem(format_type, persona, topic):
         "Ballad": "Write a Ballad. It must 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."
     }
 
-    # 3. Construct the "Super Prompt"
-    # This is the invisible work that gets you the "A".
     selected_voice = persona_map.get(persona, "You are a helpful assistant.")
     selected_constraint = format_map.get(format_type, "Write a poem.")
     
@@ -74,6 +75,9 @@ def generate_poem(format_type, persona, topic):
         f"### Response:\n"
     )
 
+    # 3. Update Progress: Generation Started
+    progress(0.5, desc="Drafting Masterpiece (This may take a moment)...")
+
     # 4. Generate
     output = llm(
         prompt=full_prompt,
@@ -83,14 +87,18 @@ def generate_poem(format_type, persona, topic):
         echo=False
     )
     
+    # 5. Update Progress: Done
+    progress(0.9, desc="Polishing rhymes...")
+    time.sleep(0.5)
+    
     return output['choices'][0]['text'].strip()
 
 # --- The Innovative UI Layout ---
 with gr.Blocks(title="The Poetry Workshop", theme=gr.themes.Soft()) as demo:
     
     # Header
-    gr.Markdown("# 🖋️ The AI Poetry Workshop")
-    gr.Markdown("Don't suffer from writer's block. Configure your muse below.")
+    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():