Update app.py

app.py

@@ -3,7 +3,7 @@ from huggingface_hub import InferenceClient
 import gspread
 from google.oauth2.service_account import Credentials
 
-# Подключение к Google Sheets
+# 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)
@@ -11,10 +11,9 @@ client_gs = gspread.authorize(credentials)
 spreadsheet = client_gs.open_by_key("1LHgb4eYDM2ImFY3e3h4kPUnDh6oRHpVz2PA29lQ3Xxg")
 worksheet = spreadsheet.sheet1
 
-# Подключение к модели zephyr-7b-beta через Inference API
+# Hugging Face Model Setup
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# Функция для получения данных из Google Sheets
 def get_apartment_data(query):
     rows = worksheet.get_all_records()
     results = []
@@ -23,26 +22,23 @@ def get_apartment_data(query):
             results.append(row)
     if results:
         return "\n".join(
-            [f"Район: {row['district']}, Застройщик: {row['developer']}, Площадь: {row['apt_area']} м²"
+            [f"District: {row['district']}, Developer: {row['developer']}, Area: {row['apt_area']} sqm"
              for row in results]
         )
     else:
-        return "Извините, подходящих вариантов не найдено."
+        return "Sorry, no matching apartments found."
 
-# Функция для обработки запросов
 def respond(message, history):
-    print("Получен запрос:", message)
-    print("История общения:", history)
+    print("Query received:", message)
+    print("Chat history:", history)
 
     if not history:
         history = []
 
     try:
-        # Если запрос связан с районами или застройщиками
-        if "район" in message.lower() or "застройщик" in message.lower():
+        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)
@@ -54,21 +50,17 @@ def respond(message, history):
             )
             response = generated_response["choices"][0]["message"]["content"]
 
-        # Формирование корректной истории общения
         history.append({"role": "user", "content": message})
         history.append({"role": "assistant", "content": response})
         return response, history
 
     except Exception as e:
-        print(f"Ошибка обработки запроса: {e}")
-        history.append({"role": "assistant", "content": "Произошла ошибка при обработке вашего запроса."})
-        return "Произошла ошибка при обработке вашего запроса.", history
+        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
 
-# Интерфейс Gradio
 chat_interface = gr.ChatInterface(
-    respond,
-    input_label="Ваш вопрос",
-    output_label="Ответ",
+    respond
 )
 
 if __name__ == "__main__":