chabot

app.py

@@ -12,12 +12,21 @@ def chat_fn(message, history):
     # Extract the generated text from the response
     reply = response[0]['generated_text']
     
-    # Update history
-    history = history + [[message, reply]]
+    # Update history with the new message format
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": reply})
     return history
 
 with gr.Blocks() as dennis_denuto:
-    chatbot = gr.Chatbot(placeholder="<strong>It's the vibe.</strong><br>Ask anything about criminal law and procedure in Australia.")
-    gr.ChatInterface(fn=chat_fn, type="messages", chatbot=chatbot, save_history=True)
+    chatbot = gr.Chatbot(
+        placeholder="<strong>It's the vibe.</strong><br>Ask anything about criminal law and procedure in Australia.",
+        type="messages"
+    )
+    gr.ChatInterface(
+        fn=chat_fn, 
+        type="messages", 
+        chatbot=chatbot, 
+        save_history=True
+    )
     
 dennis_denuto.launch()