Create app.py

app.py

@@ -0,0 +1,153 @@
+import streamlit as st
+import pandas as pd
+import folium
+from streamlit_folium import st_folium
+from datetime import datetime
+from geopy.distance import geodesic
+import requests
+import os
+import json
+
+# -----------------------------
+# GROQ API Configuration
+# -----------------------------
+GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
+GROQ_API_KEY = st.secrets["TMS"] if "GROQ_API_KEY" in st.secrets else os.getenv("GROQ_API_KEY")
+
+HEADERS = {
+    "Authorization": f"Bearer {GROQ_API_KEY}",
+    "Content-Type": "application/json"
+}
+
+# -----------------------------
+# Traccar API Configuration (Simulated Setup)
+# -----------------------------
+TRACCAR_API_URL = "https://demo.traccar.org/api"
+TRACCAR_USERNAME = st.secrets["TRACCAR_USERNAME"] if "TRACCAR_USERNAME" in st.secrets else os.getenv("TRACCAR_USERNAME")
+TRACCAR_PASSWORD = st.secrets["TRACCAR_PASSWORD"] if "TRACCAR_PASSWORD" in st.secrets else os.getenv("TRACCAR_PASSWORD")
+
+# -----------------------------
+# Simulated Databases
+# -----------------------------
+employee_data = pd.DataFrame({
+    "Employee ID": ["E001", "E002", "E003"],
+    "Name": ["Alice", "Bob", "Charlie"],
+    "Pickup Location": [(28.6139, 77.2090), (28.7041, 77.1025), (28.5355, 77.3910)]
+})
+
+vehicle_data = pd.DataFrame({
+    "Vehicle ID": ["V001", "V002"],
+    "Driver Name": ["Ravi", "Sunil"],
+    "Capacity": [2, 2],
+    "Current Location": [(28.6448, 77.2167), (28.4595, 77.0266)]
+})
+
+allocation = {}
+
+# -----------------------------
+# Utility Functions
+# -----------------------------
+def allocate_vehicles():
+    global allocation
+    allocation = {v_id: [] for v_id in vehicle_data["Vehicle ID"]}
+    emp_index = 0
+    for v_id in allocation:
+        cap = int(vehicle_data[vehicle_data["Vehicle ID"] == v_id]["Capacity"].values[0])
+        while cap > 0 and emp_index < len(employee_data):
+            allocation[v_id].append(employee_data.iloc[emp_index].to_dict())
+            emp_index += 1
+            cap -= 1
+
+def show_map():
+    m = folium.Map(location=[28.6139, 77.2090], zoom_start=10)
+    # Employee locations
+    for _, row in employee_data.iterrows():
+        folium.Marker(location=row["Pickup Location"], tooltip=f"{row['Name']}").add_to(m)
+    # Vehicle locations
+    for _, row in vehicle_data.iterrows():
+        folium.Marker(location=row["Current Location"], tooltip=f"Vehicle: {row['Vehicle ID']} (Driver: {row['Driver Name']})", icon=folium.Icon(color='green')).add_to(m)
+    st_folium(m, width=700)
+
+def generate_summary(log_text):
+    if not GROQ_API_KEY:
+        return "GROQ API key not provided."
+    payload = {
+        "model": "mixtral-8x7b-32768",
+        "messages": [
+            {"role": "system", "content": "You are a helpful assistant summarizing GPS log data."},
+            {"role": "user", "content": log_text}
+        ]
+    }
+    try:
+        response = requests.post(GROQ_API_URL, headers=HEADERS, data=json.dumps(payload))
+        result = response.json()
+        return result['choices'][0]['message']['content']
+    except Exception as e:
+        return f"Error: {e}"
+
+def fetch_traccar_device_locations():
+    if not TRACCAR_USERNAME or not TRACCAR_PASSWORD:
+        return []
+    try:
+        response = requests.get(
+            f"{TRACCAR_API_URL}/positions",
+            auth=(TRACCAR_USERNAME, TRACCAR_PASSWORD)
+        )
+        if response.status_code == 200:
+            return response.json()
+        else:
+            st.warning("Failed to fetch real-time data from Traccar API.")
+            return []
+    except Exception as e:
+        st.error(f"Traccar API error: {e}")
+        return []
+
+# -----------------------------
+# Streamlit UI
+# -----------------------------
+st.title("Company Vehicle Transport Management System")
+
+menu = st.sidebar.selectbox("Menu", ["Dashboard", "Allocate Vehicles", "Track Vehicles", "Summarize Logs"])
+
+if menu == "Dashboard":
+    st.subheader("Employee Data")
+    st.dataframe(employee_data)
+    st.subheader("Vehicle Data")
+    st.dataframe(vehicle_data)
+
+elif menu == "Allocate Vehicles":
+    allocate_vehicles()
+    st.subheader("Allocation Result")
+    for v_id, emp_list in allocation.items():
+        st.write(f"**Vehicle {v_id}**")
+        for emp in emp_list:
+            st.write(f"- {emp['Name']} (ID: {emp['Employee ID']})")
+
+elif menu == "Track Vehicles":
+    st.subheader("Live Vehicle and Employee Location Map")
+    positions = fetch_traccar_device_locations()
+    if positions:
+        m = folium.Map(location=[28.6139, 77.2090], zoom_start=10)
+        for pos in positions:
+            lat = pos.get("latitude")
+            lon = pos.get("longitude")
+            device_id = pos.get("deviceId")
+            folium.Marker(location=[lat, lon], tooltip=f"Device ID: {device_id}", icon=folium.Icon(color='blue')).add_to(m)
+        st_folium(m, width=700)
+    else:
+        show_map()
+
+elif menu == "Summarize Logs":
+    st.subheader("Summarize Driver/Vehicle Logs")
+    log_text = st.text_area("Paste GPS or Driver Log Text:")
+    if st.button("Summarize with GROQ"):
+        with st.spinner("Generating summary..."):
+            summary = generate_summary(log_text)
+            st.success("Summary:")
+            st.write(summary)
+
+# -----------------------------
+# TODO: Integration Points
+# -----------------------------
+# - Add user authentication system
+# - Store and retrieve persistent data from a database like Firebase or Supabase