Update tools.py

tools.py

@@ -6,7 +6,7 @@ import time
 import folium
 import os
 
-from api  import geocode_address_nominatim,get_route_osrm,get_alternative_routes_osrm, get_detailed_route_with_instructions
+from api  import geocode_address_nominatim,get_route_osrm,get_alternative_routes_osrm, get_detailed_route_with_instructions,get_realtime_traffic
 
 def estimate_traffic_from_time() -> float:
     """
@@ -61,6 +61,34 @@ REAL_VEHICLES = [
     },
 ]
 
+def calculate_traffic(origin_coords: Tuple[float, float], dest_coords: Tuple[float, float]):
+    """Analyze the traffic using the realtime data"""
+    traffic_data = get_realtime_traffic(origin_coords, dest_coords)
+
+    print("Traffic Data:",traffic_data)
+
+    if not traffic_data:
+        print("sing fallback: estimate_traffic_from_time()")
+        return estimate_traffic_from_time()
+
+    try:
+        base_time = traffic_data["noTrafficTravelTimeInSeconds"]
+        current_time = traffic_data["travelTimeInSeconds"]
+        delay = traffic_data["trafficDelayInSeconds"]
+
+        if base_time == 0:
+            return 1.0 
+
+        traffic_factor = current_time / base_time
+
+        traffic_factor = max(0.8, min(traffic_factor, 2.0))
+        return round(traffic_factor, 2)
+
+    except KeyError:
+        print("Missing fields in traffic data, fallback to estimate_traffic_from_time()")
+        return estimate_traffic_from_time()
+
+
 def format_route_instructions(route_data: Dict) -> str:
     """
     Format route instructions in readable format
@@ -120,7 +148,7 @@ def find_optimal_route(origin_coords: Tuple[float, float],
     if not detailed_routes:
         return None
     
-    traffic_factor = estimate_traffic_from_time()
+    traffic_factor = calculate_traffic(origin_coords,dest_coords)
     
     scored_routes = []
     for route_data in detailed_routes:
@@ -278,7 +306,7 @@ def real_route_planner(origin: str, destination: str) -> str:
 
     return result
 
-def real_cost_optimizer(distance_km: float, weight_kg: float, duration_min: float) -> str:
+def real_cost_optimizer(origin_coords:float,dest_coords:float,distance_km: float, weight_kg: float, duration_min: float) -> str:
     """
     Calculate real-world delivery costs
     """
@@ -298,7 +326,7 @@ def real_cost_optimizer(distance_km: float, weight_kg: float, duration_min: floa
         
         base_fee = 150
         
-        traffic_multiplier = estimate_traffic_from_time()
+        traffic_multiplier = calculate_traffic(origin_coords,dest_coords)
         
         capacity_usage = weight_kg / vehicle["capacity_kg"]
         capacity_multiplier = 1.15 if capacity_usage > 0.8 else 1.0
@@ -351,7 +379,7 @@ def real_traffic_analyzer(origin: str, destination: str) -> str:
         return "No route found for traffic analysis"
     
     current_hour = datetime.now().hour
-    traffic_factor = estimate_traffic_from_time()
+    traffic_factor = calculate_traffic(origin_coords,dest_coords)
     
     if traffic_factor >= 1.5:
         traffic_level = "Heavy"