ErzhanAb commited on
Commit
013a5f0
·
verified ·
1 Parent(s): ebacc7b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +50 -49
app.py CHANGED
@@ -71,18 +71,30 @@ def find_fastest_route(G, start_node, end_node, weight='travel_time'):
71
  return None, None
72
 
73
  def plot_route_on_map(G, route, start_point, end_point):
74
- """Отрисовывает один маршрут на стильной карте, как в скриншоте."""
75
- m = folium.Map(tiles="CartoDB positron", attr='&copy; <a href="https://www.openstreetmap.org/copyright">OpenStreetMap</a> contributors &copy; <a href="https://carto.com/attributions">CARTO</a>')
 
76
 
77
  if route:
78
  points = [(G.nodes[node]['y'], G.nodes[node]['x']) for node in route]
79
- main_line = folium.PolyLine(points, color="#007FFF", weight=5, opacity=0.9)
 
 
 
 
80
  main_line.add_to(m)
 
81
  m.fit_bounds(main_line.get_bounds(), padding=(30, 30))
82
 
83
- # Маркеры, как в скриншоте: красный - старт, зеленый - финиш
84
- folium.Marker(location=start_point, popup="<b>Старт</b>", icon=folium.Icon(color="red")).add_to(m)
85
- folium.Marker(location=end_point, popup="<b>Финиш</b>", icon=folium.Icon(color="green")).add_to(m)
 
 
 
 
 
 
86
 
87
  return m
88
 
@@ -93,7 +105,7 @@ print("✅ Вспомогательные функции готовы.")
93
  # ==============================================================================
94
  print("\n🚀 Шаг 5: Настройка и запуск веб-интерфейса Gradio...")
95
 
96
- geolocator = Nominatim(user_agent="bishkek_navigator_app_v4")
97
  geocode = RateLimiter(geolocator.geocode, min_delay_seconds=1)
98
 
99
  def find_and_plot_route_by_address(start_address, end_address, day_of_week, hour, minute):
@@ -128,73 +140,62 @@ def find_and_plot_route_by_address(start_address, end_address, day_of_week, hour
128
 
129
  final_map = plot_route_on_map(G, route, start_point, end_point)
130
 
131
- # Форматирование вывода в Markdown, как на скриншоте
132
  output_md = f"""
133
- **✅ Маршрут построен!**
134
-
135
- * Время в пути: ~{total_time_min:.1f} мин.
136
- * Расстояние: {distance_km:.2f} км
137
- * Светофоров на пути: {signals_on_route}
138
  """
139
- return final_map._repr_html_(), output_md.replace(" *", "*") # Убираем лишние отступы для Markdown
140
 
141
  except Exception as e:
142
  return None, f"### 💥 Произошла внутренняя ошибка:\n`{e}`"
143
 
144
- # CSS для стилизации интерфейса под дизайн со скриншота
145
  css = """
146
- body, .gradio-container { background-color: #f0fdfa; font-family: 'Inter', sans-serif; }
147
- .gradio-container { max-width: 1200px !important; margin: auto !important; padding-top: 1.5rem; }
148
- /* Стили для карты */
149
- #map-output { border-radius: 1.5rem; overflow: hidden; box-shadow: 0 10px 15px -3px rgba(0, 0, 0, 0.1), 0 4px 6px -2px rgba(0, 0, 0, 0.05); }
150
- /* Стили для кнопки */
151
  .gr-button-primary {
152
- background-color: #14b8a6; color: white; font-weight: bold; border-radius: 0.75rem; border: none;
153
- box-shadow: 0 4px 6px rgba(20, 184, 166, 0.2); transition: all 0.2s ease; padding: 12px 0;
154
- }
155
- .gr-button-primary:hover { background-color: #0d9488; box-shadow: 0 7px 10px rgba(20, 184, 166, 0.3); transform: translateY(-2px); }
156
- /* Стили для информационного блока с результатом */
157
- #info-box { background-color: white; padding: 1.5rem !important; border-radius: 1rem; box-shadow: 0 4px 6px -1px rgba(0,0,0,0.1), 0 2px 4px -1px rgba(0,0,0,0.06); }
158
- /* Стили для меток (Откуда?, Куда?, и т.д.) */
159
- .gradio-container .form .gr-label span {
160
- background-color: #ccfbf1; color: #134e4a; padding: 5px 12px;
161
- border-radius: 9999px; font-size: 0.8em; font-weight: 600;
162
- display: inline-block; margin-bottom: 6px;
163
- }
164
- /* Стили для полей ввода */
165
- .gradio-container .gr-input-text, .gradio-container .gr-dropdown {
166
- border-radius: 0.75rem !important; border-color: #cbd5e1 !important;
167
  }
168
- .gradio-container .gr-input-text:focus, .gradio-container .gr-dropdown:focus {
169
- border-color: #14b8a6 !important; box-shadow: 0 0 0 2px rgba(20, 184, 166, 0.3) !important;
 
170
  }
 
 
171
  """
172
 
173
- with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="gray", font=gr.themes.GoogleFont("Inter")), css=css) as demo:
174
  gr.Markdown(
175
  """
176
- <div style="text-align: center; margin-bottom: 2rem;">
177
- <h1 style="font-size: 2.5rem; font-weight: 700; color: #0f766e;">🗺️ Умный навигатор по Бишкеку</h1>
178
- <p style="color: #0d9488; font-size: 1.1rem;">Постройте оптимальный маршрут с учётом прогноза пробок</p>
179
  </div>
180
  """
181
  )
