Update app.py

app.py

@@ -19,6 +19,7 @@ def get_apartment_data(query):
     rows = worksheet.get_all_records()
     results = []
     for row in rows:
+        # Basic check if the query is found in "district" or "developer"
         if query.lower() in row["district"].lower() or query.lower() in row["developer"].lower():
             results.append(row)
     if results:
@@ -31,36 +32,48 @@ def get_apartment_data(query):
 
 # Function to handle queries
 def respond(message, history):
+    """
+    message (str): latest user message
+    history (list): the entire conversation so far, as a list of dicts like:
+        [
+          {"role": "user", "content": "..."},
+          {"role": "assistant", "content": "..."},
+          ...
+        ]
+    Returns a single string (the assistant's answer).
+    """
+
     print("Query received:", message)
     print("Chat history:", history)
 
-    if not history:
-        history = []
-
     try:
-        # If the query is related to districts or developers
+        # If the query is about district or developer, fetch from Sheets
         if "district" in message.lower() or "developer" in message.lower():
             response = get_apartment_data(message)
         else:
-            # Process the query through the model
-            messages = [{"role": "user", "content": message}]
+            # Build messages from history for context
+            conversation = []
+            for turn in history:
+                conversation.append({"role": turn["role"], "content": turn["content"]})
+
+            # Add the new user message
+            conversation.append({"role": "user", "content": message})
+
+            # Query Zephyr
             generated_response = client.chat_completion(
-                messages=messages,
+                messages=conversation,
                 max_tokens=512,
                 temperature=0.7,
                 top_p=0.95,
             )
             response = generated_response["choices"][0]["message"]["content"]
 
-        # Append to history in the required format
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": response})
-        return response, history
+        # Just return the assistant's text
+        return response
 
     except Exception as e:
         print(f"Error processing query: {e}")
-        history.append({"role": "assistant", "content": "An error occurred while processing your query."})
-        return "An error occurred while processing your query.", history
+        return "An error occurred while processing your query."
 
 # Gradio Interface
 chat_interface = gr.ChatInterface(