Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 import plotly.graph_objects as go
+import time
 
 # Load the Excel file
 file_path = 'Dhaka Metro Rail Fare 2.XLSX'  # Ensure the correct file path
@@ -54,7 +55,7 @@ else:
     1. Select your **Location station** from the dropdown menu.
     2. Select your **destination(s)** by clicking the destination buttons.
     3. The fare from your location to the selected destination(s) will be displayed below.
-    4. You can also see the stations marked on a map.
+    4. You can also see the stations marked on a map with an animated train journey!
     
     **Note:** The map highlights your location in green and destinations in blue.
     
@@ -107,7 +108,6 @@ else:
         fare_data = df[(df['Origin'] == origin) & (df['Destination'].isin(destinations))]
 
         # Display the fare data
-        # Display the fare data in a creative and aesthetic format
         if not fare_data.empty:
             for index, row in fare_data.iterrows():
                 origin_to_dest_fare = row['Fare (৳)']
@@ -133,65 +133,91 @@ else:
     else:
         st.write("Please select both an origin and at least one destination.")
 
-    st.write("Below is the Map for Dhaka Metro Rail 💶:")
+    st.write("Below is the Animated Map for Dhaka Metro Rail 💶:")
 
-    # Plotting the map using Plotly
+    # Plotting the map using Plotly with animation
     fig = go.Figure()
 
     # Add markers for each unique station
     unique_stations = pd.concat([df[['Origin', 'Origin_Lat', 'Origin_Lon']].rename(columns={'Origin': 'Station', 'Origin_Lat': 'Lat', 'Origin_Lon': 'Lon'}), 
                                  df[['Destination', 'Destination_Lat', 'Destination_Lon']].rename(columns={'Destination': 'Station', 'Destination_Lat': 'Lat', 'Destination_Lon': 'Lon'})]).drop_duplicates()
 
-    # Add a map marker for each station
-    for i, row in unique_stations.iterrows():
-        if row['Station'] == origin:
-            # Mark the origin as green
-            fig.add_trace(go.Scattermapbox(
-                mode="markers+text",
-                lon=[row['Lon']],
-                lat=[row['Lat']],
-                marker={'size': 12, 'color': 'green'},
-                text=row['Station'],
-                textposition="top center",
-                name=row['Station'],
-                customdata=[row['Station']]  # customdata is now a list
-            ))
-        elif row['Station'] in destinations:
-            # Mark the destination as blue
-            fig.add_trace(go.Scattermapbox(
-                mode="markers+text",
-                lon=[row['Lon']],
-                lat=[row['Lat']],
-                marker={'size': 12, 'color': 'blue'},
-                text=row['Station'],
-                textposition="top center",
-                name=row['Station'],
-                customdata=[row['Station']]  # customdata is now a list
-            ))
-        else:
-            # Mark all other stations as red
-            fig.add_trace(go.Scattermapbox(
-                mode="markers+text",
-                lon=[row['Lon']],
-                lat=[row['Lat']],
-                marker={'size': 12, 'color': 'red'},
-                text=row['Station'],
-                textposition="top center",
-                name=row['Station'],
-                customdata=[row['Station']]  # customdata is now a list
+    # Store the animation frames for the train's movement
+    frames = []
+
+    # For each selected destination, animate the train from the origin to the destination
+    if origin != default_origin and destinations:
+        for dest in destinations:
+            origin_coords = (coordinates[origin][0], coordinates[origin][1])
+            destination_coords = (coordinates[dest][0], coordinates[dest][1])
+
+            # Create frames for animation (move from origin to destination)
+            num_frames = 30
+            lat_diff = (destination_coords[0] - origin_coords[0]) / num_frames
+            lon_diff = (destination_coords[1] - origin_coords[1]) / num_frames
+
+            frames = [
+                go.Frame(
+                    data=[go.Scattermapbox(
+                        mode="markers+text",
+                        lon=[origin_coords[1] + lon_diff * i],
+                        lat=[origin_coords[0] + lat_diff * i],
+                        marker={'size': 15, 'color': 'orange'},
+                        text="🚆 Train",
+                        textposition="top center"
+                    )],
+                    name=f"frame_{i}"
+                ) for i in range(num_frames)
+            ]
+
+            # Add final frame for destination
+            frames.append(go.Frame(
+                data=[go.Scattermapbox(
+                    mode="markers+text",
+                    lon=[destination_coords[1]],
+                    lat=[destination_coords[0]],
+                    marker={'size': 15, 'color': 'orange'},
+                    text="🚆 Train",
+                    textposition="top center"
+                )],
+                name=f"frame_{num_frames}"
             ))
 
-    # Map layout
+    # Add base station markers
+    for i, row in unique_stations.iterrows():
+        fig.add_trace(go.Scattermapbox(
+            mode="markers+text",
+            lon=[row['Lon']],
+            lat=[row['Lat']],
+            marker={'size': 10, 'color': 'blue'},
+            text=row['Station'],
+            textposition="top center",
+            name=row['Station']
+        ))
+
+    # Map layout with animation
     fig.update_layout(
         mapbox=dict(
             style="open-street-map",
             center=go.layout.mapbox.Center(lat=23.780, lon=90.400),  # Center on Dhaka
             zoom=11
         ),
+        updatemenus=[dict(
+            type="buttons",
+            showactive=False,
+            buttons=[dict(
+                label="Play",
+                method="animate",
+                args=[None, dict(frame=dict(duration=100, redraw=True), fromcurrent=True)]
+            )]
+        )],
         margin={"r":0,"t":0,"l":0,"b":0},
         showlegend=False,
-        title="Dhaka Metro Rail Location Map"
+        title="Dhaka Metro Rail Location Map with Train Animation"
     )
 
+    # Add frames for animation
+    fig.frames = frames
+
     # Show plot in Streamlit
     st.plotly_chart(fig)