Update app.py

app.py

@@ -97,7 +97,7 @@ def analyze_query(query):
             return int(number)
 
     # Поиск цены
-    price_keywords = ["руб", "р.", "рублей", "рубля"]
+    """price_keywords = ["руб", "р.", "рублей", "рубля"]
     price_match = re.findall(rf"(до|\s*)(\d+[\s]*[.,]?\d*)\s*(млн|тр|т\.р|тыс|тысруб|тыс\.руб|тр|т\.р.|тыр|тыр.|тыщ|тыш|тысяч|млнруб|млн\.руб|мн|М|миллионов|милионов|лямов|лимонов)?\s*({'|'.join(price_keywords)})\b", query)
     if price_match:
         operator = price_match[0][0]
@@ -128,10 +128,44 @@ def analyze_query(query):
             filters["area_max"] = int(area_max_match[0][1])
 
     return filters
+    """
+
+    # Поиск цены. из стримлит
+    price_keywords = ["руб", "р.", "рублей", "рубля"]
+    price_match = re.findall(rf"(до|<|>\s*)(\d+[\s]*[.,]?\d*)\s*(млн|тр|т\.р|тыс|тысруб|тыс\.руб|тр|т\.р.|тыр|тыр.|тыщ|тыш|тысяч|млнруб|млн\.руб|мн|М|миллионов|милионов|лямов|лимонов)?\s*({'|'.join(price_keywords)})\b", query)
+    if price_match:
+        operator = price_match[0][0]
+        raw_price = price_match[0][1] + (price_match[0][2] or "")
+        normalized_price = normalize_number(raw_price)
+        if normalized_price is not None:
+            if operator in ["до", "<"]:
+                filters["price_max"] = normalized_price
+            elif operator in ["от", ">"]:
+                filters["price_min"] = normalized_price
+    # Поиск площади
+    area_keywords = ["квм", "кв.м", "кв метров", "кв м", "кв.м.", "квадратов", "квадрата", "м2", "метров", "м²"]
+    # Сначала ищем диапазоны
+    area_range_match_1 = re.findall(rf"(\d+)\s*-\s*(\d+)\s*({'|'.join(area_keywords)})", query)
+    area_range_match_2 = re.findall(rf"(\d+)\s*[.\s]+\s*(\d+)\s*({'|'.join(area_keywords)})", query)
+    if area_range_match_1:
+        filters["area_min"] = int(area_range_match_1[0][0])
+        filters["area_max"] = int(area_range_match_1[0][1])
+    elif area_range_match_2:
+        filters["area_min"] = int(area_range_match_2[0][0])
+        filters["area_max"] = int(area_range_match_2[0][1])
+    else:
+        # Если не найдены диапазоны, ищем "от" или "до"
+        area_min_match = re.findall(rf"\b(от)\s*(\d+)\s*({'|'.join(area_keywords)})", query)
+        area_max_match = re.findall(rf"\b(до)\s*(\d+)\s*({'|'.join(area_keywords)})", query)
+        if area_min_match:
+            filters["area_min"] = int(area_min_match[0][1])  # Преобразуем число в целый тип
+        if area_max_match:
+            filters["area_max"] = int(area_max_match[0][1])  # Преобразуем число в целый тип
+    return filters
 
 
 # Поиск объектов недвижимости в CSV
-def search_real_estate(dataframe, filters):
+"""def search_real_estate(dataframe, filters):
     if not filters:
         return []
 
@@ -163,6 +197,61 @@ def search_real_estate(dataframe, filters):
 
     # Возвращаем все подходящие объекты
     return dataframe[mask].to_dict(orient="records")
+"""
+# Поиск объектов недвижимости в CSV
+def search_real_estate(dataframe, filters):
+    if not filters:
+        return []
+    mask = pd.Series([True] * len(dataframe))
+    # Фильтрация по типу недвижимости
+    if "type" in filters:
+        mask &= dataframe["type"].str.lower().isin(filters["type"])
+        
+    # Фильтрация по виду сделки (Аренда/Продажа)
+    if "rent" in filters:
+        mask &= dataframe["RENT"].str.lower() == filters["rent"].lower()
+  
+    # Фильтрация по цене
+    if "price_min" in filters and "price_max" in filters:
+        mask &= (dataframe["price"] >= filters["price_min"]) & (dataframe["price"] <= filters["price_max"])
+    elif "price_min" in filters:
+        mask &= dataframe["price"] >= filters["price_min"]
+    elif "price_max" in filters:
+        mask &= dataframe["price"] <= filters["price_max"]
+    # Фильтрация по площади
+    if "area_min" in filters and "area_max" in filters:
+        mask &= (dataframe["area"] >= filters["area_min"]) & (dataframe["area"] <= filters["area_max"])
+    elif "area_min" in filters:
+        mask &= dataframe["area"] >= filters["area_min"]
+    elif "area_max" in filters:
+        mask &= dataframe["area"] <= filters["area_max"]
+    results = dataframe[mask]
+    # Проверяем, есть ли результаты
+    if not results.empty:
+        return results.to_dict(orient="records")  # Возвращаем список словарей
+    # Fallback: Если нет подходящих объектов, предлагаем все объекты данного типа
+    if "type" in filters:
+        fallback_mask = dataframe["type"].str.lower().isin(filters["type"])
+        fallback_results = dataframe[fallback_mask]
+        if not fallback_results.empty:
+            return {"fallback": True, "results": fallback_results.to_dict(orient="records")}  # Возвращаем fallback-результаты
+    return []  # Возвращаем пустой список, если ничего не найдено
+
+# Создание контекста для модели
+def create_context(dataframe, filters):
+    if "type" in filters and "rent" in filters:
+        # Создаем контекст только для релевантных данных
+        relevant_data = dataframe[
+            (dataframe["type"].str.lower().isin(filters["type"])) &
+            (dataframe["RENT"].str.lower() == filters["rent"].lower())
+        ]
+    else:
+        # Если параметры не указаны, используем весь датасет
+        relevant_data = dataframe
+    context = " ".join(relevant_data["description"].dropna().tolist()) + " " + \
+              " ".join(relevant_data["type"].dropna().tolist()) + " " + \
+              " ".join(relevant_data["address"].dropna().tolist())
+    return context
 
 
 # Функция для получения ответа от модели
@@ -200,13 +289,15 @@ def predict(message, history=None):
         for result in filtered_results:
             response += (
                 f"ID: {result.get('item_id', 'Не указано')}\n"
+                f"Тип: {result.get('type', 'Не указано')}\n"
                 f"Адрес: {result.get('address', 'Не указано')}\n"
                 f"Цена: {result.get('price', 'Не указано')} руб.\n"
                 f"Площадь: {result.get('area', 'Не указано')} м²\n"
                 f"Метро: {result.get('metro', 'Не указано')}\n"
                 f"До метро: {result.get('met_range', 'Не указано')}\n"
                 f"Описание: {result.get('description', 'Не указано')}\n"
-                f"Тип: {result.get('type', 'Не указано')}\n"
+                f"Фото: {result.get('foto', 'Не указано')}\n"
+                f"Записаться на просмотр: {result.get('order', 'Не указано')}\n"
                 f"---\n"
             )
         log_dialog(message, response)