182
  with gr.Row(equal_height=False):
183
  with gr.Column(scale=1):
184
  with gr.Group():
185
- start_address_input = gr.Textbox(label="Откуда?", placeholder="7 микрорайон")
186
- end_address_input = gr.Textbox(label="Куда?", placeholder="Ошский рынок")
187
 
188
- with gr.Accordion("🗓️ Указать время и дату", open=False):
189
- day_dropdown = gr.Dropdown(label="День недели", choices=["Понедельник", "Вторник", "Среда", "Четверг", "Пятница", "Суббота", "Воскресенье"], value="Понедельник", type="index")
190
- current_time = datetime.datetime.now()
191
- hour_slider = gr.Slider(label="Час", minimum=0, maximum=23, step=1, value=current_time.hour)
192
- minute_slider = gr.Slider(label="Минута", minimum=0, maximum=59, step=1, value=current_time.minute)
 
193
 
194
  build_btn = gr.Button("🚀 Найти лучший маршрут", variant="primary")
195
  output_info = gr.Markdown(label="Информация о маршруте", elem_id="info-box")
196
 
197
- with gr.Column(scale=2):
198
  output_map_html = gr.HTML(label="Карта с маршрутом", elem_id="map-output")
199
 
200
  build_btn.click(
 
71
  return None, None
72
 
73
  def plot_route_on_map(G, route, start_point, end_point):
74
+ """Отрисовывает один маршрут на большой цветной карте со стильными маркерами."""
75
+ # Возвращаем цветную карту OpenStreetMap
76
+ m = folium.Map(tiles="OpenStreetMap")
77
 
78
  if route:
79
  points = [(G.nodes[node]['y'], G.nodes[node]['x']) for node in route]
80
+
81
+ # Белая "подложка" для контраста и лучшей видимости
82
+ folium.PolyLine(points, color="white", weight=10, opacity=0.8).add_to(m)
83
+ # Основная линия маршрута - яркий, приятный синий цвет
84
+ main_line = folium.PolyLine(points, color="#3b82f6", weight=5, opacity=1, tooltip="Оптимальный маршрут")
85
  main_line.add_to(m)
86
+
87
  m.fit_bounds(main_line.get_bounds(), padding=(30, 30))
88
 
89
+ # Новые, более стильные маркеры с использованием иконок Font Awesome
90
+ folium.Marker(
91
+ location=start_point, popup="<b>Точка А (Старт)</b>",
92
+ icon=folium.Icon(color="blue", icon="play-circle", prefix='fa')
93
+ ).add_to(m)
94
+ folium.Marker(
95
+ location=end_point, popup="<b>Точка Б (Финиш)</b>",
96
+ icon=folium.Icon(color="red", icon="flag-checkered", prefix='fa')
97
+ ).add_to(m)
98
 
99
  return m
100
 
 
105
  # ==============================================================================
106
  print("\n🚀 Шаг 5: Настройка и запуск веб-интерфейса Gradio...")
107
 
108
+ geolocator = Nominatim(user_agent="bishkek_navigator_app_v5")
109
  geocode = RateLimiter(geolocator.geocode, min_delay_seconds=1)
110
 
111
  def find_and_plot_route_by_address(start_address, end_address, day_of_week, hour, minute):
 
140
 
141
  final_map = plot_route_on_map(G, route, start_point, end_point)
142
 
143
+ # Форматирование вывода в Markdown с новой цветовой схемой
144
  output_md = f"""
145
+ ### ✅ Маршрут построен!
146
+ - **Время в пути:** <span style="font-size: 1.1em; color: #3b82f6;">**~{total_time_min:.1f} мин.**</span>
147
+ - **Расстояние:** **{distance_km:.2f} км**
148
+ - **Светофоров на пути:** **{signals_on_route}**
 
149
  """
150
+ return final_map._repr_html_(), output_md
151
 
152
  except Exception as e:
153
  return None, f"### 💥 Произошла внутренняя ошибка:\n`{e}`"
154
 
155
+ # CSS для новой сине-голубой темы
156
  css = """
157
+ body { font-family: 'Inter', sans-serif; }
158
+ .gradio-container { max-width: 100% !important; }
159
+ /* Кнопка */
 
 
160
  .gr-button-primary {
161
+ background: linear-gradient(to right, #60a5fa, #3b82f6);
162
+ color: white; font-weight: bold; border: none; transition: all 0.3s ease;
163
+ box-shadow: 0 4px 14px 0 rgba(59, 130, 246, 0.39);
 
 
 
 
 
 
 
 
 
 
 
 
164
  }
165
+ .gr-button-primary:hover {
166
+ background: linear-gradient(to right, #3b82f6, #2563eb);
167
+ box-shadow: 0 6px 20px 0 rgba(59, 130, 246, 0.45); transform: translateY(-1px);
168
  }
169
+ /* Информационный блок */
170
+ #info-box { background-color: #f9fafb; padding: 1.5rem !important; border-radius: 0.75rem; border: 1px solid #e5e7eb;}
171
  """
172
 
173
+ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="blue", font=gr.themes.GoogleFont("Inter")), css=css) as demo:
174
  gr.Markdown(
175
  """
176
+ <div style="text-align: center;">
177
+ <h1 style="font-size: 2.5rem; font-weight: 700;">🗺️ Умный навигатор по Бишкеку</h1>
178
+ <p style="color: #4b5563; font-size: 1.1rem;">Постройте оптимальный маршрут с учётом прогноза пробок</p>
179
  </div>
180
  """
181
  )
