Update app.py

app.py

@@ -14,7 +14,12 @@ if device == "cuda":
 else:
     text_model_name = "google/flan-t5-base"  # CPU-friendly
 
-chat_model = pipeline("text-generation", model=text_model_name, device=0 if device=="cuda" else -1)
+chat_model = pipeline(
+    "text-generation",
+    model=text_model_name,
+    device=0 if device=="cuda" else -1,
+    return_full_text=False
+)
 
 # ----------------------------
 # Try to load Stable Diffusion (only if GPU)
@@ -53,7 +58,7 @@ def query_llm(prompt, history, persona):
     input_text += f"User: {prompt}\nAssistant:"
 
     out = chat_model(input_text, max_new_tokens=300, do_sample=True, temperature=0.7)
-    return out[0]["generated_text"].split("Assistant:")[-1].strip()
+    return out[0]["generated_text"].strip()
 
 
 def multimodal_chat(user_msg, history, persona):
@@ -94,7 +99,15 @@ def multimodal_chat(user_msg, history, persona):
 
 
 with gr.Blocks(css="style.css") as demo:
-    gr.Markdown("🧠 **ZEN Research Lab (Adaptive Edition)** — works everywhere, GPU unlocks extra powers", elem_id="zen-header")
+    gr.Markdown("🧠 **ZEN Research Lab (Adaptive Edition)**", elem_id="zen-header")
+
+    # Capabilities banner
+    cap_text = "✅ Text  ✅ Charts  ✅ Simulation"
+    if sd_model is not None:
+        cap_text += "  ✅ Images"
+    else:
+        cap_text += "  ❌ Images (GPU required)"
+    gr.Markdown(cap_text)
 
     persona = gr.Dropdown(["Default","Analyst","Artist","Futurist","Philosopher"], label="Mode", value="Default")
     chatbot = gr.Chatbot(label="Conversation", height=400)
@@ -119,5 +132,18 @@ with gr.Blocks(css="style.css") as demo:
     user_msg.submit(respond, inputs=[user_msg, chatbot, persona],
                     outputs=[chatbot, img_out, chart_out])
 
+    # Example starter buttons
+    with gr.Accordion("✨ Try these examples"):
+        with gr.Row():
+            gr.Examples(
+                examples=[
+                    ["Draw a futuristic city skyline at night"],
+                    ["Simulate first contact with an alien civilization"],
+                    ["Make a chart of AI adoption from 2010 to 2030"],
+                    ["Explain quantum entanglement in simple terms"],
+                ],
+                inputs=[user_msg]
+            )
+
 if __name__ == "__main__":
     demo.queue(max_size=50).launch()