Update app.py

app.py

@@ -67,19 +67,28 @@ def chat_with_docs(message, history, api_key):
     global chat_engine, conversation_history
     
     if not api_key:
-        return history + [("Please enter your OpenAI API key first.", None)]
+        return history + [{"role": "assistant", "content": "Please enter your OpenAI API key first."}]
     
     if chat_engine is None:
-        return history + [(message, "Please upload and load documents first before asking questions.")]
+        return history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": "Please upload and load documents first before asking questions."}
+        ]
     
     try:
         response = chat_engine.chat(message)
         conversation_history.append({"role": "user", "content": message})
         conversation_history.append({"role": "assistant", "content": response.response})
         
-        return history + [(message, response.response)]
+        return history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": response.response}
+        ]
     except Exception as e:
-        return history + [(message, f"Error: {str(e)}")]
+        return history + [
+            {"role": "user", "content": message},
+            {"role": "assistant", "content": f"Error: {str(e)}"}
+        ]
 
 # Function to save conversation
 def save_conversation():
@@ -171,7 +180,11 @@ with gr.Blocks(title="Chat with Documents 💬 📚") as demo:
     
     with gr.Row():
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(label="Chat", height=400)
+            chatbot = gr.Chatbot(
+                label="Chat",
+                height=400,
+                type="messages"  # Explicitly set type to messages
+            )
             msg = gr.Textbox(
                 label="Your Question",
                 placeholder="Ask a question about your documents..."