Update app_low.py

app_low.py

@@ -52,16 +52,26 @@ def enhance_prompt(user_prompt, temperature, max_tokens, chat_history):
     return chat_history
 
 
+# ===================== Prompt Enhancer Function =====================
 def enhance_prompt1(user_prompt, temperature, max_tokens, chat_history):
+    # Build the messages for the model
     messages = [
         {"role": "system", "content": "Enhance and expand the following prompt with more details and context:"},
         {"role": "user", "content": user_prompt}
     ]
+    
+    # Generate model prompt
     prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
     output = pipe(prompt, max_new_tokens=256)
-    print(output[0]['generated_text'])
+    enhanced_text = output[0]['generated_text']
+    print(enhanced_text)
 
-    return output[0]['generated_text']
+    # Wrap in a chat message format
+    chat_history = chat_history or []
+    chat_history.append({"role": "user", "content": user_prompt})
+    chat_history.append({"role": "assistant", "content": enhanced_text})
+
+    return chat_history
 
 # ============================================================
 # 3️⃣ Gradio UI