Update app.py

app.py

@@ -17,6 +17,7 @@ client = InferenceClient(MODEL_REPO)
 
 # ---------------- CHAT FUNCTION ----------------
 def stream_response(message, chat_history, system_message, max_tokens, temperature, top_p, response_style):
+    # adjust style
     if response_style == "Concise":
         system_message += " Keep answers short and direct."
     elif response_style == "Detailed":
@@ -24,10 +25,8 @@ def stream_response(message, chat_history, system_message, max_tokens, temperatu
     elif response_style == "Essay":
         system_message += " Write long, structured, essay-style responses."
 
-    messages = [{"role": "system", "content": system_message}]
-    for user, bot in chat_history:
-        messages.append({"role": "user", "content": user})
-        messages.append({"role": "assistant", "content": bot})
+    # build conversation
+    messages = [{"role": "system", "content": system_message}] + chat_history
     messages.append({"role": "user", "content": message})
 
     response = ""
@@ -40,7 +39,11 @@ def stream_response(message, chat_history, system_message, max_tokens, temperatu
     ):
         token = msg.choices[0].delta.content or ""
         response += token
-        yield "", chat_history + [(message, response)]
+        # yield new history in messages format
+        yield "", chat_history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": response}
+        ]
 
 
 # ---------------- UI ----------------
@@ -54,7 +57,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet", secondary_hue="pink"))
     )
 
     chatbot = gr.Chatbot(
-        type="messages",  # ✅ new format
+        type="messages",  # ✅ use messages format
         height=500,
         show_copy_button=True,
         label="Chat"
@@ -95,7 +98,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="violet", secondary_hue="pink"))
         [msg, chatbot, system_prompt, max_tokens, temperature, top_p, response_style],
         [msg, chatbot]
     )
-    clear_btn.click(lambda: None, None, chatbot, queue=False)
+    clear_btn.click(lambda: [], None, chatbot, queue=False)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()