updated code for Gradio 6.9.0

app.py

@@ -22,7 +22,7 @@ def make_system_message(system_message, demographics, occupation, psychographics
 
 def stream_chat(
     message,
-    history,                # list[list[str, str]] from gr.Chatbot
+    history,                # In Gradio 6, this is a list of dicts: [{"role": "user", "content": "..."}]
     system_message,
     demographics,
     occupation,
@@ -34,43 +34,49 @@ def stream_chat(
     top_p,
 ):
     """
-    Streaming generator that yields the progressively updated chat history.
+    Streaming generator that yields the progressively updated chat history using Gradio 6's message format.
     """
     # 1) Build system + conversation messages
     sys_msg = make_system_message(system_message, demographics, occupation, psychographics, buying_habits, critical_mode)
 
     messages = [{"role": "system", "content": sys_msg}]
-    for user_msg, assistant_msg in (history or []):
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
-        if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
+    
+    # Append past history (Gradio 6 format is already compatible with OpenAI's format!)
+    for msg in (history or []):
+        messages.append({"role": msg["role"], "content": msg["content"]})
+        
+    # Append current user message
     messages.append({"role": "user", "content": message})
 
-    # 2) Start streaming back to the Chatbot
-    running_reply = ""
+    # 2) Setup the history to send back to the Gradio UI
+    running_history = (history or []).copy()
+    running_history.append({"role": "user", "content": message})
+    running_history.append({"role": "assistant", "content": ""})
+    
     # Optimistically show the assistant "typing"
-    running_history = (history or []) + [[message, ""]]
     yield running_history
 
     try:
         response = client.chat.completions.create(
-            model="gpt-4o-mini",   # adjust if needed
+            model="gpt-4o-mini",
             messages=messages,
             max_tokens=int(max_tokens),
             temperature=float(temp),
             top_p=float(top_p),
             stream=True,
         )
+        
+        running_reply = ""
         for chunk in response:
             if chunk.choices and chunk.choices[0].delta and chunk.choices[0].delta.content:
                 token = chunk.choices[0].delta.content
                 running_reply += token
-                running_history[-1][1] = running_reply
+                # Update the content of the last message in the history
+                running_history[-1]["content"] = running_reply
                 # Yield the whole history each time so the UI updates
                 yield running_history
     except Exception as e:
-        running_history[-1][1] = f"❌ An error occurred: {str(e)}"
+        running_history[-1]["content"] = f"❌ An error occurred: {str(e)}"
         yield running_history
 
 
@@ -82,7 +88,8 @@ with gr.Blocks(title="Virtual Consumer Persona – Live Focus Group!") as demo:
     *Powered by OpenAI GPT-4o-mini. Developed by wn.*
     """)
 
-    chatbot = gr.Chatbot(height=450)
+    # Explicitly set type="messages" for Gradio 6 compatibility
+    chatbot = gr.Chatbot(height=450, type="messages")
 
     with gr.Column():
         instructions = gr.Textbox(