182
  with gr.Row(equal_height=False):
183
  with gr.Column(scale=1):
184
  with gr.Group():
185
+ start_address_input = gr.Textbox(label="Откуда?", placeholder="Например, ЦУМ")
186
+ end_address_input = gr.Textbox(label="Куда?", placeholder="Например, парк Ата-Тюрк")
187
 
188
+ # Убрали Accordion, теперь настройки времени всегда видны
189
+ gr.Markdown("#### 🗓️ Укажите время и дату")
190
+ day_dropdown = gr.Dropdown(label="День недели", choices=["Понедельник", "Вторник", "Среда", "Четверг", "Пятница", "Суббота", "Воскресенье"], value="Понедельник", type="index")
191
+ current_time = datetime.datetime.now()
192
+ hour_slider = gr.Slider(label="Час", minimum=0, maximum=23, step=1, value=current_time.hour)
193
+ minute_slider = gr.Slider(label="Минута", minimum=0, maximum=59, step=1, value=current_time.minute)
194
 
195
  build_btn = gr.Button("🚀 Найти лучший маршрут", variant="primary")
196
  output_info = gr.Markdown(label="Информация о маршруте", elem_id="info-box")
197
 
198
+ with gr.Column(scale=3): # Увеличили масштаб для карты, чтобы она была больше
199
  output_map_html = gr.HTML(label="Карта с маршрутом", elem_id="map-output")
200
 
201
  build_btn.click(