Update app.py

app.py

@@ -1,7 +1,7 @@
 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
+import time
 
 # --- Configuration ---
 MODEL_REPO = "Kezovic/iris-q4gguf-lora-test"
@@ -30,14 +30,14 @@ def load_llm():
 load_llm()
 
 # --- The "A-Grade" Logic: Hardcoded Prompt Engineering ---
-# UPDATE: Added 'progress' argument to the function signature
+# UPDATE: Added 'progress=gr.Progress()' here to enable the loading bar
 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...")
+    # 1. VISUAL FEEDBACK: Immediately show the bar
+    progress(0, desc="Consulting the Muse...")
     
     if not llm:
         return "Error: Model not loaded."
@@ -45,10 +45,11 @@ def generate_poem(format_type, persona, topic, progress=gr.Progress()):
     if not topic:
         return "Please enter a topic!"
 
-    # 2. Update Progress: Constructing Prompt
+    # 2. VISUAL FEEDBACK: Update status while we build the prompt
     progress(0.2, desc=f"Summoning {persona}...")
-    time.sleep(0.5) # Artificial pause so user sees the message (optional)
+    time.sleep(0.3) # Small pause so the user actually sees the message
 
+    # 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.",
@@ -56,6 +57,7 @@ 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.",
@@ -75,8 +77,8 @@ def generate_poem(format_type, persona, topic, progress=gr.Progress()):
         f"### Response:\n"
     )
 
-    # 3. Update Progress: Generation Started
-    progress(0.5, desc="Drafting Masterpiece (This may take a moment)...")
+    # 3. VISUAL FEEDBACK: Indicate the heavy lifting is starting
+    progress(0.4, desc="Writing your masterpiece (this may take a moment)...")
 
     # 4. Generate
     output = llm(
@@ -87,9 +89,9 @@ def generate_poem(format_type, persona, topic, progress=gr.Progress()):
         echo=False
     )
     
-    # 5. Update Progress: Done
-    progress(0.9, desc="Polishing rhymes...")
-    time.sleep(0.5)
+    # 4. VISUAL FEEDBACK: Done
+    progress(0.9, desc="Final polish...")
+    time.sleep(0.2)
     
     return output['choices'][0]['text'].strip()