Spaces:
Sleeping
Sleeping
Zane Falcao commited on
Commit ·
4917185
1
Parent(s): 2a1e3e4
updates
Browse files
app.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
from fastapi import FastAPI,
|
| 2 |
-
from fastapi.responses import
|
| 3 |
from fastapi.middleware.cors import CORSMiddleware
|
| 4 |
import pandas as pd
|
| 5 |
import numpy as np
|
|
@@ -16,12 +16,11 @@ import os
|
|
| 16 |
import time
|
| 17 |
from functools import lru_cache
|
| 18 |
from rtree import index
|
| 19 |
-
import gc
|
| 20 |
-
import shutil
|
| 21 |
-
from typing import Optional, List, Dict, Any, Union
|
| 22 |
from pydantic import BaseModel, Field
|
|
|
|
| 23 |
|
| 24 |
-
|
|
|
|
| 25 |
|
| 26 |
# Add CORS middleware
|
| 27 |
app.add_middleware(
|
|
@@ -35,8 +34,11 @@ app.add_middleware(
|
|
| 35 |
# Create temp directory for files
|
| 36 |
os.makedirs('temp', exist_ok=True)
|
| 37 |
|
| 38 |
-
#
|
| 39 |
-
|
|
|
|
|
|
|
|
|
|
| 40 |
def default(self, obj):
|
| 41 |
if isinstance(obj, np.integer):
|
| 42 |
return int(obj)
|
|
@@ -44,13 +46,10 @@ class NumpyEncoder(json.JSONEncoder):
|
|
| 44 |
return float(obj)
|
| 45 |
elif isinstance(obj, np.ndarray):
|
| 46 |
return obj.tolist()
|
| 47 |
-
return super(
|
| 48 |
|
| 49 |
-
#
|
| 50 |
-
|
| 51 |
-
|
| 52 |
-
# Define Pydantic models for API responses
|
| 53 |
-
class Location(BaseModel):
|
| 54 |
lat: float
|
| 55 |
lon: float
|
| 56 |
|
|
@@ -61,7 +60,7 @@ class Store(BaseModel):
|
|
| 61 |
distance: float
|
| 62 |
estimated_delivery_time: int
|
| 63 |
product_categories: str
|
| 64 |
-
location:
|
| 65 |
|
| 66 |
class StoresResponse(BaseModel):
|
| 67 |
status: str
|
|
@@ -70,7 +69,7 @@ class StoresResponse(BaseModel):
|
|
| 70 |
class ErrorResponse(BaseModel):
|
| 71 |
status: str
|
| 72 |
message: str
|
| 73 |
-
|
| 74 |
class StoreLocator:
|
| 75 |
def __init__(self, stores_dataframe):
|
| 76 |
self.stores_df = stores_dataframe
|
|
@@ -79,30 +78,16 @@ class StoreLocator:
|
|
| 79 |
self.spatial_index = self._build_spatial_index()
|
| 80 |
|
| 81 |
@lru_cache(maxsize=50)
|
| 82 |
-
def initialize_graph(self, center_point, dist=
|
| 83 |
"""Initialize road network graph with caching"""
|
| 84 |
cache_key = f"{center_point[0]}_{center_point[1]}"
|
| 85 |
if cache_key in self.graph_cache:
|
| 86 |
self.network_graph = self.graph_cache[cache_key]
|
| 87 |
return True
|
| 88 |
try:
|
| 89 |
-
|
| 90 |
-
self.network_graph = ox.graph_from_point(
|
| 91 |
-
center_point,
|
| 92 |
-
dist=dist,
|
| 93 |
-
network_type="drive",
|
| 94 |
-
simplify=True,
|
| 95 |
-
retain_all=False
|
| 96 |
-
)
|
| 97 |
self.network_graph = ox.add_edge_speeds(self.network_graph)
|
| 98 |
self.network_graph = ox.add_edge_travel_times(self.network_graph)
|
| 99 |
-
|
| 100 |
-
# Store in cache
|
| 101 |
-
self.graph_cache[cache_key] = self.network_graph
|
| 102 |
-
|
| 103 |
-
# Force garbage collection
|
| 104 |
-
gc.collect()
|
| 105 |
-
|
| 106 |
return True
|
| 107 |
except Exception as e:
|
| 108 |
print(f"Error initializing graph: {str(e)}")
|
|
@@ -137,7 +122,7 @@ class StoreLocator:
|
|
| 137 |
multiplier = 1.0
|
| 138 |
|
| 139 |
return round(base_minutes * multiplier)
|
| 140 |
-
|
| 141 |
def find_nearby_stores(self, lat, lon, radius=5):
|
| 142 |
"""Find stores within radius using spatial index"""
|
| 143 |
nearby_stores = []
|
|
@@ -167,24 +152,20 @@ class StoreLocator:
|
|
| 167 |
return sorted(nearby_stores, key=lambda x: x['distance'])
|
| 168 |
|
| 169 |
def create_store_map(self, center_lat, center_lon, radius=5):
|
| 170 |
-
"""Create an interactive map with store locations
|
| 171 |
# Create base map
|
| 172 |
-
m = folium.Map(
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
| 176 |
-
|
| 177 |
|
| 178 |
# Create marker cluster for better performance with many markers
|
| 179 |
marker_cluster = plugins.MarkerCluster().add_to(m)
|
| 180 |
-
|
| 181 |
# Add stores to map
|
| 182 |
nearby_stores = self.find_nearby_stores(center_lat, center_lon, radius)
|
| 183 |
|
| 184 |
-
|
| 185 |
-
max_stores = min(len(nearby_stores), 50) # Cap at 50 stores
|
| 186 |
-
|
| 187 |
-
for store in nearby_stores[:max_stores]:
|
| 188 |
# Prepare popup content
|
| 189 |
popup_content = f"""
|
| 190 |
<div style='width: 200px'>
|
|
@@ -203,15 +184,14 @@ class StoreLocator:
|
|
| 203 |
icon=folium.Icon(color='red', icon='info-sign')
|
| 204 |
).add_to(marker_cluster)
|
| 205 |
|
| 206 |
-
# Add line to show distance from center
|
| 207 |
-
|
| 208 |
-
|
| 209 |
-
|
| 210 |
-
|
| 211 |
-
|
| 212 |
-
|
| 213 |
-
|
| 214 |
-
).add_to(m)
|
| 215 |
|
| 216 |
# Add current location marker
|
| 217 |
folium.Marker(
|
|
@@ -220,16 +200,84 @@ class StoreLocator:
|
|
| 220 |
icon=folium.Icon(color='green', icon='home')
|
| 221 |
).add_to(m)
|
| 222 |
|
| 223 |
-
# Add
|
|
|
|
|
|
|
|
|
|
| 224 |
folium.LayerControl().add_to(m)
|
| 225 |
|
| 226 |
return m
|
| 227 |
-
|
| 228 |
-
# Initialize store locator
|
| 229 |
store_locator = StoreLocator(stores_df)
|
| 230 |
|
| 231 |
-
|
| 232 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 233 |
temp_dir = 'temp'
|
| 234 |
if os.path.exists(temp_dir):
|
| 235 |
for file in os.listdir(temp_dir):
|
|
@@ -241,24 +289,13 @@ def cleanup_temp_files():
|
|
| 241 |
os.remove(file_path)
|
| 242 |
except Exception as e:
|
| 243 |
print(f"Error cleaning up temp files: {e}")
|
|
|
|
|
|
|
|
|
|
| 244 |
|
| 245 |
-
# Register cleanup on startup and shutdown
|
| 246 |
-
@app.on_event("startup")
|
| 247 |
-
async def startup_event():
|
| 248 |
-
cleanup_temp_files()
|
| 249 |
-
|
| 250 |
-
@app.on_event("shutdown")
|
| 251 |
-
async def shutdown_event():
|
| 252 |
-
# Clean up all temporary files on shutdown
|
| 253 |
-
try:
|
| 254 |
-
shutil.rmtree('temp')
|
| 255 |
-
except Exception as e:
|
| 256 |
-
print(f"Error cleaning up temp directory: {e}")
|
| 257 |
-
|
| 258 |
-
# Routes
|
| 259 |
@app.get("/", response_class=HTMLResponse)
|
| 260 |
async def home():
|
| 261 |
-
"""API
|
| 262 |
html_content = f"""
|
| 263 |
<!DOCTYPE html>
|
| 264 |
<html lang="en">
|
|
@@ -288,30 +325,30 @@ async def home():
|
|
| 288 |
|
| 289 |
<h2>1. Find Nearby Stores (JSON Response)</h2>
|
| 290 |
<pre>
|
| 291 |
-
/api/stores/nearby?lat=18.9695&lon=72.8320&radius=1
|
| 292 |
</pre>
|
| 293 |
<p>Use this to get store details near Market Road area within 1km</p>
|
| 294 |
|
| 295 |
<h2>2. View Basic Store Map</h2>
|
| 296 |
<pre>
|
| 297 |
-
/api/stores/map?lat=18.9701&lon=72.8330&radius=0.5
|
| 298 |
</pre>
|
| 299 |
<p>Shows map centered at Main Street with 500m radius</p>
|
| 300 |
|
| 301 |
<h2>3. View All Store Locations with Color Coding</h2>
|
| 302 |
<pre>
|
| 303 |
-
/api/stores/locations?lat=18.9685&lon=72.8325&radius=2
|
| 304 |
</pre>
|
| 305 |
<p>Shows detailed map with color-coded stores within 2km</p>
|
| 306 |
|
| 307 |
<h2>4. Get Route Between Points</h2>
|
| 308 |
<p>Example routes:</p>
|
| 309 |
<pre>
|
| 310 |
-
# Route from Park Avenue to Hill Road stores
|
| 311 |
-
/api/stores/route?user_lat=18.9710&user_lon=72.8335&store_lat=18.9705&store_lon=72.8345&viz_type=simple
|
| 312 |
|
| 313 |
# Route from Main Street to Market Road stores
|
| 314 |
-
/api/stores/route?user_lat=18.9701&user_lon=72.8330&store_lat=18.9695&store_lon=72.8320&viz_type=simple
|
| 315 |
</pre>
|
| 316 |
|
| 317 |
<h2>Key Location Points in Dataset:</h2>
|
|
@@ -322,21 +359,20 @@ async def home():
|
|
| 322 |
<li>Shopping Center: 18.9670, 72.8300</li>
|
| 323 |
<li>Commercial Street: 18.9690, 72.8340</li>
|
| 324 |
</ul>
|
| 325 |
-
|
| 326 |
-
<h2>API Documentation</h2>
|
| 327 |
-
<p>You can view the interactive API documentation at: <a href="/docs">/docs</a></p>
|
| 328 |
</body>
|
| 329 |
</html>
|
| 330 |
"""
|
| 331 |
-
return html_content
|
| 332 |
|
| 333 |
@app.get("/api/stores/nearby", response_model=StoresResponse, responses={400: {"model": ErrorResponse}})
|
| 334 |
async def get_nearby_stores(
|
| 335 |
lat: float = Query(..., description="Latitude of user location"),
|
| 336 |
lon: float = Query(..., description="Longitude of user location"),
|
| 337 |
-
radius: float = Query(5
|
| 338 |
):
|
| 339 |
-
"""
|
|
|
|
|
|
|
| 340 |
try:
|
| 341 |
nearby_stores = store_locator.find_nearby_stores(lat, lon, radius)
|
| 342 |
return {"status": "success", "stores": nearby_stores}
|
|
@@ -347,13 +383,12 @@ async def get_nearby_stores(
|
|
| 347 |
async def get_stores_map(
|
| 348 |
lat: float = Query(..., description="Latitude of center point"),
|
| 349 |
lon: float = Query(..., description="Longitude of center point"),
|
| 350 |
-
radius: float = Query(5
|
| 351 |
):
|
| 352 |
-
"""
|
|
|
|
|
|
|
| 353 |
try:
|
| 354 |
-
# Clean up temp files before creating new ones
|
| 355 |
-
cleanup_temp_files()
|
| 356 |
-
|
| 357 |
store_map = store_locator.create_store_map(lat, lon, radius)
|
| 358 |
|
| 359 |
# Create complete HTML content
|
|
@@ -396,31 +431,26 @@ async def get_stores_map(
|
|
| 396 |
with open(file_path, 'w', encoding='utf-8') as f:
|
| 397 |
f.write(html_content)
|
| 398 |
|
| 399 |
-
# Return the file
|
| 400 |
-
return
|
| 401 |
|
| 402 |
except Exception as e:
|
| 403 |
raise HTTPException(status_code=400, detail=str(e))
|
| 404 |
-
|
| 405 |
@app.get("/api/stores/route", response_class=HTMLResponse, responses={400: {"model": ErrorResponse}, 404: {"model": ErrorResponse}})
|
| 406 |
async def get_store_route(
|
| 407 |
-
user_lat: float = Query(..., description="
|
| 408 |
-
user_lon: float = Query(..., description="
|
| 409 |
-
store_lat: float = Query(..., description="
|
| 410 |
-
store_lon: float = Query(..., description="
|
| 411 |
-
viz_type: str = Query("simple", description="Visualization type (simple or advanced)")
|
| 412 |
):
|
| 413 |
-
"""
|
|
|
|
|
|
|
| 414 |
try:
|
| 415 |
-
# Clean up temp files before creating new ones
|
| 416 |
-
cleanup_temp_files()
|
| 417 |
-
|
| 418 |
# Initialize graph if not already initialized
|
| 419 |
-
# Use a smaller distance to reduce memory usage
|
| 420 |
if store_locator.network_graph is None:
|
| 421 |
-
|
| 422 |
-
if not success:
|
| 423 |
-
raise HTTPException(status_code=400, detail="Unable to initialize graph, try a different location")
|
| 424 |
|
| 425 |
# Get nearest nodes
|
| 426 |
start_node = ox.distance.nearest_nodes(
|
|
@@ -429,7 +459,7 @@ async def get_store_route(
|
|
| 429 |
store_locator.network_graph, store_lon, store_lat)
|
| 430 |
|
| 431 |
try:
|
| 432 |
-
# Calculate
|
| 433 |
path_time = nx.shortest_path(
|
| 434 |
store_locator.network_graph,
|
| 435 |
start_node,
|
|
@@ -437,285 +467,206 @@ async def get_store_route(
|
|
| 437 |
weight='travel_time'
|
| 438 |
)
|
| 439 |
|
| 440 |
-
|
| 441 |
-
|
| 442 |
-
|
| 443 |
-
|
| 444 |
-
|
| 445 |
-
|
| 446 |
-
|
| 447 |
-
|
| 448 |
-
|
| 449 |
-
|
| 450 |
-
|
| 451 |
-
|
| 452 |
-
|
| 453 |
-
|
| 454 |
-
|
| 455 |
-
|
| 456 |
-
|
| 457 |
-
|
| 458 |
-
|
| 459 |
-
|
| 460 |
-
|
| 461 |
-
|
| 462 |
-
|
| 463 |
-
|
| 464 |
-
|
| 465 |
-
|
| 466 |
-
|
| 467 |
-
|
| 468 |
-
|
| 469 |
-
|
| 470 |
-
|
| 471 |
-
|
| 472 |
-
|
| 473 |
-
|
| 474 |
-
|
| 475 |
-
|
| 476 |
-
|
| 477 |
-
|
| 478 |
-
|
| 479 |
-
|
| 480 |
-
|
| 481 |
-
|
| 482 |
-
|
| 483 |
-
|
| 484 |
-
|
| 485 |
-
|
| 486 |
-
|
| 487 |
-
|
| 488 |
-
|
| 489 |
-
|
| 490 |
-
|
| 491 |
-
|
| 492 |
-
|
| 493 |
-
|
| 494 |
-
|
| 495 |
-
|
| 496 |
-
|
| 497 |
-
|
| 498 |
-
|
| 499 |
-
|
| 500 |
-
|
| 501 |
-
|
| 502 |
-
|
| 503 |
-
|
| 504 |
-
|
| 505 |
-
# Create complete HTML content
|
| 506 |
-
html_content = f"""
|
| 507 |
-
<!DOCTYPE html>
|
| 508 |
-
<html>
|
| 509 |
-
<head>
|
| 510 |
-
<meta charset="utf-8">
|
| 511 |
-
<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no" />
|
| 512 |
-
<title>Simple Route Map</title>
|
| 513 |
-
<style>
|
| 514 |
-
body {{
|
| 515 |
-
margin: 0;
|
| 516 |
-
padding: 0;
|
| 517 |
-
width: 100vw;
|
| 518 |
-
height: 100vh;
|
| 519 |
-
overflow: hidden;
|
| 520 |
-
}}
|
| 521 |
-
#map {{
|
| 522 |
-
width: 100%;
|
| 523 |
-
height: 100%;
|
| 524 |
-
}}
|
| 525 |
-
</style>
|
| 526 |
-
</head>
|
| 527 |
-
<body>
|
| 528 |
-
{m.get_root().render()}
|
| 529 |
-
<script>
|
| 530 |
-
window.onload = function() {{
|
| 531 |
-
setTimeout(function() {{
|
| 532 |
-
window.dispatchEvent(new Event('resize'));
|
| 533 |
-
}}, 1000);
|
| 534 |
-
}};
|
| 535 |
-
</script>
|
| 536 |
-
</body>
|
| 537 |
-
</html>
|
| 538 |
-
"""
|
| 539 |
-
|
| 540 |
-
# Save the HTML to a file
|
| 541 |
-
file_path = 'temp/simple_route_map.html'
|
| 542 |
-
with open(file_path, 'w', encoding='utf-8') as f:
|
| 543 |
-
f.write(html_content)
|
| 544 |
-
|
| 545 |
-
# Return the HTML content
|
| 546 |
-
return html_content
|
| 547 |
-
|
| 548 |
-
else:
|
| 549 |
-
# WARNING: Advanced visualization - may cause memory issues on limited resources
|
| 550 |
-
|
| 551 |
-
# Limit the path nodes to reduce memory usage
|
| 552 |
-
# Only include every Nth node
|
| 553 |
-
step = max(1, len(path_time) // 30) # Maximum 30 points
|
| 554 |
-
simplified_path = path_time[::step]
|
| 555 |
-
if path_time[-1] not in simplified_path:
|
| 556 |
-
simplified_path.append(path_time[-1])
|
| 557 |
-
|
| 558 |
-
# Create animation data (simplified)
|
| 559 |
-
lst_start, lst_end = [], []
|
| 560 |
-
start_x, start_y = [], []
|
| 561 |
-
end_x, end_y = [], []
|
| 562 |
-
lst_length, lst_time = [], []
|
| 563 |
-
|
| 564 |
-
for a, b in zip(simplified_path[:-1], simplified_path[1:]):
|
| 565 |
-
lst_start.append(a)
|
| 566 |
-
lst_end.append(b)
|
| 567 |
-
|
| 568 |
-
# Calculate accumulated length and time between simplified points
|
| 569 |
-
segment_length = 0
|
| 570 |
-
segment_time = 0
|
| 571 |
-
path_segment = nx.shortest_path(
|
| 572 |
-
store_locator.network_graph, a, b, weight='travel_time')
|
| 573 |
-
|
| 574 |
-
for i in range(len(path_segment) - 1):
|
| 575 |
-
u, v = path_segment[i], path_segment[i + 1]
|
| 576 |
-
segment_length += store_locator.network_graph.edges[(u, v, 0)]['length']
|
| 577 |
-
segment_time += store_locator.network_graph.edges[(u, v, 0)]['travel_time']
|
| 578 |
-
|
| 579 |
-
lst_length.append(round(segment_length))
|
| 580 |
-
lst_time.append(round(segment_time))
|
| 581 |
-
start_x.append(store_locator.network_graph.nodes[a]['x'])
|
| 582 |
-
start_y.append(store_locator.network_graph.nodes[a]['y'])
|
| 583 |
-
end_x.append(store_locator.network_graph.nodes[b]['x'])
|
| 584 |
-
end_y.append(store_locator.network_graph.nodes[b]['y'])
|
| 585 |
-
|
| 586 |
-
df = pd.DataFrame(
|
| 587 |
-
list(zip(lst_start, lst_end, start_x, start_y, end_x, end_y,
|
| 588 |
-
lst_length, lst_time)),
|
| 589 |
-
columns=["start", "end", "start_x", "start_y",
|
| 590 |
-
"end_x", "end_y", "length", "travel_time"]
|
| 591 |
-
).reset_index().rename(columns={"index": "id"})
|
| 592 |
-
|
| 593 |
-
# Create animation using plotly (reduced complexity)
|
| 594 |
-
df_start = df[df["start"] == lst_start[0]]
|
| 595 |
-
df_end = df[df["end"] == lst_end[-1]]
|
| 596 |
-
|
| 597 |
-
fig = px.scatter_mapbox(
|
| 598 |
data_frame=df,
|
| 599 |
lon="start_x",
|
| 600 |
-
lat="start_y"
|
| 601 |
-
|
| 602 |
-
|
| 603 |
-
|
| 604 |
-
|
| 605 |
-
|
| 606 |
-
|
| 607 |
-
|
| 608 |
-
|
| 609 |
-
|
| 610 |
-
|
| 611 |
-
|
| 612 |
-
|
| 613 |
-
|
| 614 |
-
|
| 615 |
-
|
| 616 |
-
|
| 617 |
-
|
| 618 |
-
|
| 619 |
-
|
| 620 |
-
|
| 621 |
-
|
| 622 |
-
|
| 623 |
-
|
| 624 |
-
|
| 625 |
-
|
| 626 |
-
|
| 627 |
-
|
| 628 |
-
|
| 629 |
-
|
| 630 |
-
|
| 631 |
-
|
| 632 |
-
|
| 633 |
-
|
| 634 |
-
|
| 635 |
-
|
| 636 |
-
|
| 637 |
-
|
| 638 |
-
|
| 639 |
-
|
| 640 |
-
|
| 641 |
-
|
| 642 |
-
|
| 643 |
-
|
| 644 |
-
|
| 645 |
-
|
| 646 |
-
|
| 647 |
-
|
| 648 |
-
|
| 649 |
-
|
| 650 |
-
|
| 651 |
-
|
| 652 |
-
|
| 653 |
-
|
| 654 |
-
|
| 655 |
-
|
| 656 |
-
|
| 657 |
-
|
| 658 |
-
|
| 659 |
-
|
| 660 |
-
|
| 661 |
-
|
| 662 |
-
|
| 663 |
-
|
| 664 |
-
|
| 665 |
-
|
| 666 |
-
|
| 667 |
-
|
| 668 |
-
|
| 669 |
-
|
| 670 |
-
|
| 671 |
-
|
| 672 |
-
|
| 673 |
-
|
| 674 |
-
|
| 675 |
-
|
| 676 |
-
|
| 677 |
-
|
| 678 |
-
|
| 679 |
-
|
| 680 |
-
|
| 681 |
-
|
| 682 |
-
|
| 683 |
-
|
| 684 |
-
|
| 685 |
-
|
| 686 |
-
|
| 687 |
-
|
| 688 |
-
|
| 689 |
-
|
| 690 |
-
|
| 691 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 692 |
|
| 693 |
except nx.NetworkXNoPath:
|
| 694 |
raise HTTPException(status_code=404, detail="No route found")
|
| 695 |
|
| 696 |
-
except HTTPException:
|
| 697 |
-
raise
|
| 698 |
except Exception as e:
|
| 699 |
raise HTTPException(status_code=400, detail=str(e))
|
| 700 |
-
|
| 701 |
@app.get("/api/stores/locations", response_class=HTMLResponse, responses={400: {"model": ErrorResponse}})
|
| 702 |
async def get_all_store_locations(
|
| 703 |
lat: float = Query(..., description="Latitude of center point"),
|
| 704 |
lon: float = Query(..., description="Longitude of center point"),
|
| 705 |
-
radius: float = Query(10
|
| 706 |
):
|
| 707 |
-
"""
|
|
|
|
|
|
|
| 708 |
try:
|
| 709 |
-
# Clean up temp files before creating new ones
|
| 710 |
-
cleanup_temp_files()
|
| 711 |
-
|
| 712 |
# Get nearby stores
|
| 713 |
nearby_stores = store_locator.find_nearby_stores(lat, lon, radius)
|
| 714 |
|
| 715 |
-
# Limit number of stores for memory optimization
|
| 716 |
-
max_stores = min(len(nearby_stores), 50)
|
| 717 |
-
nearby_stores = nearby_stores[:max_stores]
|
| 718 |
-
|
| 719 |
# Create base map centered on user location
|
| 720 |
m = folium.Map(
|
| 721 |
location=[lat, lon],
|
|
@@ -740,14 +691,16 @@ async def get_all_store_locations(
|
|
| 740 |
else:
|
| 741 |
color = 'blue' # Further away
|
| 742 |
|
| 743 |
-
# Create
|
| 744 |
popup_content = f"""
|
| 745 |
<div style='width: 200px; font-size: 14px;'>
|
| 746 |
<h4 style='color: {color}; margin: 0 0 8px 0;'>{store['store_name']}</h4>
|
|
|
|
| 747 |
<b>Distance:</b> {store['distance']} km<br>
|
| 748 |
<b>Est. Delivery:</b> {store['estimated_delivery_time']} mins<br>
|
|
|
|
| 749 |
<b>Categories:</b> {store['product_categories']}<br>
|
| 750 |
-
<button onclick="window.location.href='/api/stores/route?user_lat={lat}&user_lon={lon}&store_lat={store['location']['lat']}&store_lon={store['location']['lon']}
|
| 751 |
style='margin-top: 8px; padding: 8px; width: 100%; background-color: #007bff; color: white; border: none; border-radius: 4px;'>
|
| 752 |
Get Route
|
| 753 |
</button>
|
|
@@ -762,24 +715,24 @@ async def get_all_store_locations(
|
|
| 762 |
tooltip=f"{store['store_name']} ({store['distance']} km)"
|
| 763 |
).add_to(m)
|
| 764 |
|
| 765 |
-
# Add circle to show distance
|
| 766 |
-
|
| 767 |
-
|
| 768 |
-
|
| 769 |
-
|
| 770 |
-
|
| 771 |
-
|
| 772 |
-
|
| 773 |
-
).add_to(m)
|
| 774 |
|
| 775 |
-
# Add distance circles from user location
|
| 776 |
-
for
|
| 777 |
folium.Circle(
|
| 778 |
location=[lat, lon],
|
| 779 |
-
radius=
|
| 780 |
color=color,
|
| 781 |
fill=False,
|
| 782 |
-
weight=1,
|
|
|
|
| 783 |
).add_to(m)
|
| 784 |
|
| 785 |
# Create mobile-friendly HTML content
|
|
@@ -854,36 +807,19 @@ async def get_all_store_locations(
|
|
| 854 |
with open(file_path, 'w', encoding='utf-8') as f:
|
| 855 |
f.write(html_content)
|
| 856 |
|
| 857 |
-
|
| 858 |
-
return html_content
|
| 859 |
|
| 860 |
except Exception as e:
|
| 861 |
raise HTTPException(status_code=400, detail=str(e))
|
| 862 |
|
| 863 |
-
# Add
|
| 864 |
-
@app.
|
| 865 |
-
async def
|
| 866 |
-
|
| 867 |
-
|
| 868 |
-
|
| 869 |
-
raise HTTPException(status_code=404, detail="File not found")
|
| 870 |
-
return FileResponse(full_path)
|
| 871 |
-
|
| 872 |
-
# Add memory monitoring and management middleware
|
| 873 |
-
@app.middleware("http")
|
| 874 |
-
async def add_memory_management(request: Request, call_next):
|
| 875 |
-
# Cleanup before processing request
|
| 876 |
-
cleanup_temp_files()
|
| 877 |
-
|
| 878 |
-
# Process the request
|
| 879 |
-
response = await call_next(request)
|
| 880 |
-
|
| 881 |
-
# Cleanup after processing request
|
| 882 |
-
gc.collect()
|
| 883 |
-
|
| 884 |
-
return response
|
| 885 |
|
| 886 |
-
#
|
| 887 |
if __name__ == "__main__":
|
| 888 |
import uvicorn
|
| 889 |
uvicorn.run(app, host="0.0.0.0", port=8000)
|
|
|
|
| 1 |
+
from fastapi import FastAPI, HTTPException, Query, Request, Response
|
| 2 |
+
from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
|
| 3 |
from fastapi.middleware.cors import CORSMiddleware
|
| 4 |
import pandas as pd
|
| 5 |
import numpy as np
|
|
|
|
| 16 |
import time
|
| 17 |
from functools import lru_cache
|
| 18 |
from rtree import index
|
|
|
|
|
|
|
|
|
|
| 19 |
from pydantic import BaseModel, Field
|
| 20 |
+
from typing import List, Dict, Any, Optional
|
| 21 |
|
| 22 |
+
# Create app instance
|
| 23 |
+
app = FastAPI(title="Store Locator API")
|
| 24 |
|
| 25 |
# Add CORS middleware
|
| 26 |
app.add_middleware(
|
|
|
|
| 34 |
# Create temp directory for files
|
| 35 |
os.makedirs('temp', exist_ok=True)
|
| 36 |
|
| 37 |
+
# Load and prepare the store data
|
| 38 |
+
stores_df = pd.read_csv('dataset of 50 stores.csv')
|
| 39 |
+
|
| 40 |
+
# Custom JSON encoder for numpy types
|
| 41 |
+
class NumpyJSONEncoder(json.JSONEncoder):
|
| 42 |
def default(self, obj):
|
| 43 |
if isinstance(obj, np.integer):
|
| 44 |
return int(obj)
|
|
|
|
| 46 |
return float(obj)
|
| 47 |
elif isinstance(obj, np.ndarray):
|
| 48 |
return obj.tolist()
|
| 49 |
+
return super().default(obj)
|
| 50 |
|
| 51 |
+
# Pydantic models for response validation
|
| 52 |
+
class StoreLocation(BaseModel):
|
|
|
|
|
|
|
|
|
|
| 53 |
lat: float
|
| 54 |
lon: float
|
| 55 |
|
|
|
|
| 60 |
distance: float
|
| 61 |
estimated_delivery_time: int
|
| 62 |
product_categories: str
|
| 63 |
+
location: StoreLocation
|
| 64 |
|
| 65 |
class StoresResponse(BaseModel):
|
| 66 |
status: str
|
|
|
|
| 69 |
class ErrorResponse(BaseModel):
|
| 70 |
status: str
|
| 71 |
message: str
|
| 72 |
+
|
| 73 |
class StoreLocator:
|
| 74 |
def __init__(self, stores_dataframe):
|
| 75 |
self.stores_df = stores_dataframe
|
|
|
|
| 78 |
self.spatial_index = self._build_spatial_index()
|
| 79 |
|
| 80 |
@lru_cache(maxsize=50)
|
| 81 |
+
def initialize_graph(self, center_point, dist=20000):
|
| 82 |
"""Initialize road network graph with caching"""
|
| 83 |
cache_key = f"{center_point[0]}_{center_point[1]}"
|
| 84 |
if cache_key in self.graph_cache:
|
| 85 |
self.network_graph = self.graph_cache[cache_key]
|
| 86 |
return True
|
| 87 |
try:
|
| 88 |
+
self.network_graph = ox.graph_from_point(center_point, dist=dist, network_type="drive")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
self.network_graph = ox.add_edge_speeds(self.network_graph)
|
| 90 |
self.network_graph = ox.add_edge_travel_times(self.network_graph)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 91 |
return True
|
| 92 |
except Exception as e:
|
| 93 |
print(f"Error initializing graph: {str(e)}")
|
|
|
|
| 122 |
multiplier = 1.0
|
| 123 |
|
| 124 |
return round(base_minutes * multiplier)
|
| 125 |
+
|
| 126 |
def find_nearby_stores(self, lat, lon, radius=5):
|
| 127 |
"""Find stores within radius using spatial index"""
|
| 128 |
nearby_stores = []
|
|
|
|
| 152 |
return sorted(nearby_stores, key=lambda x: x['distance'])
|
| 153 |
|
| 154 |
def create_store_map(self, center_lat, center_lon, radius=5):
|
| 155 |
+
"""Create an interactive map with store locations"""
|
| 156 |
# Create base map
|
| 157 |
+
m = folium.Map(location=[center_lat, center_lon],
|
| 158 |
+
zoom_start=13,
|
| 159 |
+
tiles="cartodbpositron",
|
| 160 |
+
prefer_canvas=True
|
| 161 |
+
)
|
| 162 |
|
| 163 |
# Create marker cluster for better performance with many markers
|
| 164 |
marker_cluster = plugins.MarkerCluster().add_to(m)
|
|
|
|
| 165 |
# Add stores to map
|
| 166 |
nearby_stores = self.find_nearby_stores(center_lat, center_lon, radius)
|
| 167 |
|
| 168 |
+
for store in nearby_stores:
|
|
|
|
|
|
|
|
|
|
| 169 |
# Prepare popup content
|
| 170 |
popup_content = f"""
|
| 171 |
<div style='width: 200px'>
|
|
|
|
| 184 |
icon=folium.Icon(color='red', icon='info-sign')
|
| 185 |
).add_to(marker_cluster)
|
| 186 |
|
| 187 |
+
# Add line to show distance from center
|
| 188 |
+
folium.PolyLine(
|
| 189 |
+
locations=[[center_lat, center_lon],
|
| 190 |
+
[store['location']['lat'], store['location']['lon']]],
|
| 191 |
+
weight=2,
|
| 192 |
+
color='blue',
|
| 193 |
+
opacity=0.3
|
| 194 |
+
).add_to(m)
|
|
|
|
| 195 |
|
| 196 |
# Add current location marker
|
| 197 |
folium.Marker(
|
|
|
|
| 200 |
icon=folium.Icon(color='green', icon='home')
|
| 201 |
).add_to(m)
|
| 202 |
|
| 203 |
+
# Add fullscreen option
|
| 204 |
+
plugins.Fullscreen().add_to(m)
|
| 205 |
+
|
| 206 |
+
# Add layer control
|
| 207 |
folium.LayerControl().add_to(m)
|
| 208 |
|
| 209 |
return m
|
| 210 |
+
|
| 211 |
+
# Initialize store locator
|
| 212 |
store_locator = StoreLocator(stores_df)
|
| 213 |
|
| 214 |
+
def create_animated_route(G, path, color, weight=3):
|
| 215 |
+
"""Create an animated route visualization"""
|
| 216 |
+
features = []
|
| 217 |
+
timestamps = []
|
| 218 |
+
|
| 219 |
+
# Convert path nodes to coordinates
|
| 220 |
+
route_coords = [
|
| 221 |
+
(G.nodes[node]['y'], G.nodes[node]['x'])
|
| 222 |
+
for node in path
|
| 223 |
+
]
|
| 224 |
+
|
| 225 |
+
# Create features for each segment of the route
|
| 226 |
+
for i in range(len(route_coords) - 1):
|
| 227 |
+
segment = {
|
| 228 |
+
'type': 'Feature',
|
| 229 |
+
'geometry': {
|
| 230 |
+
'type': 'LineString',
|
| 231 |
+
'coordinates': [
|
| 232 |
+
[route_coords[i][1], route_coords[i][0]],
|
| 233 |
+
[route_coords[i+1][1], route_coords[i+1][0]]
|
| 234 |
+
]
|
| 235 |
+
},
|
| 236 |
+
'properties': {
|
| 237 |
+
'times': [datetime.now().isoformat()],
|
| 238 |
+
'style': {
|
| 239 |
+
'color': color,
|
| 240 |
+
'weight': weight,
|
| 241 |
+
'opacity': 0.8
|
| 242 |
+
}
|
| 243 |
+
}
|
| 244 |
+
}
|
| 245 |
+
features.append(segment)
|
| 246 |
+
timestamps.append(datetime.now().isoformat())
|
| 247 |
+
|
| 248 |
+
return features
|
| 249 |
+
|
| 250 |
+
def create_route_animation_data(G, path_time, path_length):
|
| 251 |
+
"""Create animation data for route visualization"""
|
| 252 |
+
lst_start, lst_end = [], []
|
| 253 |
+
start_x, start_y = [], []
|
| 254 |
+
end_x, end_y = [], []
|
| 255 |
+
lst_length, lst_time = [], []
|
| 256 |
+
|
| 257 |
+
# Process time-based path
|
| 258 |
+
for a, b in zip(path_time[:-1], path_time[1:]):
|
| 259 |
+
lst_start.append(a)
|
| 260 |
+
lst_end.append(b)
|
| 261 |
+
lst_length.append(round(G.edges[(a,b,0)]['length']))
|
| 262 |
+
lst_time.append(round(G.edges[(a,b,0)]['travel_time']))
|
| 263 |
+
start_x.append(G.nodes[a]['x'])
|
| 264 |
+
start_y.append(G.nodes[a]['y'])
|
| 265 |
+
end_x.append(G.nodes[b]['x'])
|
| 266 |
+
end_y.append(G.nodes[b]['y'])
|
| 267 |
+
|
| 268 |
+
# Create DataFrame
|
| 269 |
+
df = pd.DataFrame(
|
| 270 |
+
list(zip(lst_start, lst_end, start_x, start_y, end_x, end_y,
|
| 271 |
+
lst_length, lst_time)),
|
| 272 |
+
columns=["start", "end", "start_x", "start_y", "end_x", "end_y",
|
| 273 |
+
"length", "travel_time"]
|
| 274 |
+
).reset_index().rename(columns={"index": "id"})
|
| 275 |
+
|
| 276 |
+
return df
|
| 277 |
+
|
| 278 |
+
# Function to clean up temporary files
|
| 279 |
+
@app.middleware("http")
|
| 280 |
+
async def cleanup_temp_files(request: Request, call_next):
|
| 281 |
temp_dir = 'temp'
|
| 282 |
if os.path.exists(temp_dir):
|
| 283 |
for file in os.listdir(temp_dir):
|
|
|
|
| 289 |
os.remove(file_path)
|
| 290 |
except Exception as e:
|
| 291 |
print(f"Error cleaning up temp files: {e}")
|
| 292 |
+
|
| 293 |
+
response = await call_next(request)
|
| 294 |
+
return response
|
| 295 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 296 |
@app.get("/", response_class=HTMLResponse)
|
| 297 |
async def home():
|
| 298 |
+
"""API documentation homepage"""
|
| 299 |
html_content = f"""
|
| 300 |
<!DOCTYPE html>
|
| 301 |
<html lang="en">
|
|
|
|
| 325 |
|
| 326 |
<h2>1. Find Nearby Stores (JSON Response)</h2>
|
| 327 |
<pre>
|
| 328 |
+
https://maps-yiv5.onrender.com/api/stores/nearby?lat=18.9695&lon=72.8320&radius=1
|
| 329 |
</pre>
|
| 330 |
<p>Use this to get store details near Market Road area within 1km</p>
|
| 331 |
|
| 332 |
<h2>2. View Basic Store Map</h2>
|
| 333 |
<pre>
|
| 334 |
+
https://maps-yiv5.onrender.com/api/stores/map?lat=18.9701&lon=72.8330&radius=0.5
|
| 335 |
</pre>
|
| 336 |
<p>Shows map centered at Main Street with 500m radius</p>
|
| 337 |
|
| 338 |
<h2>3. View All Store Locations with Color Coding</h2>
|
| 339 |
<pre>
|
| 340 |
+
https://maps-yiv5.onrender.com/api/stores/locations?lat=18.9685&lon=72.8325&radius=2
|
| 341 |
</pre>
|
| 342 |
<p>Shows detailed map with color-coded stores within 2km</p>
|
| 343 |
|
| 344 |
<h2>4. Get Route Between Points</h2>
|
| 345 |
<p>Example routes:</p>
|
| 346 |
<pre>
|
| 347 |
+
# Route from Park Avenue to Hill Road stores
|
| 348 |
+
https://maps-yiv5.onrender.com/api/stores/route?user_lat=18.9710&user_lon=72.8335&store_lat=18.9705&store_lon=72.8345&viz_type=simple
|
| 349 |
|
| 350 |
# Route from Main Street to Market Road stores
|
| 351 |
+
https://maps-yiv5.onrender.com/api/stores/route?user_lat=18.9701&user_lon=72.8330&store_lat=18.9695&store_lon=72.8320&viz_type=simple
|
| 352 |
</pre>
|
| 353 |
|
| 354 |
<h2>Key Location Points in Dataset:</h2>
|
|
|
|
| 359 |
<li>Shopping Center: 18.9670, 72.8300</li>
|
| 360 |
<li>Commercial Street: 18.9690, 72.8340</li>
|
| 361 |
</ul>
|
|
|
|
|
|
|
|
|
|
| 362 |
</body>
|
| 363 |
</html>
|
| 364 |
"""
|
| 365 |
+
return HTMLResponse(content=html_content)
|
| 366 |
|
| 367 |
@app.get("/api/stores/nearby", response_model=StoresResponse, responses={400: {"model": ErrorResponse}})
|
| 368 |
async def get_nearby_stores(
|
| 369 |
lat: float = Query(..., description="Latitude of user location"),
|
| 370 |
lon: float = Query(..., description="Longitude of user location"),
|
| 371 |
+
radius: float = Query(5, description="Search radius in kilometers")
|
| 372 |
):
|
| 373 |
+
"""
|
| 374 |
+
Get nearby stores based on user location
|
| 375 |
+
"""
|
| 376 |
try:
|
| 377 |
nearby_stores = store_locator.find_nearby_stores(lat, lon, radius)
|
| 378 |
return {"status": "success", "stores": nearby_stores}
|
|
|
|
| 383 |
async def get_stores_map(
|
| 384 |
lat: float = Query(..., description="Latitude of center point"),
|
| 385 |
lon: float = Query(..., description="Longitude of center point"),
|
| 386 |
+
radius: float = Query(5, description="Search radius in kilometers")
|
| 387 |
):
|
| 388 |
+
"""
|
| 389 |
+
Get HTML map with store locations
|
| 390 |
+
"""
|
| 391 |
try:
|
|
|
|
|
|
|
|
|
|
| 392 |
store_map = store_locator.create_store_map(lat, lon, radius)
|
| 393 |
|
| 394 |
# Create complete HTML content
|
|
|
|
| 431 |
with open(file_path, 'w', encoding='utf-8') as f:
|
| 432 |
f.write(html_content)
|
| 433 |
|
| 434 |
+
# Return the file
|
| 435 |
+
return FileResponse(file_path, media_type='text/html')
|
| 436 |
|
| 437 |
except Exception as e:
|
| 438 |
raise HTTPException(status_code=400, detail=str(e))
|
| 439 |
+
|
| 440 |
@app.get("/api/stores/route", response_class=HTMLResponse, responses={400: {"model": ErrorResponse}, 404: {"model": ErrorResponse}})
|
| 441 |
async def get_store_route(
|
| 442 |
+
user_lat: float = Query(..., description="Latitude of user location"),
|
| 443 |
+
user_lon: float = Query(..., description="Longitude of user location"),
|
| 444 |
+
store_lat: float = Query(..., description="Latitude of store location"),
|
| 445 |
+
store_lon: float = Query(..., description="Longitude of store location")
|
|
|
|
| 446 |
):
|
| 447 |
+
"""
|
| 448 |
+
Get route between user location and store
|
| 449 |
+
"""
|
| 450 |
try:
|
|
|
|
|
|
|
|
|
|
| 451 |
# Initialize graph if not already initialized
|
|
|
|
| 452 |
if store_locator.network_graph is None:
|
| 453 |
+
store_locator.initialize_graph((user_lat, user_lon))
|
|
|
|
|
|
|
| 454 |
|
| 455 |
# Get nearest nodes
|
| 456 |
start_node = ox.distance.nearest_nodes(
|
|
|
|
| 459 |
store_locator.network_graph, store_lon, store_lat)
|
| 460 |
|
| 461 |
try:
|
| 462 |
+
# Calculate paths
|
| 463 |
path_time = nx.shortest_path(
|
| 464 |
store_locator.network_graph,
|
| 465 |
start_node,
|
|
|
|
| 467 |
weight='travel_time'
|
| 468 |
)
|
| 469 |
|
| 470 |
+
# Create animation data
|
| 471 |
+
lst_start, lst_end = [], []
|
| 472 |
+
start_x, start_y = [], []
|
| 473 |
+
end_x, end_y = [], []
|
| 474 |
+
lst_length, lst_time = [], []
|
| 475 |
+
|
| 476 |
+
for a, b in zip(path_time[:-1], path_time[1:]):
|
| 477 |
+
lst_start.append(a)
|
| 478 |
+
lst_end.append(b)
|
| 479 |
+
lst_length.append(round(store_locator.network_graph.edges[(a,b,0)]['length']))
|
| 480 |
+
lst_time.append(round(store_locator.network_graph.edges[(a,b,0)]['travel_time']))
|
| 481 |
+
start_x.append(store_locator.network_graph.nodes[a]['x'])
|
| 482 |
+
start_y.append(store_locator.network_graph.nodes[a]['y'])
|
| 483 |
+
end_x.append(store_locator.network_graph.nodes[b]['x'])
|
| 484 |
+
end_y.append(store_locator.network_graph.nodes[b]['y'])
|
| 485 |
+
|
| 486 |
+
df = pd.DataFrame(
|
| 487 |
+
list(zip(lst_start, lst_end, start_x, start_y, end_x, end_y,
|
| 488 |
+
lst_length, lst_time)),
|
| 489 |
+
columns=["start", "end", "start_x", "start_y",
|
| 490 |
+
"end_x", "end_y", "length", "travel_time"]
|
| 491 |
+
).reset_index().rename(columns={"index": "id"})
|
| 492 |
+
|
| 493 |
+
# Create animation using plotly
|
| 494 |
+
df_start = df[df["start"] == start_node]
|
| 495 |
+
df_end = df[df["end"] == end_node]
|
| 496 |
+
|
| 497 |
+
fig = px.scatter_mapbox(
|
| 498 |
+
data_frame=df,
|
| 499 |
+
lon="start_x",
|
| 500 |
+
lat="start_y",
|
| 501 |
+
zoom=15,
|
| 502 |
+
width=1000,
|
| 503 |
+
height=800,
|
| 504 |
+
animation_frame="id",
|
| 505 |
+
mapbox_style="carto-positron"
|
| 506 |
+
)
|
| 507 |
+
|
| 508 |
+
# Add driver marker
|
| 509 |
+
fig.data[0].marker = {"size": 12}
|
| 510 |
+
|
| 511 |
+
# Add start point
|
| 512 |
+
fig.add_trace(
|
| 513 |
+
px.scatter_mapbox(
|
| 514 |
+
data_frame=df_start,
|
| 515 |
+
lon="start_x",
|
| 516 |
+
lat="start_y"
|
| 517 |
+
).data[0]
|
| 518 |
+
)
|
| 519 |
+
fig.data[1].marker = {"size": 15, "color": "red"}
|
| 520 |
+
|
| 521 |
+
# Add end point
|
| 522 |
+
fig.add_trace(
|
| 523 |
+
px.scatter_mapbox(
|
| 524 |
+
data_frame=df_end,
|
| 525 |
+
lon="start_x",
|
| 526 |
+
lat="start_y"
|
| 527 |
+
).data[0]
|
| 528 |
+
)
|
| 529 |
+
fig.data[2].marker = {"size": 15, "color": "green"}
|
| 530 |
+
|
| 531 |
+
# Add route
|
| 532 |
+
fig.add_trace(
|
| 533 |
+
px.line_mapbox(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 534 |
data_frame=df,
|
| 535 |
lon="start_x",
|
| 536 |
+
lat="start_y"
|
| 537 |
+
).data[0]
|
| 538 |
+
)
|
| 539 |
+
|
| 540 |
+
# Update layout with slower animation settings
|
| 541 |
+
fig.update_layout(
|
| 542 |
+
showlegend=False,
|
| 543 |
+
margin={"r":0,"t":0,"l":0,"b":0},
|
| 544 |
+
autosize=True,
|
| 545 |
+
height=None,
|
| 546 |
+
updatemenus=[{
|
| 547 |
+
"type": "buttons",
|
| 548 |
+
"showactive": False,
|
| 549 |
+
"y": 0,
|
| 550 |
+
"x": 0,
|
| 551 |
+
"xanchor": "left",
|
| 552 |
+
"yanchor": "bottom",
|
| 553 |
+
"buttons": [
|
| 554 |
+
{
|
| 555 |
+
"label": "Play",
|
| 556 |
+
"method": "animate",
|
| 557 |
+
"args": [
|
| 558 |
+
None,
|
| 559 |
+
{
|
| 560 |
+
"frame": {"duration": 1000, "redraw": True},
|
| 561 |
+
"fromcurrent": True,
|
| 562 |
+
"transition": {"duration": 800}
|
| 563 |
+
}
|
| 564 |
+
]
|
| 565 |
+
},
|
| 566 |
+
{
|
| 567 |
+
"label": "Pause",
|
| 568 |
+
"method": "animate",
|
| 569 |
+
"args": [
|
| 570 |
+
[None],
|
| 571 |
+
{
|
| 572 |
+
"frame": {"duration": 0, "redraw": False},
|
| 573 |
+
"mode": "immediate",
|
| 574 |
+
"transition": {"duration": 0}
|
| 575 |
+
}
|
| 576 |
+
]
|
| 577 |
+
}
|
| 578 |
+
]
|
| 579 |
+
}],
|
| 580 |
+
sliders=[{
|
| 581 |
+
"currentvalue": {"prefix": "Step: "},
|
| 582 |
+
"pad": {"t": 20},
|
| 583 |
+
"len": 0.9,
|
| 584 |
+
"x": 0.1,
|
| 585 |
+
"xanchor": "left",
|
| 586 |
+
"y": 0.02,
|
| 587 |
+
"yanchor": "bottom",
|
| 588 |
+
"steps": [
|
| 589 |
+
{
|
| 590 |
+
"args": [
|
| 591 |
+
[k],
|
| 592 |
+
{
|
| 593 |
+
"frame": {"duration": 1000, "redraw": True},
|
| 594 |
+
"transition": {"duration": 500},
|
| 595 |
+
"mode": "immediate"
|
| 596 |
+
}
|
| 597 |
+
],
|
| 598 |
+
"label": str(k),
|
| 599 |
+
"method": "animate"
|
| 600 |
+
}
|
| 601 |
+
for k in range(len(df))
|
| 602 |
+
]
|
| 603 |
+
}]
|
| 604 |
+
)
|
| 605 |
+
|
| 606 |
+
# Create complete HTML content
|
| 607 |
+
html_content = f"""
|
| 608 |
+
<!DOCTYPE html>
|
| 609 |
+
<html>
|
| 610 |
+
<head>
|
| 611 |
+
<meta charset="utf-8">
|
| 612 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no" />
|
| 613 |
+
<title>Route Map</title>
|
| 614 |
+
<style>
|
| 615 |
+
body {{
|
| 616 |
+
margin: 0;
|
| 617 |
+
padding: 0;
|
| 618 |
+
width: 100vw;
|
| 619 |
+
height: 100vh;
|
| 620 |
+
overflow: hidden;
|
| 621 |
+
}}
|
| 622 |
+
#map-container {{
|
| 623 |
+
width: 100%;
|
| 624 |
+
height: 100%;
|
| 625 |
+
}}
|
| 626 |
+
</style>
|
| 627 |
+
</head>
|
| 628 |
+
<body>
|
| 629 |
+
<div id="map-container">
|
| 630 |
+
{fig.to_html(include_plotlyjs=True, full_html=False)}
|
| 631 |
+
</div>
|
| 632 |
+
<script>
|
| 633 |
+
window.onload = function() {{
|
| 634 |
+
setTimeout(function() {{
|
| 635 |
+
window.dispatchEvent(new Event('resize'));
|
| 636 |
+
}}, 1000);
|
| 637 |
+
}};
|
| 638 |
+
</script>
|
| 639 |
+
</body>
|
| 640 |
+
</html>
|
| 641 |
+
"""
|
| 642 |
+
|
| 643 |
+
# Save the HTML to a file
|
| 644 |
+
file_path = 'temp/route_map.html'
|
| 645 |
+
with open(file_path, 'w', encoding='utf-8') as f:
|
| 646 |
+
f.write(html_content)
|
| 647 |
+
|
| 648 |
+
# Return the file
|
| 649 |
+
return FileResponse(file_path, media_type='text/html')
|
| 650 |
|
| 651 |
except nx.NetworkXNoPath:
|
| 652 |
raise HTTPException(status_code=404, detail="No route found")
|
| 653 |
|
|
|
|
|
|
|
| 654 |
except Exception as e:
|
| 655 |
raise HTTPException(status_code=400, detail=str(e))
|
| 656 |
+
|
| 657 |
@app.get("/api/stores/locations", response_class=HTMLResponse, responses={400: {"model": ErrorResponse}})
|
| 658 |
async def get_all_store_locations(
|
| 659 |
lat: float = Query(..., description="Latitude of center point"),
|
| 660 |
lon: float = Query(..., description="Longitude of center point"),
|
| 661 |
+
radius: float = Query(10, description="Search radius in kilometers")
|
| 662 |
):
|
| 663 |
+
"""
|
| 664 |
+
Get a map showing all stores in the given radius with colors based on distance
|
| 665 |
+
"""
|
| 666 |
try:
|
|
|
|
|
|
|
|
|
|
| 667 |
# Get nearby stores
|
| 668 |
nearby_stores = store_locator.find_nearby_stores(lat, lon, radius)
|
| 669 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 670 |
# Create base map centered on user location
|
| 671 |
m = folium.Map(
|
| 672 |
location=[lat, lon],
|
|
|
|
| 691 |
else:
|
| 692 |
color = 'blue' # Further away
|
| 693 |
|
| 694 |
+
# Create detailed popup content with mobile-friendly styling
|
| 695 |
popup_content = f"""
|
| 696 |
<div style='width: 200px; font-size: 14px;'>
|
| 697 |
<h4 style='color: {color}; margin: 0 0 8px 0;'>{store['store_name']}</h4>
|
| 698 |
+
<b>Address:</b> {store['address']}<br>
|
| 699 |
<b>Distance:</b> {store['distance']} km<br>
|
| 700 |
<b>Est. Delivery:</b> {store['estimated_delivery_time']} mins<br>
|
| 701 |
+
<b>Contact:</b> {store['contact']}<br>
|
| 702 |
<b>Categories:</b> {store['product_categories']}<br>
|
| 703 |
+
<button onclick="window.location.href='/api/stores/route?user_lat={lat}&user_lon={lon}&store_lat={store['location']['lat']}&store_lon={store['location']['lon']}'"
|
| 704 |
style='margin-top: 8px; padding: 8px; width: 100%; background-color: #007bff; color: white; border: none; border-radius: 4px;'>
|
| 705 |
Get Route
|
| 706 |
</button>
|
|
|
|
| 715 |
tooltip=f"{store['store_name']} ({store['distance']} km)"
|
| 716 |
).add_to(m)
|
| 717 |
|
| 718 |
+
# Add circle to show distance
|
| 719 |
+
folium.Circle(
|
| 720 |
+
location=[store['location']['lat'], store['location']['lon']],
|
| 721 |
+
radius=store['distance'] * 100,
|
| 722 |
+
color=color,
|
| 723 |
+
fill=True,
|
| 724 |
+
opacity=0.1
|
| 725 |
+
).add_to(m)
|
|
|
|
| 726 |
|
| 727 |
+
# Add distance circles from user location
|
| 728 |
+
for radius_circle, color in [(2000, 'red'), (5000, 'orange'), (radius * 1000, 'blue')]:
|
| 729 |
folium.Circle(
|
| 730 |
location=[lat, lon],
|
| 731 |
+
radius=radius_circle,
|
| 732 |
color=color,
|
| 733 |
fill=False,
|
| 734 |
+
weight=1,
|
| 735 |
+
dash_array='5, 5'
|
| 736 |
).add_to(m)
|
| 737 |
|
| 738 |
# Create mobile-friendly HTML content
|
|
|
|
| 807 |
with open(file_path, 'w', encoding='utf-8') as f:
|
| 808 |
f.write(html_content)
|
| 809 |
|
| 810 |
+
return FileResponse(file_path, media_type='text/html')
|
|
|
|
| 811 |
|
| 812 |
except Exception as e:
|
| 813 |
raise HTTPException(status_code=400, detail=str(e))
|
| 814 |
|
| 815 |
+
# Add swagger UI customization
|
| 816 |
+
@app.on_event("startup")
|
| 817 |
+
async def startup_event():
|
| 818 |
+
app.title = "Store Locator API"
|
| 819 |
+
app.description = "API for locating nearby stores and generating routes"
|
| 820 |
+
app.version = "1.0.0"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 821 |
|
| 822 |
+
# Entry point for running the application
|
| 823 |
if __name__ == "__main__":
|
| 824 |
import uvicorn
|
| 825 |
uvicorn.run(app, host="0.0.0.0", port=8000)
|
dapp.py
ADDED
|
@@ -0,0 +1,889 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from fastapi import FastAPI, Request, Query, HTTPException, Response
|
| 2 |
+
from fastapi.responses import JSONResponse, HTMLResponse, FileResponse
|
| 3 |
+
from fastapi.middleware.cors import CORSMiddleware
|
| 4 |
+
import pandas as pd
|
| 5 |
+
import numpy as np
|
| 6 |
+
from geopy.distance import geodesic
|
| 7 |
+
import folium
|
| 8 |
+
from folium import plugins
|
| 9 |
+
import osmnx as ox
|
| 10 |
+
import networkx as nx
|
| 11 |
+
from datetime import datetime
|
| 12 |
+
import json
|
| 13 |
+
import matplotlib.pyplot as plt
|
| 14 |
+
import plotly.express as px
|
| 15 |
+
import os
|
| 16 |
+
import time
|
| 17 |
+
from functools import lru_cache
|
| 18 |
+
from rtree import index
|
| 19 |
+
import gc
|
| 20 |
+
import shutil
|
| 21 |
+
from typing import Optional, List, Dict, Any, Union
|
| 22 |
+
from pydantic import BaseModel, Field
|
| 23 |
+
|
| 24 |
+
app = FastAPI(title="falcao-maps API", description="Store locator and route planning API")
|
| 25 |
+
|
| 26 |
+
# Add CORS middleware
|
| 27 |
+
app.add_middleware(
|
| 28 |
+
CORSMiddleware,
|
| 29 |
+
allow_origins=["*"],
|
| 30 |
+
allow_credentials=True,
|
| 31 |
+
allow_methods=["*"],
|
| 32 |
+
allow_headers=["*"],
|
| 33 |
+
)
|
| 34 |
+
|
| 35 |
+
# Create temp directory for files
|
| 36 |
+
os.makedirs('temp', exist_ok=True)
|
| 37 |
+
|
| 38 |
+
# Custom JSON encoder for NumPy types
|
| 39 |
+
class NumpyEncoder(json.JSONEncoder):
|
| 40 |
+
def default(self, obj):
|
| 41 |
+
if isinstance(obj, np.integer):
|
| 42 |
+
return int(obj)
|
| 43 |
+
elif isinstance(obj, np.floating):
|
| 44 |
+
return float(obj)
|
| 45 |
+
elif isinstance(obj, np.ndarray):
|
| 46 |
+
return obj.tolist()
|
| 47 |
+
return super(NumpyEncoder, self).default(obj)
|
| 48 |
+
|
| 49 |
+
# Load and prepare the store data
|
| 50 |
+
stores_df = pd.read_csv('dataset of 50 stores.csv')
|
| 51 |
+
|
| 52 |
+
# Define Pydantic models for API responses
|
| 53 |
+
class Location(BaseModel):
|
| 54 |
+
lat: float
|
| 55 |
+
lon: float
|
| 56 |
+
|
| 57 |
+
class Store(BaseModel):
|
| 58 |
+
store_name: str
|
| 59 |
+
address: str
|
| 60 |
+
contact: str
|
| 61 |
+
distance: float
|
| 62 |
+
estimated_delivery_time: int
|
| 63 |
+
product_categories: str
|
| 64 |
+
location: Location
|
| 65 |
+
|
| 66 |
+
class StoresResponse(BaseModel):
|
| 67 |
+
status: str
|
| 68 |
+
stores: List[Store]
|
| 69 |
+
|
| 70 |
+
class ErrorResponse(BaseModel):
|
| 71 |
+
status: str
|
| 72 |
+
message: str
|
| 73 |
+
|
| 74 |
+
class StoreLocator:
|
| 75 |
+
def __init__(self, stores_dataframe):
|
| 76 |
+
self.stores_df = stores_dataframe
|
| 77 |
+
self.network_graph = None
|
| 78 |
+
self.graph_cache = {} # Cache for network graphs
|
| 79 |
+
self.spatial_index = self._build_spatial_index()
|
| 80 |
+
|
| 81 |
+
@lru_cache(maxsize=50)
|
| 82 |
+
def initialize_graph(self, center_point, dist=30000): # Reduced distance for memory optimization
|
| 83 |
+
"""Initialize road network graph with caching"""
|
| 84 |
+
cache_key = f"{center_point[0]}_{center_point[1]}"
|
| 85 |
+
if cache_key in self.graph_cache:
|
| 86 |
+
self.network_graph = self.graph_cache[cache_key]
|
| 87 |
+
return True
|
| 88 |
+
try:
|
| 89 |
+
# Use simplify=True and increased tolerance for lower memory usage
|
| 90 |
+
self.network_graph = ox.graph_from_point(
|
| 91 |
+
center_point,
|
| 92 |
+
dist=dist,
|
| 93 |
+
network_type="drive",
|
| 94 |
+
simplify=True,
|
| 95 |
+
retain_all=False
|
| 96 |
+
)
|
| 97 |
+
self.network_graph = ox.add_edge_speeds(self.network_graph)
|
| 98 |
+
self.network_graph = ox.add_edge_travel_times(self.network_graph)
|
| 99 |
+
|
| 100 |
+
# Store in cache
|
| 101 |
+
self.graph_cache[cache_key] = self.network_graph
|
| 102 |
+
|
| 103 |
+
# Force garbage collection
|
| 104 |
+
gc.collect()
|
| 105 |
+
|
| 106 |
+
return True
|
| 107 |
+
except Exception as e:
|
| 108 |
+
print(f"Error initializing graph: {str(e)}")
|
| 109 |
+
return False
|
| 110 |
+
|
| 111 |
+
def _build_spatial_index(self):
|
| 112 |
+
idx = index.Index()
|
| 113 |
+
for i, row in self.stores_df.iterrows():
|
| 114 |
+
idx.insert(i, (row['Latitude'], row['Longitude'],
|
| 115 |
+
row['Latitude'], row['Longitude']))
|
| 116 |
+
return idx
|
| 117 |
+
|
| 118 |
+
def calculate_distance(self, lat1, lon1, lat2, lon2):
|
| 119 |
+
"""Calculate direct distance between two points"""
|
| 120 |
+
return geodesic((lat1, lon1), (lat2, lon2)).kilometers
|
| 121 |
+
|
| 122 |
+
def estimate_delivery_time(self, distance, current_time=None):
|
| 123 |
+
"""Estimate delivery time based on distance and current time"""
|
| 124 |
+
if current_time is None:
|
| 125 |
+
current_time = datetime.now()
|
| 126 |
+
|
| 127 |
+
# Base time: 5 mins base + 2 mins per km
|
| 128 |
+
base_minutes = 5 + (distance * 2)
|
| 129 |
+
|
| 130 |
+
# Apply traffic multiplier based on time of day
|
| 131 |
+
hour = current_time.hour
|
| 132 |
+
if hour in [8, 9, 10, 17, 18, 19]: # Peak hours
|
| 133 |
+
multiplier = 1.5
|
| 134 |
+
elif hour in [23, 0, 1, 2, 3, 4]: # Off-peak hours
|
| 135 |
+
multiplier = 0.8
|
| 136 |
+
else: # Normal hours
|
| 137 |
+
multiplier = 1.0
|
| 138 |
+
|
| 139 |
+
return round(base_minutes * multiplier)
|
| 140 |
+
|
| 141 |
+
def find_nearby_stores(self, lat, lon, radius=5):
|
| 142 |
+
"""Find stores within radius using spatial index"""
|
| 143 |
+
nearby_stores = []
|
| 144 |
+
bbox = (lat - radius/111.0, lon - radius/111.0,
|
| 145 |
+
lat + radius/111.0, lon + radius/111.0)
|
| 146 |
+
|
| 147 |
+
for store_id in self.spatial_index.intersection(bbox):
|
| 148 |
+
store = self.stores_df.iloc[store_id]
|
| 149 |
+
distance = self.calculate_distance(lat, lon,
|
| 150 |
+
store['Latitude'],
|
| 151 |
+
store['Longitude'])
|
| 152 |
+
if distance <= radius:
|
| 153 |
+
delivery_time = self.estimate_delivery_time(distance)
|
| 154 |
+
nearby_stores.append({
|
| 155 |
+
'store_name': store['Store Name'],
|
| 156 |
+
'address': store['Address'],
|
| 157 |
+
'contact': str(store['Contact Number']), # Convert to string to avoid int64 issues
|
| 158 |
+
'distance': round(distance, 2),
|
| 159 |
+
'estimated_delivery_time': int(delivery_time), # Ensure integer type
|
| 160 |
+
'product_categories': store['Product Categories'],
|
| 161 |
+
'location': {
|
| 162 |
+
'lat': float(store['Latitude']), # Ensure float type
|
| 163 |
+
'lon': float(store['Longitude']) # Ensure float type
|
| 164 |
+
}
|
| 165 |
+
})
|
| 166 |
+
|
| 167 |
+
return sorted(nearby_stores, key=lambda x: x['distance'])
|
| 168 |
+
|
| 169 |
+
def create_store_map(self, center_lat, center_lon, radius=5):
|
| 170 |
+
"""Create an interactive map with store locations - optimized for memory"""
|
| 171 |
+
# Create base map
|
| 172 |
+
m = folium.Map(
|
| 173 |
+
location=[center_lat, center_lon],
|
| 174 |
+
zoom_start=13,
|
| 175 |
+
tiles="cartodbpositron"
|
| 176 |
+
)
|
| 177 |
+
|
| 178 |
+
# Create marker cluster for better performance with many markers
|
| 179 |
+
marker_cluster = plugins.MarkerCluster().add_to(m)
|
| 180 |
+
|
| 181 |
+
# Add stores to map
|
| 182 |
+
nearby_stores = self.find_nearby_stores(center_lat, center_lon, radius)
|
| 183 |
+
|
| 184 |
+
# Limit the number of stores to reduce memory usage
|
| 185 |
+
max_stores = min(len(nearby_stores), 50) # Cap at 50 stores
|
| 186 |
+
|
| 187 |
+
for store in nearby_stores[:max_stores]:
|
| 188 |
+
# Prepare popup content
|
| 189 |
+
popup_content = f"""
|
| 190 |
+
<div style='width: 200px'>
|
| 191 |
+
<b>{store['store_name']}</b><br>
|
| 192 |
+
Address: {store['address']}<br>
|
| 193 |
+
Distance: {store['distance']} km<br>
|
| 194 |
+
Est. Delivery: {store['estimated_delivery_time']} mins<br>
|
| 195 |
+
Categories: {store['product_categories']}
|
| 196 |
+
</div>
|
| 197 |
+
"""
|
| 198 |
+
|
| 199 |
+
# Add store marker
|
| 200 |
+
folium.Marker(
|
| 201 |
+
location=[store['location']['lat'], store['location']['lon']],
|
| 202 |
+
popup=folium.Popup(popup_content, max_width=300),
|
| 203 |
+
icon=folium.Icon(color='red', icon='info-sign')
|
| 204 |
+
).add_to(marker_cluster)
|
| 205 |
+
|
| 206 |
+
# Add line to show distance from center (only for closer stores)
|
| 207 |
+
if store['distance'] <= nearby_stores[min(9, len(nearby_stores)-1)]['distance']:
|
| 208 |
+
folium.PolyLine(
|
| 209 |
+
locations=[[center_lat, center_lon],
|
| 210 |
+
[store['location']['lat'], store['location']['lon']]],
|
| 211 |
+
weight=2,
|
| 212 |
+
color='blue',
|
| 213 |
+
opacity=0.3
|
| 214 |
+
).add_to(m)
|
| 215 |
+
|
| 216 |
+
# Add current location marker
|
| 217 |
+
folium.Marker(
|
| 218 |
+
location=[center_lat, center_lon],
|
| 219 |
+
popup='Your Location',
|
| 220 |
+
icon=folium.Icon(color='green', icon='home')
|
| 221 |
+
).add_to(m)
|
| 222 |
+
|
| 223 |
+
# Add layer control only
|
| 224 |
+
folium.LayerControl().add_to(m)
|
| 225 |
+
|
| 226 |
+
return m
|
| 227 |
+
|
| 228 |
+
# Initialize store locator
|
| 229 |
+
store_locator = StoreLocator(stores_df)
|
| 230 |
+
|
| 231 |
+
# Helper functions for cleaning temporary files
|
| 232 |
+
def cleanup_temp_files():
|
| 233 |
+
temp_dir = 'temp'
|
| 234 |
+
if os.path.exists(temp_dir):
|
| 235 |
+
for file in os.listdir(temp_dir):
|
| 236 |
+
file_path = os.path.join(temp_dir, file)
|
| 237 |
+
try:
|
| 238 |
+
if os.path.isfile(file_path) and file.endswith('.html'):
|
| 239 |
+
# Delete files older than 1 hour
|
| 240 |
+
if os.path.getmtime(file_path) < time.time() - 3600:
|
| 241 |
+
os.remove(file_path)
|
| 242 |
+
except Exception as e:
|
| 243 |
+
print(f"Error cleaning up temp files: {e}")
|
| 244 |
+
|
| 245 |
+
# Register cleanup on startup and shutdown
|
| 246 |
+
@app.on_event("startup")
|
| 247 |
+
async def startup_event():
|
| 248 |
+
cleanup_temp_files()
|
| 249 |
+
|
| 250 |
+
@app.on_event("shutdown")
|
| 251 |
+
async def shutdown_event():
|
| 252 |
+
# Clean up all temporary files on shutdown
|
| 253 |
+
try:
|
| 254 |
+
shutil.rmtree('temp')
|
| 255 |
+
except Exception as e:
|
| 256 |
+
print(f"Error cleaning up temp directory: {e}")
|
| 257 |
+
|
| 258 |
+
# Routes
|
| 259 |
+
@app.get("/", response_class=HTMLResponse)
|
| 260 |
+
async def home():
|
| 261 |
+
"""API Documentation Homepage"""
|
| 262 |
+
html_content = f"""
|
| 263 |
+
<!DOCTYPE html>
|
| 264 |
+
<html lang="en">
|
| 265 |
+
<head>
|
| 266 |
+
<meta charset="UTF-8">
|
| 267 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
| 268 |
+
<title>falcao-maps API Documentation</title>
|
| 269 |
+
<style>
|
| 270 |
+
body {{
|
| 271 |
+
font-family: Arial, sans-serif;
|
| 272 |
+
margin: 20px;
|
| 273 |
+
}}
|
| 274 |
+
h1, h2 {{
|
| 275 |
+
color: #333;
|
| 276 |
+
}}
|
| 277 |
+
pre {{
|
| 278 |
+
background-color: #f4f4f4;
|
| 279 |
+
padding: 10px;
|
| 280 |
+
border: 1px solid #ddd;
|
| 281 |
+
border-radius: 5px;
|
| 282 |
+
}}
|
| 283 |
+
</style>
|
| 284 |
+
</head>
|
| 285 |
+
<body>
|
| 286 |
+
<h1>Welcome to falcao-maps</h1>
|
| 287 |
+
<p>Based on your uploaded dataset and deployed API, here are example API calls for your client:</p>
|
| 288 |
+
|
| 289 |
+
<h2>1. Find Nearby Stores (JSON Response)</h2>
|
| 290 |
+
<pre>
|
| 291 |
+
/api/stores/nearby?lat=18.9695&lon=72.8320&radius=1
|
| 292 |
+
</pre>
|
| 293 |
+
<p>Use this to get store details near Market Road area within 1km</p>
|
| 294 |
+
|
| 295 |
+
<h2>2. View Basic Store Map</h2>
|
| 296 |
+
<pre>
|
| 297 |
+
/api/stores/map?lat=18.9701&lon=72.8330&radius=0.5
|
| 298 |
+
</pre>
|
| 299 |
+
<p>Shows map centered at Main Street with 500m radius</p>
|
| 300 |
+
|
| 301 |
+
<h2>3. View All Store Locations with Color Coding</h2>
|
| 302 |
+
<pre>
|
| 303 |
+
/api/stores/locations?lat=18.9685&lon=72.8325&radius=2
|
| 304 |
+
</pre>
|
| 305 |
+
<p>Shows detailed map with color-coded stores within 2km</p>
|
| 306 |
+
|
| 307 |
+
<h2>4. Get Route Between Points</h2>
|
| 308 |
+
<p>Example routes:</p>
|
| 309 |
+
<pre>
|
| 310 |
+
# Route from Park Avenue to Hill Road stores (use simple visualization for memory optimization)
|
| 311 |
+
/api/stores/route?user_lat=18.9710&user_lon=72.8335&store_lat=18.9705&store_lon=72.8345&viz_type=simple
|
| 312 |
+
|
| 313 |
+
# Route from Main Street to Market Road stores
|
| 314 |
+
/api/stores/route?user_lat=18.9701&user_lon=72.8330&store_lat=18.9695&store_lon=72.8320&viz_type=simple
|
| 315 |
+
</pre>
|
| 316 |
+
|
| 317 |
+
<h2>Key Location Points in Dataset:</h2>
|
| 318 |
+
<ul>
|
| 319 |
+
<li>Main Street Area: 18.9701, 72.8330</li>
|
| 320 |
+
<li>Park Avenue: 18.9710, 72.8335</li>
|
| 321 |
+
<li>Market Road: 18.9695, 72.8320</li>
|
| 322 |
+
<li>Shopping Center: 18.9670, 72.8300</li>
|
| 323 |
+
<li>Commercial Street: 18.9690, 72.8340</li>
|
| 324 |
+
</ul>
|
| 325 |
+
|
| 326 |
+
<h2>API Documentation</h2>
|
| 327 |
+
<p>You can view the interactive API documentation at: <a href="/docs">/docs</a></p>
|
| 328 |
+
</body>
|
| 329 |
+
</html>
|
| 330 |
+
"""
|
| 331 |
+
return html_content
|
| 332 |
+
|
| 333 |
+
@app.get("/api/stores/nearby", response_model=StoresResponse, responses={400: {"model": ErrorResponse}})
|
| 334 |
+
async def get_nearby_stores(
|
| 335 |
+
lat: float = Query(..., description="Latitude of user location"),
|
| 336 |
+
lon: float = Query(..., description="Longitude of user location"),
|
| 337 |
+
radius: float = Query(5.0, description="Search radius in kilometers")
|
| 338 |
+
):
|
| 339 |
+
"""Get nearby stores based on user location"""
|
| 340 |
+
try:
|
| 341 |
+
nearby_stores = store_locator.find_nearby_stores(lat, lon, radius)
|
| 342 |
+
return {"status": "success", "stores": nearby_stores}
|
| 343 |
+
except Exception as e:
|
| 344 |
+
raise HTTPException(status_code=400, detail=str(e))
|
| 345 |
+
|
| 346 |
+
@app.get("/api/stores/map", response_class=HTMLResponse, responses={400: {"model": ErrorResponse}})
|
| 347 |
+
async def get_stores_map(
|
| 348 |
+
lat: float = Query(..., description="Latitude of center point"),
|
| 349 |
+
lon: float = Query(..., description="Longitude of center point"),
|
| 350 |
+
radius: float = Query(5.0, description="Search radius in kilometers")
|
| 351 |
+
):
|
| 352 |
+
"""Get HTML map with store locations"""
|
| 353 |
+
try:
|
| 354 |
+
# Clean up temp files before creating new ones
|
| 355 |
+
cleanup_temp_files()
|
| 356 |
+
|
| 357 |
+
store_map = store_locator.create_store_map(lat, lon, radius)
|
| 358 |
+
|
| 359 |
+
# Create complete HTML content
|
| 360 |
+
html_content = f"""
|
| 361 |
+
<!DOCTYPE html>
|
| 362 |
+
<html>
|
| 363 |
+
<head>
|
| 364 |
+
<meta charset="utf-8">
|
| 365 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no" />
|
| 366 |
+
<title>Stores Map</title>
|
| 367 |
+
<style>
|
| 368 |
+
body {{
|
| 369 |
+
margin: 0;
|
| 370 |
+
padding: 0;
|
| 371 |
+
width: 100vw;
|
| 372 |
+
height: 100vh;
|
| 373 |
+
overflow: hidden;
|
| 374 |
+
}}
|
| 375 |
+
#map {{
|
| 376 |
+
width: 100%;
|
| 377 |
+
height: 100%;
|
| 378 |
+
}}
|
| 379 |
+
</style>
|
| 380 |
+
</head>
|
| 381 |
+
<body>
|
| 382 |
+
{store_map.get_root().render()}
|
| 383 |
+
<script>
|
| 384 |
+
window.onload = function() {{
|
| 385 |
+
setTimeout(function() {{
|
| 386 |
+
window.dispatchEvent(new Event('resize'));
|
| 387 |
+
}}, 1000);
|
| 388 |
+
}};
|
| 389 |
+
</script>
|
| 390 |
+
</body>
|
| 391 |
+
</html>
|
| 392 |
+
"""
|
| 393 |
+
|
| 394 |
+
# Save the HTML to a file
|
| 395 |
+
file_path = 'temp/stores_map.html'
|
| 396 |
+
with open(file_path, 'w', encoding='utf-8') as f:
|
| 397 |
+
f.write(html_content)
|
| 398 |
+
|
| 399 |
+
# Return the file as HTML response
|
| 400 |
+
return html_content
|
| 401 |
+
|
| 402 |
+
except Exception as e:
|
| 403 |
+
raise HTTPException(status_code=400, detail=str(e))
|
| 404 |
+
|
| 405 |
+
@app.get("/api/stores/route", response_class=HTMLResponse, responses={400: {"model": ErrorResponse}, 404: {"model": ErrorResponse}})
|
| 406 |
+
async def get_store_route(
|
| 407 |
+
user_lat: float = Query(..., description="User location latitude"),
|
| 408 |
+
user_lon: float = Query(..., description="User location longitude"),
|
| 409 |
+
store_lat: float = Query(..., description="Store location latitude"),
|
| 410 |
+
store_lon: float = Query(..., description="Store location longitude"),
|
| 411 |
+
viz_type: str = Query("simple", description="Visualization type (simple or advanced)")
|
| 412 |
+
):
|
| 413 |
+
"""Get route between user and store locations with visualization"""
|
| 414 |
+
try:
|
| 415 |
+
# Clean up temp files before creating new ones
|
| 416 |
+
cleanup_temp_files()
|
| 417 |
+
|
| 418 |
+
# Initialize graph if not already initialized
|
| 419 |
+
# Use a smaller distance to reduce memory usage
|
| 420 |
+
if store_locator.network_graph is None:
|
| 421 |
+
success = store_locator.initialize_graph((user_lat, user_lon), dist=10000)
|
| 422 |
+
if not success:
|
| 423 |
+
raise HTTPException(status_code=400, detail="Unable to initialize graph, try a different location")
|
| 424 |
+
|
| 425 |
+
# Get nearest nodes
|
| 426 |
+
start_node = ox.distance.nearest_nodes(
|
| 427 |
+
store_locator.network_graph, user_lon, user_lat)
|
| 428 |
+
end_node = ox.distance.nearest_nodes(
|
| 429 |
+
store_locator.network_graph, store_lon, store_lat)
|
| 430 |
+
|
| 431 |
+
try:
|
| 432 |
+
# Calculate path using the travel_time weight
|
| 433 |
+
path_time = nx.shortest_path(
|
| 434 |
+
store_locator.network_graph,
|
| 435 |
+
start_node,
|
| 436 |
+
end_node,
|
| 437 |
+
weight='travel_time'
|
| 438 |
+
)
|
| 439 |
+
|
| 440 |
+
if viz_type == "simple":
|
| 441 |
+
# Create a simple folium map for low-resource environments
|
| 442 |
+
m = folium.Map(
|
| 443 |
+
location=[(user_lat + store_lat) / 2, (user_lon + store_lon) / 2],
|
| 444 |
+
zoom_start=15,
|
| 445 |
+
tiles="cartodbpositron"
|
| 446 |
+
)
|
| 447 |
+
|
| 448 |
+
# Add markers for start and end points
|
| 449 |
+
folium.Marker(
|
| 450 |
+
[user_lat, user_lon],
|
| 451 |
+
popup='Your Location',
|
| 452 |
+
icon=folium.Icon(color='green', icon='home')
|
| 453 |
+
).add_to(m)
|
| 454 |
+
|
| 455 |
+
folium.Marker(
|
| 456 |
+
[store_lat, store_lon],
|
| 457 |
+
popup='Store Location',
|
| 458 |
+
icon=folium.Icon(color='red', icon='info-sign')
|
| 459 |
+
).add_to(m)
|
| 460 |
+
|
| 461 |
+
# Extract coordinates from the path
|
| 462 |
+
path_coords = []
|
| 463 |
+
for node in path_time:
|
| 464 |
+
x = store_locator.network_graph.nodes[node]['x']
|
| 465 |
+
y = store_locator.network_graph.nodes[node]['y']
|
| 466 |
+
path_coords.append([y, x]) # Note the y, x order for folium
|
| 467 |
+
|
| 468 |
+
# Add the route line
|
| 469 |
+
folium.PolyLine(
|
| 470 |
+
locations=path_coords,
|
| 471 |
+
weight=5,
|
| 472 |
+
color='blue',
|
| 473 |
+
opacity=0.7
|
| 474 |
+
).add_to(m)
|
| 475 |
+
|
| 476 |
+
# Add distance and time estimate
|
| 477 |
+
total_distance = 0
|
| 478 |
+
total_time = 0
|
| 479 |
+
|
| 480 |
+
for i in range(len(path_time) - 1):
|
| 481 |
+
a, b = path_time[i], path_time[i + 1]
|
| 482 |
+
total_distance += store_locator.network_graph.edges[(a, b, 0)]['length']
|
| 483 |
+
total_time += store_locator.network_graph.edges[(a, b, 0)]['travel_time']
|
| 484 |
+
|
| 485 |
+
# Convert to km and minutes
|
| 486 |
+
total_distance_km = round(total_distance / 1000, 2)
|
| 487 |
+
total_time_min = round(total_time / 60, 1)
|
| 488 |
+
|
| 489 |
+
# Add info box
|
| 490 |
+
html_content = f"""
|
| 491 |
+
<div style="position: fixed; top: 10px; left: 50px; z-index: 9999;
|
| 492 |
+
background-color: white; padding: 10px; border-radius: 5px;
|
| 493 |
+
box-shadow: 0 0 10px rgba(0,0,0,0.3);">
|
| 494 |
+
<h4 style="margin: 0 0 5px 0;">Route Information</h4>
|
| 495 |
+
<p><b>Distance:</b> {total_distance_km} km<br>
|
| 496 |
+
<b>Est. Time:</b> {total_time_min} minutes</p>
|
| 497 |
+
</div>
|
| 498 |
+
"""
|
| 499 |
+
|
| 500 |
+
m.get_root().html.add_child(folium.Element(html_content))
|
| 501 |
+
|
| 502 |
+
# Add layer control
|
| 503 |
+
folium.LayerControl().add_to(m)
|
| 504 |
+
|
| 505 |
+
# Create complete HTML content
|
| 506 |
+
html_content = f"""
|
| 507 |
+
<!DOCTYPE html>
|
| 508 |
+
<html>
|
| 509 |
+
<head>
|
| 510 |
+
<meta charset="utf-8">
|
| 511 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no" />
|
| 512 |
+
<title>Simple Route Map</title>
|
| 513 |
+
<style>
|
| 514 |
+
body {{
|
| 515 |
+
margin: 0;
|
| 516 |
+
padding: 0;
|
| 517 |
+
width: 100vw;
|
| 518 |
+
height: 100vh;
|
| 519 |
+
overflow: hidden;
|
| 520 |
+
}}
|
| 521 |
+
#map {{
|
| 522 |
+
width: 100%;
|
| 523 |
+
height: 100%;
|
| 524 |
+
}}
|
| 525 |
+
</style>
|
| 526 |
+
</head>
|
| 527 |
+
<body>
|
| 528 |
+
{m.get_root().render()}
|
| 529 |
+
<script>
|
| 530 |
+
window.onload = function() {{
|
| 531 |
+
setTimeout(function() {{
|
| 532 |
+
window.dispatchEvent(new Event('resize'));
|
| 533 |
+
}}, 1000);
|
| 534 |
+
}};
|
| 535 |
+
</script>
|
| 536 |
+
</body>
|
| 537 |
+
</html>
|
| 538 |
+
"""
|
| 539 |
+
|
| 540 |
+
# Save the HTML to a file
|
| 541 |
+
file_path = 'temp/simple_route_map.html'
|
| 542 |
+
with open(file_path, 'w', encoding='utf-8') as f:
|
| 543 |
+
f.write(html_content)
|
| 544 |
+
|
| 545 |
+
# Return the HTML content
|
| 546 |
+
return html_content
|
| 547 |
+
|
| 548 |
+
else:
|
| 549 |
+
# WARNING: Advanced visualization - may cause memory issues on limited resources
|
| 550 |
+
|
| 551 |
+
# Limit the path nodes to reduce memory usage
|
| 552 |
+
# Only include every Nth node
|
| 553 |
+
step = max(1, len(path_time) // 30) # Maximum 30 points
|
| 554 |
+
simplified_path = path_time[::step]
|
| 555 |
+
if path_time[-1] not in simplified_path:
|
| 556 |
+
simplified_path.append(path_time[-1])
|
| 557 |
+
|
| 558 |
+
# Create animation data (simplified)
|
| 559 |
+
lst_start, lst_end = [], []
|
| 560 |
+
start_x, start_y = [], []
|
| 561 |
+
end_x, end_y = [], []
|
| 562 |
+
lst_length, lst_time = [], []
|
| 563 |
+
|
| 564 |
+
for a, b in zip(simplified_path[:-1], simplified_path[1:]):
|
| 565 |
+
lst_start.append(a)
|
| 566 |
+
lst_end.append(b)
|
| 567 |
+
|
| 568 |
+
# Calculate accumulated length and time between simplified points
|
| 569 |
+
segment_length = 0
|
| 570 |
+
segment_time = 0
|
| 571 |
+
path_segment = nx.shortest_path(
|
| 572 |
+
store_locator.network_graph, a, b, weight='travel_time')
|
| 573 |
+
|
| 574 |
+
for i in range(len(path_segment) - 1):
|
| 575 |
+
u, v = path_segment[i], path_segment[i + 1]
|
| 576 |
+
segment_length += store_locator.network_graph.edges[(u, v, 0)]['length']
|
| 577 |
+
segment_time += store_locator.network_graph.edges[(u, v, 0)]['travel_time']
|
| 578 |
+
|
| 579 |
+
lst_length.append(round(segment_length))
|
| 580 |
+
lst_time.append(round(segment_time))
|
| 581 |
+
start_x.append(store_locator.network_graph.nodes[a]['x'])
|
| 582 |
+
start_y.append(store_locator.network_graph.nodes[a]['y'])
|
| 583 |
+
end_x.append(store_locator.network_graph.nodes[b]['x'])
|
| 584 |
+
end_y.append(store_locator.network_graph.nodes[b]['y'])
|
| 585 |
+
|
| 586 |
+
df = pd.DataFrame(
|
| 587 |
+
list(zip(lst_start, lst_end, start_x, start_y, end_x, end_y,
|
| 588 |
+
lst_length, lst_time)),
|
| 589 |
+
columns=["start", "end", "start_x", "start_y",
|
| 590 |
+
"end_x", "end_y", "length", "travel_time"]
|
| 591 |
+
).reset_index().rename(columns={"index": "id"})
|
| 592 |
+
|
| 593 |
+
# Create animation using plotly (reduced complexity)
|
| 594 |
+
df_start = df[df["start"] == lst_start[0]]
|
| 595 |
+
df_end = df[df["end"] == lst_end[-1]]
|
| 596 |
+
|
| 597 |
+
fig = px.scatter_mapbox(
|
| 598 |
+
data_frame=df,
|
| 599 |
+
lon="start_x",
|
| 600 |
+
lat="start_y",
|
| 601 |
+
zoom=15,
|
| 602 |
+
width=800, # Reduced size
|
| 603 |
+
height=600, # Reduced size
|
| 604 |
+
animation_frame="id",
|
| 605 |
+
mapbox_style="carto-positron"
|
| 606 |
+
)
|
| 607 |
+
|
| 608 |
+
# Basic visualization elements only
|
| 609 |
+
fig.data[0].marker = {"size": 12}
|
| 610 |
+
|
| 611 |
+
# Add start point
|
| 612 |
+
fig.add_trace(
|
| 613 |
+
px.scatter_mapbox(
|
| 614 |
+
data_frame=df_start,
|
| 615 |
+
lon="start_x",
|
| 616 |
+
lat="start_y"
|
| 617 |
+
).data[0]
|
| 618 |
+
)
|
| 619 |
+
fig.data[1].marker = {"size": 15, "color": "red"}
|
| 620 |
+
|
| 621 |
+
# Add end point
|
| 622 |
+
fig.add_trace(
|
| 623 |
+
px.scatter_mapbox(
|
| 624 |
+
data_frame=df_end,
|
| 625 |
+
lon="start_x",
|
| 626 |
+
lat="start_y"
|
| 627 |
+
).data[0]
|
| 628 |
+
)
|
| 629 |
+
fig.data[2].marker = {"size": 15, "color": "green"}
|
| 630 |
+
|
| 631 |
+
# Add route
|
| 632 |
+
fig.add_trace(
|
| 633 |
+
px.line_mapbox(
|
| 634 |
+
data_frame=df,
|
| 635 |
+
lon="start_x",
|
| 636 |
+
lat="start_y"
|
| 637 |
+
).data[0]
|
| 638 |
+
)
|
| 639 |
+
|
| 640 |
+
# Simplified layout with fewer options to reduce complexity
|
| 641 |
+
fig.update_layout(
|
| 642 |
+
showlegend=False,
|
| 643 |
+
margin={"r":0,"t":0,"l":0,"b":0},
|
| 644 |
+
autosize=True,
|
| 645 |
+
height=None
|
| 646 |
+
)
|
| 647 |
+
|
| 648 |
+
# Create complete HTML content
|
| 649 |
+
html_content = f"""
|
| 650 |
+
<!DOCTYPE html>
|
| 651 |
+
<html>
|
| 652 |
+
<head>
|
| 653 |
+
<meta charset="utf-8">
|
| 654 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no" />
|
| 655 |
+
<title>Route Map</title>
|
| 656 |
+
<style>
|
| 657 |
+
body {{
|
| 658 |
+
margin: 0;
|
| 659 |
+
padding: 0;
|
| 660 |
+
width: 100vw;
|
| 661 |
+
height: 100vh;
|
| 662 |
+
overflow: hidden;
|
| 663 |
+
}}
|
| 664 |
+
#map-container {{
|
| 665 |
+
width: 100%;
|
| 666 |
+
height: 100%;
|
| 667 |
+
}}
|
| 668 |
+
</style>
|
| 669 |
+
</head>
|
| 670 |
+
<body>
|
| 671 |
+
<div id="map-container">
|
| 672 |
+
{fig.to_html(include_plotlyjs=True, full_html=False, config={'staticPlot': True})}
|
| 673 |
+
</div>
|
| 674 |
+
<script>
|
| 675 |
+
window.onload = function() {{
|
| 676 |
+
setTimeout(function() {{
|
| 677 |
+
window.dispatchEvent(new Event('resize'));
|
| 678 |
+
}}, 1000);
|
| 679 |
+
}};
|
| 680 |
+
</script>
|
| 681 |
+
</body>
|
| 682 |
+
</html>
|
| 683 |
+
"""
|
| 684 |
+
|
| 685 |
+
# Save the HTML to a file
|
| 686 |
+
file_path = 'temp/route_map.html'
|
| 687 |
+
with open(file_path, 'w', encoding='utf-8') as f:
|
| 688 |
+
f.write(html_content)
|
| 689 |
+
|
| 690 |
+
# Return the HTML content
|
| 691 |
+
return html_content
|
| 692 |
+
|
| 693 |
+
except nx.NetworkXNoPath:
|
| 694 |
+
raise HTTPException(status_code=404, detail="No route found")
|
| 695 |
+
|
| 696 |
+
except HTTPException:
|
| 697 |
+
raise
|
| 698 |
+
except Exception as e:
|
| 699 |
+
raise HTTPException(status_code=400, detail=str(e))
|
| 700 |
+
|
| 701 |
+
@app.get("/api/stores/locations", response_class=HTMLResponse, responses={400: {"model": ErrorResponse}})
|
| 702 |
+
async def get_all_store_locations(
|
| 703 |
+
lat: float = Query(..., description="Latitude of center point"),
|
| 704 |
+
lon: float = Query(..., description="Longitude of center point"),
|
| 705 |
+
radius: float = Query(10.0, description="Search radius in kilometers")
|
| 706 |
+
):
|
| 707 |
+
"""Get a map showing all stores in the given radius with colors based on distance"""
|
| 708 |
+
try:
|
| 709 |
+
# Clean up temp files before creating new ones
|
| 710 |
+
cleanup_temp_files()
|
| 711 |
+
|
| 712 |
+
# Get nearby stores
|
| 713 |
+
nearby_stores = store_locator.find_nearby_stores(lat, lon, radius)
|
| 714 |
+
|
| 715 |
+
# Limit number of stores for memory optimization
|
| 716 |
+
max_stores = min(len(nearby_stores), 50)
|
| 717 |
+
nearby_stores = nearby_stores[:max_stores]
|
| 718 |
+
|
| 719 |
+
# Create base map centered on user location
|
| 720 |
+
m = folium.Map(
|
| 721 |
+
location=[lat, lon],
|
| 722 |
+
zoom_start=12,
|
| 723 |
+
tiles="cartodbpositron"
|
| 724 |
+
)
|
| 725 |
+
|
| 726 |
+
# Add user location marker
|
| 727 |
+
folium.Marker(
|
| 728 |
+
[lat, lon],
|
| 729 |
+
popup='Your Location',
|
| 730 |
+
icon=folium.Icon(color='green', icon='home')
|
| 731 |
+
).add_to(m)
|
| 732 |
+
|
| 733 |
+
# Add markers for each store with color coding based on distance
|
| 734 |
+
for store in nearby_stores:
|
| 735 |
+
# Color code based on distance
|
| 736 |
+
if store['distance'] <= 2:
|
| 737 |
+
color = 'red' # Very close
|
| 738 |
+
elif store['distance'] <= 5:
|
| 739 |
+
color = 'orange' # Moderate distance
|
| 740 |
+
else:
|
| 741 |
+
color = 'blue' # Further away
|
| 742 |
+
|
| 743 |
+
# Create simplified popup content
|
| 744 |
+
popup_content = f"""
|
| 745 |
+
<div style='width: 200px; font-size: 14px;'>
|
| 746 |
+
<h4 style='color: {color}; margin: 0 0 8px 0;'>{store['store_name']}</h4>
|
| 747 |
+
<b>Distance:</b> {store['distance']} km<br>
|
| 748 |
+
<b>Est. Delivery:</b> {store['estimated_delivery_time']} mins<br>
|
| 749 |
+
<b>Categories:</b> {store['product_categories']}<br>
|
| 750 |
+
<button onclick="window.location.href='/api/stores/route?user_lat={lat}&user_lon={lon}&store_lat={store['location']['lat']}&store_lon={store['location']['lon']}&viz_type=simple'"
|
| 751 |
+
style='margin-top: 8px; padding: 8px; width: 100%; background-color: #007bff; color: white; border: none; border-radius: 4px;'>
|
| 752 |
+
Get Route
|
| 753 |
+
</button>
|
| 754 |
+
</div>
|
| 755 |
+
"""
|
| 756 |
+
|
| 757 |
+
# Add store marker
|
| 758 |
+
folium.Marker(
|
| 759 |
+
location=[store['location']['lat'], store['location']['lon']],
|
| 760 |
+
popup=folium.Popup(popup_content, max_width=300),
|
| 761 |
+
icon=folium.Icon(color=color, icon='info-sign'),
|
| 762 |
+
tooltip=f"{store['store_name']} ({store['distance']} km)"
|
| 763 |
+
).add_to(m)
|
| 764 |
+
|
| 765 |
+
# Add circle to show distance - only for closer stores to reduce complexity
|
| 766 |
+
if store['distance'] <= 5:
|
| 767 |
+
folium.Circle(
|
| 768 |
+
location=[store['location']['lat'], store['location']['lon']],
|
| 769 |
+
radius=store['distance'] * 100,
|
| 770 |
+
color=color,
|
| 771 |
+
fill=True,
|
| 772 |
+
opacity=0.1
|
| 773 |
+
).add_to(m)
|
| 774 |
+
|
| 775 |
+
# Add distance circles from user location - reduced to save memory
|
| 776 |
+
for circle_radius, color in [(2000, 'red'), (5000, 'orange')]:
|
| 777 |
+
folium.Circle(
|
| 778 |
+
location=[lat, lon],
|
| 779 |
+
radius=circle_radius,
|
| 780 |
+
color=color,
|
| 781 |
+
fill=False,
|
| 782 |
+
weight=1,dash_array='5, 5'
|
| 783 |
+
).add_to(m)
|
| 784 |
+
|
| 785 |
+
# Create mobile-friendly HTML content
|
| 786 |
+
html_content = f"""
|
| 787 |
+
<!DOCTYPE html>
|
| 788 |
+
<html>
|
| 789 |
+
<head>
|
| 790 |
+
<meta charset="utf-8">
|
| 791 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0, maximum-scale=1.0, user-scalable=no" />
|
| 792 |
+
<title>Nearby Stores</title>
|
| 793 |
+
<style>
|
| 794 |
+
body {{
|
| 795 |
+
margin: 0;
|
| 796 |
+
padding: 0;
|
| 797 |
+
width: 100vw;
|
| 798 |
+
height: 100vh;
|
| 799 |
+
overflow: hidden;
|
| 800 |
+
}}
|
| 801 |
+
#map {{
|
| 802 |
+
width: 100%;
|
| 803 |
+
height: 100%;
|
| 804 |
+
}}
|
| 805 |
+
.legend {{
|
| 806 |
+
position: fixed;
|
| 807 |
+
bottom: 20px;
|
| 808 |
+
right: 20px;
|
| 809 |
+
background: white;
|
| 810 |
+
padding: 10px;
|
| 811 |
+
border-radius: 5px;
|
| 812 |
+
box-shadow: 0 1px 5px rgba(0,0,0,0.2);
|
| 813 |
+
font-size: 12px;
|
| 814 |
+
z-index: 1000;
|
| 815 |
+
}}
|
| 816 |
+
.info-box {{
|
| 817 |
+
position: fixed;
|
| 818 |
+
top: 20px;
|
| 819 |
+
left: 20px;
|
| 820 |
+
background: white;
|
| 821 |
+
padding: 10px;
|
| 822 |
+
border-radius: 5px;
|
| 823 |
+
box-shadow: 0 1px 5px rgba(0,0,0,0.2);
|
| 824 |
+
font-size: 12px;
|
| 825 |
+
z-index: 1000;
|
| 826 |
+
}}
|
| 827 |
+
</style>
|
| 828 |
+
</head>
|
| 829 |
+
<body>
|
| 830 |
+
{m.get_root().render()}
|
| 831 |
+
<div class="legend">
|
| 832 |
+
<b>Distance Zones</b><br>
|
| 833 |
+
<span style="color: red;">●</span> < 2 km<br>
|
| 834 |
+
<span style="color: orange;">●</span> 2-5 km<br>
|
| 835 |
+
<span style="color: blue;">●</span> > 5 km
|
| 836 |
+
</div>
|
| 837 |
+
<div class="info-box">
|
| 838 |
+
<b>Search Radius:</b> {radius} km<br>
|
| 839 |
+
<b>Stores Found:</b> {len(nearby_stores)}
|
| 840 |
+
</div>
|
| 841 |
+
<script>
|
| 842 |
+
window.onload = function() {{
|
| 843 |
+
setTimeout(function() {{
|
| 844 |
+
window.dispatchEvent(new Event('resize'));
|
| 845 |
+
}}, 1000);
|
| 846 |
+
}};
|
| 847 |
+
</script>
|
| 848 |
+
</body>
|
| 849 |
+
</html>
|
| 850 |
+
"""
|
| 851 |
+
|
| 852 |
+
# Save and return the file
|
| 853 |
+
file_path = 'temp/locations_map.html'
|
| 854 |
+
with open(file_path, 'w', encoding='utf-8') as f:
|
| 855 |
+
f.write(html_content)
|
| 856 |
+
|
| 857 |
+
# Return the HTML content
|
| 858 |
+
return html_content
|
| 859 |
+
|
| 860 |
+
except Exception as e:
|
| 861 |
+
raise HTTPException(status_code=400, detail=str(e))
|
| 862 |
+
|
| 863 |
+
# Add endpoint to serve static files directly
|
| 864 |
+
@app.get("/temp/{file_path:path}", response_class=FileResponse)
|
| 865 |
+
async def get_temp_file(file_path: str):
|
| 866 |
+
"""Serve temporary files like HTML maps"""
|
| 867 |
+
full_path = os.path.join("temp", file_path)
|
| 868 |
+
if not os.path.exists(full_path):
|
| 869 |
+
raise HTTPException(status_code=404, detail="File not found")
|
| 870 |
+
return FileResponse(full_path)
|
| 871 |
+
|
| 872 |
+
# Add memory monitoring and management middleware
|
| 873 |
+
@app.middleware("http")
|
| 874 |
+
async def add_memory_management(request: Request, call_next):
|
| 875 |
+
# Cleanup before processing request
|
| 876 |
+
cleanup_temp_files()
|
| 877 |
+
|
| 878 |
+
# Process the request
|
| 879 |
+
response = await call_next(request)
|
| 880 |
+
|
| 881 |
+
# Cleanup after processing request
|
| 882 |
+
gc.collect()
|
| 883 |
+
|
| 884 |
+
return response
|
| 885 |
+
|
| 886 |
+
# For running the application directly (development mode)
|
| 887 |
+
if __name__ == "__main__":
|
| 888 |
+
import uvicorn
|
| 889 |
+
uvicorn.run(app, host="0.0.0.0", port=8000)
|