Update app.py

app.py

@@ -1,81 +1,37 @@
-import gradio as gr
-from huggingface_hub import InferenceClient
-import gspread
-from google.oauth2.service_account import Credentials
-
-# Google Sheets Setup
-SERVICE_ACCOUNT_FILE = "sheet-integration-agent@bold-physics-445007-g4.iam.gserviceaccount.com.json"
-scopes = ["https://www.googleapis.com/auth/spreadsheets"]
-credentials = Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=scopes)
-client_gs = gspread.authorize(credentials)
-spreadsheet = client_gs.open_by_key("1LHgb4eYDM2ImFY3e3h4kPUnDh6oRHpVz2PA29lQ3Xxg")
-worksheet = spreadsheet.sheet1
-
-# Hugging Face Model Setup
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-
-def get_apartment_data(query):
-    rows = worksheet.get_all_records()
-    results = []
-    for row in rows:
-        if query.lower() in row["district"].lower() or query.lower() in row["developer"].lower():
-            results.append(row)
-    if results:
-        return "\n".join(
-            [f"District: {row['district']}, Developer: {row['developer']}, Area: {row['apt_area']} sqm"
-             for row in results]
-        )
-    else:
-        return "Sorry, no matching apartments found."
-
 def respond(message, history):
     print("Query received:", message)
     print("Chat history:", history)
 
-    # Initialize history as a list of dictionaries if empty
+    # Ensure history is a list of dictionaries
     if not history:
         history = []
 
     try:
-        if "district" in message.lower() or "developer" in message.lower():
+        # Check for specific keywords to invoke Google Sheets query
+        if "bostandyk" in message.lower() or "developer" in message.lower():
             response = get_apartment_data(message)
         else:
+            # Process the query using the model
             messages = [{"role": "system", "content": "You are a helpful assistant."}]
             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": entry["role"], "content": entry["content"]})
 
             messages.append({"role": "user", "content": message})
 
             # Generate response from model
-            generated_response = client.chat_completion(
+            model_response = client.chat_completion(
                 messages=messages,
                 max_tokens=512,
                 temperature=0.7,
                 top_p=0.95,
             )
-            response = generated_response["choices"][0]["message"]["content"]
+            response = model_response["choices"][0]["message"]["content"]
 
-        # Append the new interaction to the history
+        # Append the interaction to the 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 [{"role": "assistant", "content": "An error occurred while processing your query."}]
-
-# Set up Gradio Chat Interface
-chat_interface = gr.ChatInterface(
-    fn=respond,
-    examples=[
-        "What apartments are available in Bostandyk?",
-        "Tell me about BI Group.",
-        "What options do you have for developers?",
-    ],
-)
-
-if __name__ == "__main__":
-    chat_interface.launch()
+        return [{"role": "assistant", "content": "An error occurred while processing your query."}]