Update app.py

app.py

@@ -32,20 +32,21 @@ def respond(message, history):
     print("Query received:", message)
     print("Chat history:", history)
 
+    # Initialize history as a list of dictionaries if empty
     if not history:
         history = []
 
     try:
-        # Determine if the query should access Google Sheets
         if "district" in message.lower() or "developer" in message.lower():
             response = get_apartment_data(message)
         else:
             messages = [{"role": "system", "content": "You are a helpful assistant."}]
-            for user_message, assistant_message in history:
-                if user_message:
-                    messages.append({"role": "user", "content": user_message})
-                if assistant_message:
-                    messages.append({"role": "assistant", "content": assistant_message})
+            for entry in history:
+                if entry["role"] == "user":
+                    messages.append({"role": "user", "content": entry["content"]})
+                elif entry["role"] == "assistant":
+                    messages.append({"role": "assistant", "content": entry["content"]})
+
             messages.append({"role": "user", "content": message})
 
             # Generate response from model
@@ -58,16 +59,18 @@ def respond(message, history):
             response = generated_response["choices"][0]["message"]["content"]
 
         # Append the new interaction to the history
-        history.append((message, response))
-        return response, history
+        history.append({"role": "user", "content": message})
+        history.append({"role": "assistant", "content": response})
+        return history
 
     except Exception as e:
         print(f"Error processing query: {e}")
-        return "An error occurred while processing your query.", history
+        return [{"role": "assistant", "content": "An error occurred while processing your query."}]
 
 # Set up Gradio Chat Interface
 chat_interface = gr.ChatInterface(
     fn=respond,
+    chatbot="Your Assistant",
     examples=[
         "What apartments are available in Bostandyk?",
         "Tell me about BI Group.",