Update app.py

app.py

@@ -4,6 +4,7 @@ import folium
 import numpy as np
 from scipy.optimize import linear_sum_assignment
 import pandas as pd
+from folium import plugins
 
 def calculate_distances(coordinates):
     """Calculate distances between all points using Haversine formula"""
@@ -13,7 +14,6 @@ def calculate_distances(coordinates):
     for i in range(n):
         for j in range(n):
             if i != j:
-                # Haversine formula
                 lat1, lon1 = coordinates[i]
                 lat2, lon2 = coordinates[j]
                 
@@ -44,6 +44,17 @@ def get_place_coordinates(place_name):
     except:
         return None
 
+def create_numbered_marker(number):
+    """Create a custom marker with a number"""
+    return plugins.BeautifyIcon(
+        number=number,
+        border_color='#b4b4b4',
+        border_width=1,
+        text_color='black',
+        background_color='white',
+        inner_icon_style='margin-top:0;'
+    )
+
 def main():
     st.title("AI Travel Route Optimizer")
     
@@ -76,17 +87,20 @@ def main():
         
         m = folium.Map(location=[center_lat, center_lon], zoom_start=4)
         
+        # Create optimized itinerary list first
+        optimized_places = [(i+1, places[idx][0]) for i, idx in enumerate(optimized_indices)]
+        
         # Add markers and route lines
         for i in range(len(optimized_indices)):
             current_idx = optimized_indices[i]
             next_idx = optimized_indices[(i + 1) % len(optimized_indices)]
             
-            # Add marker
+            # Add numbered marker
             place_name, (lat, lon) = places[current_idx]
             folium.Marker(
                 [lat, lon],
                 popup=f"Stop {i+1}: {place_name}",
-                icon=folium.Icon(color='red', icon='info-sign')
+                icon=create_numbered_marker(i+1)
             ).add_to(m)
             
             # Add line to next point
@@ -101,10 +115,10 @@ def main():
         # Display map
         st.components.v1.html(m._repr_html_(), height=600)
         
-        # Display itinerary
+        # Display optimized itinerary
         st.header("Optimized Itinerary")
-        for i in range(len(optimized_indices)):
-            st.write(f"{i+1}. {places[optimized_indices[i]][0]}")
+        for stop_num, place_name in optimized_places:
+            st.write(f"{stop_num}. {place_name}")
         
         # Calculate total distance
         total_distance = 0