Create app.py

app.py

@@ -0,0 +1,200 @@
+# ==============================================================================
+#           ШАГ 1: ИМПОРТ И НАСТРОЙКА
+# ==============================================================================
+print("🚀 Шаг 1: Импорт всех библиотек...")
+
+import osmnx as ox
+import networkx as nx
+import geopandas as gpd
+import pandas as pd
+import numpy as np
+from catboost import CatBoostRegressor
+import datetime
+import folium
+import gradio as gr
+from geopy.geocoders import Nominatim
+from geopy.extra.rate_limiter import RateLimiter
+
+print("✅ Импорт завершен.")
+
+# ==============================================================================
+#           ШАГ 2: ЗАГРУЗКА ДАННЫХ ГОРОДА (ГРАФ, СВЕТОФОРЫ, POI)
+#           Этот блок выполняется один раз при запуске приложения.
+# ==============================================================================
+print("\n🌍 Шаг 2: Подготовка данных о городе Бишкек...")
+
+place_name = "Bishkek, Kyrgyzstan"
+
+# --- Загрузка графа дорог ---
+print("   - Загружаем дорожный граф... (это может занять несколько минут)")
+G = ox.graph_from_place(place_name, network_type='drive', simplify=True)
+G = ox.add_edge_speeds(G)
+G = ox.add_edge_travel_times(G)
+print("   - ✅ Граф дорог загружен.")
+
+# --- Преобразование в таблицы и подготовка фичей ---
+print("   - Готовим фичи для ребер графа...")
+gdf_nodes, gdf_edges_raw = ox.graph_to_gdfs(G)
+gdf_edges = gdf_edges_raw.copy()
+
+# Очистка данных (взято из твоего ноутбука)
+gdf_edges['maxspeed'] = gdf_edges['maxspeed'].apply(lambda x: x[0] if isinstance(x, list) else x).astype(str).str.extract('(\d+)').astype(float).fillna(40)
+gdf_edges['lanes'] = gdf_edges['lanes'].apply(lambda x: x[0] if isinstance(x, list) else x)
+gdf_edges['lanes'] = pd.to_numeric(gdf_edges['lanes'], errors='coerce').fillna(1)
+gdf_edges['highway'] = gdf_edges['highway'].apply(lambda x: x[0] if isinstance(x, list) else x).fillna(gdf_edges['highway'].mode()[0])
+
+# --- Загрузка и привязка светофоров ---
+print("   - Загружаем данные о светофорах...")
+tags_signals = {"highway": "traffic_signals"}
+gdf_signals = ox.features_from_place(place_name, tags_signals)
+signal_nodes = ox.nearest_nodes(G, X=gdf_signals.geometry.x, Y=gdf_signals.geometry.y)
+signal_nodes_set = set(signal_nodes)
+print(f"   - ✅ Найдено и привязано {len(signal_nodes_set)} перекрестков со светофорами.")
+
+
+# --- Загрузка и подсчет POI (точек интереса) ---
+print("   - Загружаем POI и обогащаем граф... (это тоже может занять время)")
+tags_poi = {
+    "amenity": ["restaurant", "cafe", "school", "university", "bank", "atm", "pharmacy", "hospital"],
+    "shop": ["supermarket", "mall"], "office": True, "public_transport": ["station", "platform"]
+}
+gdf_poi = ox.features_from_place(place_name, tags_poi)
+
+# Расчет POI
+utm_crs = gdf_edges.estimate_utm_crs()
+projected_gdf_edges = gdf_edges.to_crs(utm_crs)
+projected_gdf_poi = gdf_poi.to_crs(utm_crs)
+buffer_gdf = gpd.GeoDataFrame(geometry=projected_gdf_edges.geometry.centroid.buffer(50), index=projected_gdf_edges.index)
+joined_gdf = gpd.sjoin(buffer_gdf, projected_gdf_poi, how="inner", predicate="intersects")
+poi_counts = joined_gdf.index.value_counts()
+gdf_edges['poi_count'] = poi_counts.reindex(gdf_edges.index, fill_value=0)
+print("   - ✅ Данные обогащены точками интереса (POI).")
+
+
+print("✅ Данные о городе полностью подготовлены!")
+
+
+# ==============================================================================
+#           ШАГ 3: ЗАГРУЗКА ОБУЧЕННОЙ МОДЕЛИ
+# ==============================================================================
+print("\n🧠 Шаг 3: Загрузка обученной модели предсказания трафика...")
+model_filename = "bishkek_traffic_model.cbm"
+model = CatBoostRegressor()
+model.load_model(model_filename)
+print("✅ Модель успешно загружена.")
+
+
+# ==============================================================================
+#           ШАГ 4: ВСПОМОГАТЕЛЬНЫЕ ФУНКЦИИ ДЛЯ НАВИГАТОРА
+# ==============================================================================
+print("\n🧩 Шаг 4: Сборка функций-компонентов навигатора...")
+
+def predict_graph_weights(graph_edges_df, model, timestamp):
+    df = graph_edges_df.copy()
+    df['hour'] = timestamp.hour
+    df['minute'] = timestamp.minute
+    df['day_of_week'] = timestamp.weekday()
+    df.rename(columns={'highway': 'highway_cat'}, inplace=True)
+    features_for_pred = ['lanes', 'maxspeed', 'length', 'poi_count', 'highway_cat', 'hour', 'minute', 'day_of_week']
+    predictions = model.predict(df[features_for_pred])
+    return predictions
+
+def find_two_routes(G, start_node, end_node, weight='travel_time'):
+    try:
+        route1 = nx.shortest_path(G, source=start_node, target=end_node, weight=weight)
+        time1 = nx.shortest_path_length(G, source=start_node, target=end_node, weight=weight)
+    except nx.NetworkXNoPath: return None, None, None, None
+    G_penalized = G.copy()
+    for u, v in zip(route1[:-1], route1[1:]):
+        if G_penalized.has_edge(u, v): G_penalized[u][v][0][weight] *= 2
+    try:
+        route2 = nx.shortest_path(G_penalized, source=start_node, target=end_node, weight=weight)
+        time2 = sum(G[u][v][0][weight] for u, v in zip(route2[:-1], route2[1:]))
+    except nx.NetworkXNoPath: route2, time2 = None, None
+    return route1, time1, route2, time2
+
+def plot_routes_on_folium_map(G, route1, route2, start_point, end_point):
+    map_center = [(start_point[0] + end_point[0]) / 2, (start_point[1] + end_point[1]) / 2]
+    m = folium.Map(location=map_center, zoom_start=13, tiles="OpenStreetMap")
+    if route1:
+        points1 = [(G.nodes[node]['y'], G.nodes[node]['x']) for node in route1]
+        folium.PolyLine(points1, color="white", weight=11, opacity=0.7).add_to(m)
+        folium.PolyLine(points1, color="#2ca02c", weight=6, opacity=0.85, tooltip="Маршрут 1").add_to(m)
+    if route2 and route1 != route2:
+        points2 = [(G.nodes[node]['y'], G.nodes[node]['x']) for node in route2]
+        folium.PolyLine(points2, color="white", weight=11, opacity=0.7).add_to(m)
+        folium.PolyLine(points2, color="#007FFF", weight=6, opacity=0.85, tooltip="Маршрут 2").add_to(m)
+    folium.Marker(location=start_point, popup="Точка А (Старт)", icon=folium.Icon(color="green")).add_to(m)
+    folium.Marker(location=end_point, popup="Точка Б (Финиш)", icon=folium.Icon(color="red")).add_to(m)
+    return m
+
+print("✅ Вспомогательные функции готовы.")
+
+# ==============================================================================
+#           ШАГ 5: ГЛАВНАЯ ФУНКЦИЯ И ИНТЕРФЕЙС GRADIO
+# ==============================================================================
+print("\n🚀 Шаг 5: Настройка и запуск веб-интерфейса Gradio...")
+
+geolocator = Nominatim(user_agent="bishkek_navigator_app")
+geocode = RateLimiter(geolocator.geocode, min_delay_seconds=1)
+
+def find_and_plot_routes_by_address(start_address, end_address, day_of_week, hour, minute):
+    if not start_address or not end_address: return None, "Ошибка: Пожалуйста, введите оба адреса."
+    if hour is None or minute is None: return None, "Ошибка: Пожалуйста, установите значения для часа и минуты."
+    try:
+        location_a = geocode(f"{start_address}, Бишкек"); location_b = geocode(f"{end_address}, Бишкек")
+        if not location_a or not location_b: return None, "Не удалось найти один или оба адреса. Попробуйте уточнить."
+        
+        start_point = (location_a.latitude, location_a.longitude); end_point = (location_b.latitude, location_b.longitude)
+        start_node = ox.nearest_nodes(G, Y=start_point[0], X=start_point[1]); end_node = ox.nearest_nodes(G, Y=end_point[0], X=end_point[1])
+        
+        selected_time = datetime.datetime(2023, 1, 2 + day_of_week, hour, minute)
+        travel_times = predict_graph_weights(gdf_edges, model, selected_time)
+        nx.set_edge_attributes(G, values=pd.Series(travel_times, index=gdf_edges.index).to_dict(), name='travel_time')
+        
+        r1, t1, r2, t2 = find_two_routes(G, start_node, end_node, weight='travel_time')
+        if r1 is None: return None, "Не удалось построить маршрут."
+        
+        SIGNAL_DELAY_SECONDS = 30
+        signals_on_route1 = sum(1 for node in r1 if node in signal_nodes_set); t1_total = t1 + signals_on_route1 * SIGNAL_DELAY_SECONDS
+        distance1_km = sum(G[u][v][0]['length'] for u, v in zip(r1[:-1], r1[1:])) / 1000
+        
+        t2_total, distance2_km = None, None
+        if r2 and r1 != r2:
+            signals_on_route2 = sum(1 for node in r2 if node in signal_nodes_set); t2_total = t2 + signals_on_route2 * SIGNAL_DELAY_SECONDS
+            distance2_km = sum(G[u][v][0]['length'] for u, v in zip(r2[:-1], r2[1:])) / 1000
+        
+        final_map = plot_routes_on_folium_map(G, r1, r2, start_point, end_point)
+        
+        output_text = f"Маршрут 1:\n  - Время: ~{t1_total / 60:.1f} мин.\n  - Расстояние: {distance1_km:.2f} км\n  - Светофоров: {signals_on_route1}\n"
+        if t2_total is not None:
+            output_text += f"\nМаршрут 2:\n  - Время: ~{t2_total / 60:.1f} мин.\n  - Расстояние: {distance2_km:.2f} км\n  - Светофоров: {signals_on_route2}"
+        else: output_text += "\nАльтернативный маршрут не найден."
+        
+        return final_map._repr_html_(), output_text
+        
+    except Exception as e:
+        return None, f"Произошла внутренняя ошибка: {e}"
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🗺️ Умный навигатор по Бишкеку")
+    with gr.Row():
+        with gr.Column(scale=1):
+            start_address_input = gr.Textbox(label="Откуда?", placeholder="Например, 7-й микрорайон")
+            end_address_input = gr.Textbox(label="Куда?", placeholder="Например, Ошский базар")
+            day_dropdown = gr.Dropdown(label="День недели", choices=["Понедельник", "Вторник", "Среда", "Четверг", "Пятница", "Суббота", "Воскресенье"], value="Понедельник", type="index")
+            hour_slider = gr.Slider(label="Час", minimum=0, maximum=23, step=1, value=8)
+            minute_slider = gr.Slider(label="Минута", minimum=0, maximum=59, step=1, value=30)
+            build_btn = gr.Button("🚀 Построить маршруты", variant="primary")
+            output_textbox = gr.Textbox(label="Информация о маршрутах", interactive=False, lines=7)
+        with gr.Column(scale=2):
+            output_map_html = gr.HTML(label="Карта с маршрутами")
+    build_btn.click(
+        fn=find_and_plot_routes_by_address,
+        inputs=[start_address_input, end_address_input, day_dropdown, hour_slider, minute_slider],
+        outputs=[output_map_html, output_textbox]
+    )
+
+demo.launch()
+
+print("\n\n🌐 Веб-интерфейс запущен!")