Update app.py

app.py

@@ -1,4 +1,4 @@
-# streamlit_app.py - Bolt Driver Recommendation Syste
+# streamlit_app.py - Bolt Driver Recommendation System
 import streamlit as st
 import pandas as pd
 import numpy as np
@@ -368,87 +368,82 @@ def main():
     with col1:
         st.markdown('<div class="section-header">Demand Map</div>', unsafe_allow_html=True)
         
-        # Create map
-        m = folium.Map(
-            location=[59.436, 24.753],  # Tallinn center
-            zoom_start=12,
-            tiles="CartoDB positron"
-        )
-        
-        # Add driver marker if location provided
-        if use_location:
-            folium.Marker(
-                location=[current_lat, current_lng],
-                popup="Your Location",
-                icon=folium.Icon(color="blue", icon="user", prefix="fa"),
-                tooltip="Your Current Location"
-            ).add_to(m)
-        
-        # Add recommendation markers
-        for i, rec in enumerate(recommendations):
-            folium.CircleMarker(
-                location=[rec["center_lat"], rec["center_lng"]],
-                radius=20,
-                color="red",
-                fill=True,
-                fill_color="red",
-                fill_opacity=0.6,
-                popup=f"""
-                <b>Recommendation {i+1}</b><br>
-                Expected rides: {rec['predicted_rides']:.1f}<br>
-                Avg value: €{rec['avg_value']:.2f}<br>
-                Expected value: €{rec['expected_value']:.2f}<br>
-                {f'Distance: {rec["distance_km"]:.2f} km' if rec["distance_km"] is not None else ''}
-                """
-            ).add_to(m)
+        try:
+            # Create map
+            m = folium.Map(
+                location=[59.436, 24.753],  # Tallinn center
+                zoom_start=12,
+                tiles="CartoDB positron"
+            )
+            
+            # Add driver marker if location provided
+            if use_location:
+                folium.Marker(
+                    location=[current_lat, current_lng],
+                    popup="Your Location",
+                    icon=folium.Icon(color="blue", icon="user", prefix="fa"),
+                    tooltip="Your Current Location"
+                ).add_to(m)
             
-            # Add number label
-            folium.Marker(
-                location=[rec["center_lat"], rec["center_lng"]],
-                icon=folium.DivIcon(
-                    html=f"""
-                    <div style="
-                        font-size: 12pt;
-                        color: white;
-                        font-weight: bold;
-                        text-align: center;
-                        width: 25px;
-                        height: 25px;
-                        line-height: 25px;
-                    ">{i+1}</div>
+            # Add recommendation markers
+            for i, rec in enumerate(recommendations):
+                folium.CircleMarker(
+                    location=[rec["center_lat"], rec["center_lng"]],
+                    radius=20,
+                    color="red",
+                    fill=True,
+                    fill_color="red",
+                    fill_opacity=0.6,
+                    popup=f"""
+                    <b>Recommendation {i+1}</b><br>
+                    Expected rides: {rec['predicted_rides']:.1f}<br>
+                    Avg value: €{rec['avg_value']:.2f}<br>
+                    Expected value: €{rec['expected_value']:.2f}<br>
+                    {f'Distance: {rec["distance_km"]:.2f} km' if rec["distance_km"] is not None else ''}
                     """
-                )
-            ).add_to(m)
-        
-        # Add heatmap if enabled
-        if show_heatmap:
-            # Get a larger set of predictions for the heatmap
-            all_predictions = model.predict(day_idx, selected_hour, top_n=100)
-            heat_data = [
-                [pred["center_lat"], pred["center_lng"], pred["predicted_rides"]] 
-                for pred in all_predictions
-            ]
+                ).add_to(m)
+                
+                # Add number label - using HTML directly to avoid the split error
+                folium.Marker(
+                    location=[rec["center_lat"], rec["center_lng"]],
+                    icon=folium.DivIcon(
+                        html=f'<div style="font-size:12pt;color:white;font-weight:bold;text-align:center;width:25px;height:25px;line-height:25px;">{i+1}</div>'
+                    )
+                ).add_to(m)
+            
+            # Add heatmap if enabled
+            if show_heatmap:
+                # Get a larger set of predictions for the heatmap
+                all_predictions = model.predict(day_idx, selected_hour, top_n=100)
+                heat_data = [
+                    [pred["center_lat"], pred["center_lng"], pred["predicted_rides"]] 
+                    for pred in all_predictions
+                ]
+                
+                # Add heatmap layer
+                HeatMap(
+                    heat_data,
+                    radius=15,
+                    gradient={
+                        0.2: 'blue',
+                        0.4: 'lime',
+                        0.6: 'yellow',
+                        0.8: 'orange',
+                        1.0: 'red'
+                    },
+                    name="Demand Heatmap",
+                    show=True
+                ).add_to(m)
+            
+            # Add layer control
+            folium.LayerControl().add_to(m)
+            
+            # Display the map
+            folium_static(m, width=700)
             
-            # Add heatmap layer
-            HeatMap(
-                heat_data,
-                radius=15,
-                gradient={
-                    0.2: 'blue',
-                    0.4: 'lime',
-                    0.6: 'yellow',
-                    0.8: 'orange',
-                    1.0: 'red'
-                },
-                name="Demand Heatmap",
-                show=True
-            ).add_to(m)
-        
-        # Add layer control
-        folium.LayerControl().add_to(m)
-        
-        # Display the map
-        folium_static(m, width=700)
+        except Exception as e:
+            st.error(f"Error rendering map: {e}")
+            st.info("Showing tabular results instead.")
     
     with col2:
         st.markdown('<div class="section-header">Recommendations</div>', unsafe_allow_html=True)