Update app.py

app.py

@@ -36,12 +36,19 @@ def respond(message, 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": "user", "content": message}]
-            for h in history:
-                messages.append(h)
+            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})
+            messages.append({"role": "user", "content": message})
+
+            # Generate response from model
             generated_response = client.chat_completion(
                 messages=messages,
                 max_tokens=512,
@@ -50,17 +57,22 @@ def respond(message, history):
             )
             response = generated_response["choices"][0]["message"]["content"]
 
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": response})
+        # Append the new interaction to the history
+        history.append((message, response))
         return response, history
 
     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
 
+# Set up Gradio Chat Interface
 chat_interface = gr.ChatInterface(
-    respond
+    fn=respond,
+    examples=[
+        "What apartments are available in Bostandyk?",
+        "Tell me about BI Group.",
+        "What options do you have for developers?",
+    ],
 )
 
 if __name__ == "__main__":