Update app.py

app.py

@@ -63,7 +63,8 @@ def chatbot(user_input, history, model_choice):
     split_res = response.split("<|assistant|>")
     final_response = split_res[-1].strip() if len(split_res) > 1 else response
 
-    history.append((user_input, final_response))
+    history.append({"role": "user", "content": user_input})
+    history.append({"role": "assistant", "content": final_response})
     return "", history
 
 with gr.Blocks() as demo:
@@ -72,7 +73,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         model_choice = gr.Dropdown(label="Select Model", choices=premium_models, value=premium_models[0])
     
-    chatbot_ui = gr.Chatbot()
+    chatbot_ui = gr.Chatbot(type="messages")
     user_input = gr.Textbox(show_label=False, placeholder="Type your message and press Enter")
     clear_btn = gr.Button("Clear")