Update app_low.py

app_low.py

@@ -42,6 +42,8 @@ def enhance_prompt(user_prompt, temperature, max_tokens, chat_history):
         temperature=float(temperature),
         do_sample=True,
     )[0]["generated_text"].strip()
+    print(output)
+    print(output[0])
 
     # Append conversation to history
     chat_history.append({"role": "user", "content": user_prompt})
@@ -50,6 +52,13 @@ def enhance_prompt(user_prompt, temperature, max_tokens, chat_history):
     return chat_history
 
 
+def enhance_prompt1(user_prompt, temperature, max_tokens, chat_history):
+    prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+    output = pipe(prompt, max_new_tokens=256)
+    print(output[0]['generated_text'])
+
+    return output[0]['generated_text']
+
 # ============================================================
 # 3️⃣ Gradio UI
 # ============================================================
@@ -74,11 +83,13 @@ with gr.Blocks(title="Prompt Enhancer – Gemma 3 270M", theme=gr.themes.Soft())
             max_tokens = gr.Slider(32, 256, value=128, step=16, label="Max Tokens")
             send_btn = gr.Button("🚀 Enhance Prompt", variant="primary")
             clear_btn = gr.Button("🧹 Clear Chat")
+            add_btn = gr.Button("🧹 add Chat")
 
     # Bind UI actions
     send_btn.click(enhance_prompt, [user_prompt, temperature, max_tokens, chatbot], chatbot)
     user_prompt.submit(enhance_prompt, [user_prompt, temperature, max_tokens, chatbot], chatbot)
     clear_btn.click(lambda: [], None, chatbot)
+    add_btn.submit(enhance_prompt1, [user_prompt, temperature, max_tokens, chatbot], chatbot)
 
     gr.Markdown(
         """