Update app_low.py

app_low.py

@@ -25,22 +25,21 @@ pipe = pipeline(
 # 2️⃣ Define the generation function
 # ============================================================
 def enhance_prompt(user_prompt, temperature, max_tokens, chat_history):
-    """Enhance user prompt and maintain chat history."""
-    if not user_prompt.strip():
-        return chat_history + [["", "⚠️ Please enter a prompt."]]
+    chat_history = chat_history or []
 
     full_prompt = f"Enhance and expand the following prompt with more details and context: {user_prompt}"
 
-    # Generate output
     output = pipe(
         full_prompt,
         max_new_tokens=int(max_tokens),
         temperature=float(temperature),
         do_sample=True,
-    )
+    )[0]['generated_text'].strip()
+
+    # Convert to Gradio messages format
+    chat_history.append({"role": "user", "content": user_prompt})
+    chat_history.append({"role": "assistant", "content": output})
 
-    result = output[0]['generated_text'].strip()
-    chat_history = chat_history + [[user_prompt, result]]
     return chat_history
 
 # ============================================================