add utils src

src/components/__init__.py

@@ -0,0 +1 @@
+# This file is intentionally left blank.

src/components/map_visualization.py

@@ -0,0 +1,197 @@
+def load_data():
+    """
+    Load delivery and vehicle data from CSV files
+    
+    Returns:
+        tuple: (delivery_data, vehicle_data)
+    """
+    # Get the project root directory
+    root_dir = Path(__file__).resolve().parent.parent.parent
+    
+    # Define data paths
+    delivery_data_path = os.path.join(root_dir, 'data', 'delivery-data', 'delivery_data.csv')
+    vehicle_data_path = os.path.join(root_dir, 'data', 'vehicle-data', 'vehicle_data.csv')
+    
+    # Load data
+    try:
+        delivery_data = pd.read_csv(delivery_data_path)
+        vehicle_data = pd.read_csv(vehicle_data_path)
+        return delivery_data, vehicle_data
+    except FileNotFoundError as e:
+        st.error(f"Could not load data: {e}")
+        st.info("Please generate the data first by running: python src/utils/generate_all_data.py")
+        return None, None
+
+def create_delivery_map(delivery_data=None, vehicle_data=None, show_deliveries=True, show_depots=True, 
+                       date_filter=None, status_filter=None, priority_filter=None):
+    """
+    Create a Folium map with markers for deliveries and vehicle depots
+    
+    Parameters:
+        delivery_data (pd.DataFrame): Delivery data
+        vehicle_data (pd.DataFrame): Vehicle data
+        show_deliveries (bool): Whether to show delivery markers
+        show_depots (bool): Whether to show depot markers
+        date_filter (str): Filter deliveries by date
+        status_filter (str): Filter deliveries by status
+        priority_filter (str): Filter deliveries by priority
+    
+    Returns:
+        folium.Map: Folium map with markers
+    """
+    # If data not provided, load it
+    if delivery_data is None or vehicle_data is None:
+        delivery_data, vehicle_data = load_data()
+        if delivery_data is None or vehicle_data is None:
+            return None
+    
+    # Apply filters to delivery data
+    if date_filter is not None:
+        delivery_data = delivery_data[delivery_data['delivery_date'] == date_filter]
+        
+    if status_filter is not None:
+        delivery_data = delivery_data[delivery_data['status'] == status_filter]
+        
+    if priority_filter is not None:
+        delivery_data = delivery_data[delivery_data['priority'] == priority_filter]
+    
+    # Create map centered around Singapore
+    singapore_coords = [1.3521, 103.8198]  # Center of Singapore
+    m = folium.Map(location=singapore_coords, zoom_start=12)
+    
+    # Add delivery markers
+    if show_deliveries and not delivery_data.empty:
+        for _, row in delivery_data.iterrows():
+            # Create popup content with delivery information
+            popup_content = f"""
+                <b>Delivery ID:</b> {row['delivery_id']}<br>
+                <b>Customer:</b> {row['customer_name']}<br>
+                <b>Address:</b> {row['address']}<br>
+                <b>Time Window:</b> {row['time_window']}<br>
+                <b>Status:</b> {row['status']}<br>
+                <b>Priority:</b> {row['priority']}<br>
+                <b>Weight:</b> {row['weight_kg']} kg<br>
+                <b>Volume:</b> {row['volume_m3']} m³
+            """
+            
+            # Set marker color based on priority
+            color_map = {'High': 'red', 'Medium': 'orange', 'Low': 'blue'}
+            color = color_map.get(row['priority'], 'blue')
+            
+            # Add marker to map
+            folium.Marker(
+                location=[row['latitude'], row['longitude']],
+                popup=folium.Popup(popup_content, max_width=300),
+                tooltip=f"Delivery {row['delivery_id']}: {row['customer_name']}",
+                icon=folium.Icon(color=color, icon="package", prefix="fa")
+            ).add_to(m)
+    
+    # Add depot markers
+    if show_depots and not vehicle_data.empty:
+        for _, row in vehicle_data.iterrows():
+            # Create popup content with vehicle information
+            popup_content = f"""
+                <b>Vehicle ID:</b> {row['vehicle_id']}<br>
+                <b>Type:</b> {row['vehicle_type']}<br>
+                <b>Driver:</b> {row['driver_name']}<br>
+                <b>Status:</b> {row['status']}<br>
+                <b>Capacity:</b> {row['max_weight_kg']} kg / {row['max_volume_m3']} m³<br>
+                <b>Working Hours:</b> {row['start_time']} - {row['end_time']}
+            """
+            
+            # Add marker to map
+            folium.Marker(
+                location=[row['depot_latitude'], row['depot_longitude']],
+                popup=folium.Popup(popup_content, max_width=300),
+                tooltip=f"Depot: {row['vehicle_id']}",
+                icon=folium.Icon(color="green", icon="truck", prefix="fa")
+            ).add_to(m)
+    
+    return m
+
+def display_map_component():
+    """
+    Display the map visualization component in Streamlit
+    """
+    st.subheader("Delivery and Depot Locations")
+    
+    # Load data
+    delivery_data, vehicle_data = load_data()
+    if delivery_data is None or vehicle_data is None:
+        return
+    
+    # Create sidebar filters
+    with st.sidebar:
+        st.subheader("Map Filters")
+        
+        # Show/hide options
+        show_deliveries = st.checkbox("Show Deliveries", value=True)
+        show_depots = st.checkbox("Show Depots", value=True)
+        
+        # Delivery date filter
+        dates = sorted(delivery_data['delivery_date'].unique())
+        selected_date = st.selectbox(
+            "Filter by Date",
+            options=["All"] + list(dates),
+            index=0
+        )
+        date_filter = None if selected_date == "All" else selected_date
+        
+        # Delivery status filter
+        statuses = sorted(delivery_data['status'].unique())
+        selected_status = st.selectbox(
+            "Filter by Status",
+            options=["All"] + list(statuses),
+            index=0
+        )
+        status_filter = None if selected_status == "All" else selected_status
+        
+        # Delivery priority filter
+        priorities = sorted(delivery_data['priority'].unique())
+        selected_priority = st.selectbox(
+            "Filter by Priority",
+            options=["All"] + list(priorities),
+            index=0
+        )
+        priority_filter = None if selected_priority == "All" else selected_priority
+    
+    # Display statistics
+    col1, col2, col3 = st.columns(3)
+    
+    # Apply filters for stats calculation
+    filtered_delivery_data = delivery_data
+    if date_filter:
+        filtered_delivery_data = filtered_delivery_data[filtered_delivery_data['delivery_date'] == date_filter]
+    if status_filter:
+        filtered_delivery_data = filtered_delivery_data[filtered_delivery_data['status'] == status_filter]
+    if priority_filter:
+        filtered_delivery_data = filtered_delivery_data[filtered_delivery_data['priority'] == priority_filter]
+    
+    with col1:
+        st.metric("Total Deliveries", filtered_delivery_data.shape[0])
+    
+    with col2:
+        st.metric("Total Weight", f"{filtered_delivery_data['weight_kg'].sum():.2f} kg")
+    
+    with col3:
+        st.metric("Available Vehicles", vehicle_data[vehicle_data['status'] == 'Available'].shape[0])
+    
+    # Create and display the map
+    delivery_map = create_delivery_map(
+        delivery_data=delivery_data,
+        vehicle_data=vehicle_data,
+        show_deliveries=show_deliveries,
+        show_depots=show_depots,
+        date_filter=date_filter,
+        status_filter=status_filter,
+        priority_filter=priority_filter
+    )
+    
+    if delivery_map:
+        folium_static(delivery_map, width=800, height=600)
+    else:
+        st.error("Could not create map. Please check that data is available.")
+
+if __name__ == "__main__":
+    # Run the map visualization component
+    display_map_component()

src/streamlit_app.py

@@ -1,40 +0,0 @@
-import altair as alt
-import numpy as np
-import pandas as pd
-import streamlit as st
-
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))