Update app.py (#3)

- Update app.py (715e443b9f3159bc32fae0192aa1475aef14f0a3)

app.py

@@ -1,110 +1,111 @@
-import streamlit as st
-import serial
-import time
-import folium
-from folium.plugins import MarkerCluster
-import io
-import base64
-
-# Setup serial connection
-arduino_port = "COM3"  # Replace with your Arduino port. If necessary, change to a different port.
-baud_rate = 9600
-
-# Check if the port is available before attempting to open it
-try:
-    ser = serial.Serial(arduino_port, baud_rate)
-except serial.SerialException as e:
-    st.error(f"Error: {e}")
-    st.stop()  # Stop the script if the serial connection fails
-
-# Dummy user credentials (for login)
-USER_CREDENTIALS = {
-    "admin": "password123",  # Username: admin, Password: password123
-    "user1": "mypassword"    # Username: user1, Password: mypassword
-}
-
-# Function to display login page
-def login_page():
-    st.title("Bus System Status Monitoring")
-    st.subheader("Please login to access the system")
-
-    # Input for username and password
-    username = st.text_input("Username", "")
-    password = st.text_input("Password", "", type="password")
-
-    # Check if the user submits the login form
-    if st.button("Login"):
-        if username in USER_CREDENTIALS and USER_CREDENTIALS[username] == password:
-            st.success("Login Successful!")
-            return username  # return username to proceed to the monitoring page
-        else:
-            st.error("Invalid credentials, please try again.")
-            return None
-
-# Function to read data from Arduino (simulating bus data including GPS location)
-def read_data_from_arduino():
-    bus_name = ""
-    seat_status = ""
-    distance = 0
-    latitude = 0.0
-    longitude = 0.0
-
-    while True:
-        if ser.in_waiting > 0:
-            data = ser.readline().decode('utf-8').strip()
-            if "Bus Name:" in data:
-                bus_name = data.split("Bus Name: ")[1]
-            if "Seat Status:" in data:
-                seat_status = data.split("Seat Status: ")[1]
-            if "Distance:" in data:
-                distance = int(data.split("Distance: ")[1])
-            if "Latitude:" in data:
-                latitude = float(data.split("Latitude: ")[1])
-            if "Longitude:" in data:
-                longitude = float(data.split("Longitude: ")[1])
-
-            # Display data in Streamlit
-            st.subheader(f"Bus Name: {bus_name}")
-            st.subheader(f"Seat Status: {seat_status}")
-            st.subheader(f"Distance to Bus: {distance} cm")
-            st.subheader(f"Latitude: {latitude}, Longitude: {longitude}")
-
-            # Plot Bus on Map
-            plot_map(latitude, longitude)
-
-            time.sleep(1)  # Delay for 1 second before reading next set of data
-
-# Function to plot the map with the bus location
-def plot_map(latitude, longitude):
-    # Create a map centered around the bus location
-    bus_map = folium.Map(location=[latitude, longitude], zoom_start=15)
-
-    # Add a marker for the bus location
-    folium.Marker([latitude, longitude], popup="Bus Location").add_to(bus_map)
-
-    # Render the map in Streamlit
-    st.subheader("Real-Time Bus Location")
-    # Save the map to a PNG image
-    map_stream = io.BytesIO()
-    bus_map.save(map_stream, close_file=False)
-
-    # Convert the map to base64 for display in Streamlit
-    map_stream.seek(0)
-    map_data = base64.b64encode(map_stream.getvalue()).decode()
-    st.markdown(f'<img src="data:image/png;base64,{map_data}" width="100%" />', unsafe_allow_html=True)
-
-# Main function for the Streamlit app
-def main():
-    # Show the login page
-    username = login_page()
-
-    # If the user is logged in, show the bus system page
-    if username:
-        st.write("Loading the Bus System Monitoring Dashboard...")
-
-        # Run the real-time data reading from Arduino
-        read_data_from_arduino()
-
-# Run the app
-if __name__ == "__main__":
-    main()
+import streamlit as st 
+import pandas as pd
+from datetime import datetime
+import folium
+from streamlit_folium import folium_static
+import groq
+
+# Load bus data
+data_path = r"C:\Users\Muthuraja\OneDrive\Attachments\Desktop\second\pdp.csv"
+df = pd.read_csv(data_path)
+
+# Dummy user credentials
+USER_CREDENTIALS = {
+    "Muthuraja":"virat",
+    "Praveen":"dhoni",
+    "Pandi":"kabadi",
+    "admin": "password123",
+    "user": "buspass2025"
+}
+
+# Groq API Key
+GROQ_API_KEY = "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"
+groq.api_key = GROQ_API_KEY
+
+# User login
+def authenticate(username, password):
+    if username in USER_CREDENTIALS and USER_CREDENTIALS[username] == password:
+        return True
+    return False
+
+# Generate bus prediction using Groq API
+def predict_bus_status(bus_number, city, area):
+    prompt = f"Predict the status and arrival time for bus {bus_number} in {city}, {area}."
+    try:
+        client = groq.Client(api_key=GROQ_API_KEY)
+        response = client.chat.completions.create(
+            model="llama3-70b-8192",  # Correct model name for Groq  # Use Groq's model, adjust if needed
+            messages=[{"role": "system", "content": prompt}]
+        )
+        return response.choices[0].message.content.strip()
+    except Exception as e:
+        return f"Error: {e}"
+
+# Plot bus locations using Folium with real coordinates
+def plot_bus_map(area_df):
+    if area_df.empty:
+        st.warning("No bus data available for this area.")
+        return
+
+    if "latitude" not in area_df.columns or "longitude" not in area_df.columns:
+        st.error("Latitude and longitude columns missing. Please update CSV.")
+        return
+
+    center_lat = area_df["latitude"].mean()
+    center_lon = area_df["longitude"].mean()
+    
+    m = folium.Map(location=[center_lat, center_lon], zoom_start=12)
+
+    for _, row in area_df.iterrows():
+        folium.Marker(
+            location=[row["latitude"], row["longitude"]],
+            popup=f"{row['bus_number']} - {row['bus_route']}\nArrival: {row['arrival_time']}\nStatus: {row['status']}",
+            icon=folium.Icon(color="blue" if row["status"].lower() == "on time" else "red")
+        ).add_to(m)
+
+    folium_static(m)
+
+# Streamlit UI
+st.title("🚍 Tamil Nadu Bus Tracking & Prediction System")
+
+# Login form
+if "authenticated" not in st.session_state:
+    st.session_state.authenticated = False
+
+if not st.session_state.authenticated:
+    username = st.text_input("Username")
+    password = st.text_input("Password", type="password")
+    if st.button("Login"):
+        if authenticate(username, password):
+            st.session_state.authenticated = True
+            st.success("Login successful!")
+        else:
+            st.error("Invalid username or password")
+
+if st.session_state.authenticated:
+    city = st.selectbox("Select City", df["city"].unique())
+    area = st.selectbox("Select Area", df[df["city"] == city]["area"].unique())
+
+    filtered_df = df[(df["city"] == city) & (df["area"] == area)]
+    
+    st.subheader(f"🚌 Bus Details for {city}, {area}")
+    st.dataframe(filtered_df)
+
+    st.subheader("🗺️ Bus Map View")
+    plot_bus_map(filtered_df)
+
+    def predict_next_bus(area_df):
+        now = datetime.now()
+        upcoming_buses = area_df[area_df["arrival_time"] > now.strftime("%Y-%m-%d %H:%M:%S")]
+        if not upcoming_buses.empty:
+            next_bus = upcoming_buses.iloc[0]
+            prediction = predict_bus_status(next_bus['bus_number'], city, area)
+            return f"🚏 Next bus: {next_bus['bus_number']} arriving at {next_bus['arrival_time']}\n🔮 Prediction: {prediction}"
+        return "⚠️ No upcoming buses available."
+
+    prediction = predict_next_bus(filtered_df)
+    st.success(prediction)
+
+    if st.button("Logout"):
+        st.session_state.authenticated = False
+        st.experimental_rerun()