KMRTRUCKING1 / app.py
kmr24's picture
Upload app.py
10c5d60 verified
import streamlit as st
from pymongo import MongoClient
from datetime import datetime
import pandas as pd
# MongoDB connection details
MONGO_URI = "mongodb+srv://kmrlogistics:[email protected]/"
DB_NAME = "trucking"
COLLECTION_NAME_DRIVERS = "drivers"
COLLECTION_NAME_TRUCKS = "trucks"
COLLECTION_NAME_OWNER_OPERATORS = "owner_operators"
# Establish a connection to MongoDB
client = MongoClient(MONGO_URI)
db = client[DB_NAME]
drivers_collection = db[COLLECTION_NAME_DRIVERS]
trucks_collection = db[COLLECTION_NAME_TRUCKS]
owner_operators_collection = db[COLLECTION_NAME_OWNER_OPERATORS]
# User authentication function
def login(username, password):
"""Check the provided username and password."""
return username == "KMR LOGISTICS" and password == "kmr2024"
# Initialize Streamlit session state for login
if "logged_in" not in st.session_state:
st.session_state.logged_in = False
# Streamlit app layout
st.title("Kmr Logistics Management")
# Login section
st.sidebar.header("User Login")
if not st.session_state.logged_in:
username = st.sidebar.text_input("Username")
password = st.sidebar.text_input("Password", type="password")
if st.sidebar.button("Login"):
if login(username, password):
st.session_state.logged_in = True
st.sidebar.success("Logged in successfully!")
else:
st.sidebar.error("Invalid username or password.")
else:
st.sidebar.success("Logged in as kmr logistics.")
# Show the main application only if the user is logged in
if st.session_state.logged_in:
# Page navigation
page = st.sidebar.radio("Select a page", [
"Add Truck",
"Delete Truck",
"Add Driver",
"Delete Driver",
"Add Owner Operator",
"Delete Owner Operator",
"Assign Truck",
"Add Load",
"Update/Delete Load",
"View Owner Operator Info",
"View Information"
])
if page == "Add Truck":
st.header("Add Truck Information")
truck_name = st.text_input("Truck Name")
truck_plate_number = st.text_input("Truck Plate Number")
if st.button("Save Truck Info"):
if truck_name and truck_plate_number:
# Check if truck exists
existing_truck = trucks_collection.find_one({"truck_plate_number": truck_plate_number})
if existing_truck:
st.warning(f"Truck with Plate Number {truck_plate_number} already exists.")
else:
truck_data = {
"truck_name": truck_name,
"truck_plate_number": truck_plate_number
}
trucks_collection.insert_one(truck_data)
st.success(f"Truck {truck_name} added successfully!")
else:
st.error("Please fill in all fields.")
elif page == "Delete Truck":
st.header("Delete Truck")
trucks = list(trucks_collection.find({}, {"truck_name": 1, "truck_plate_number": 1}))
truck_names = [f"{t.get('truck_name', '')} ({t.get('truck_plate_number', '')})" for t in trucks]
selected_truck = st.selectbox("Select Truck to Delete", truck_names)
if st.button("Delete Truck"):
if selected_truck:
truck_plate_number = selected_truck.split(" (")[1][:-1]
trucks_collection.delete_one({"truck_plate_number": truck_plate_number})
st.success(f"Truck {selected_truck} deleted successfully!")
else:
st.error("Please select a truck to delete.")
elif page == "Add Driver":
st.header("Add Driver Information")
first_name = st.text_input("First Name")
last_name = st.text_input("Last Name")
contact_info = st.text_input("Contact Info")
pay_per_mile = st.number_input("Driver Pay per Mile", min_value=0.0, format="%f")
if st.button("Save Driver Info"):
if first_name and last_name and contact_info and pay_per_mile:
full_name = f"{first_name} {last_name}"
# Check if driver exists
existing_driver = drivers_collection.find_one({"contact_info": contact_info})
if existing_driver:
st.warning(f"Driver with Contact Info {contact_info} already exists.")
else:
driver_data = {
"first_name": first_name,
"last_name": last_name,
"contact_info": contact_info,
"pay_per_mile": pay_per_mile,
"trucks": [],
"loads": []
}
drivers_collection.insert_one(driver_data)
st.success(f"Driver {full_name} added successfully!")
else:
st.error("Please fill in all fields.")
elif page == "Delete Driver":
st.header("Delete Driver")
drivers = list(drivers_collection.find({}, {"first_name": 1, "last_name": 1}))
driver_names = [f"{d.get('first_name', '')} {d.get('last_name', '')}" for d in drivers]
selected_driver = st.selectbox("Select Driver to Delete", driver_names)
if st.button("Delete Driver"):
if selected_driver:
first_name, last_name = selected_driver.split()
drivers_collection.delete_one({"first_name": first_name, "last_name": last_name})
st.success(f"Driver {selected_driver} deleted successfully!")
else:
st.error("Please select a driver to delete.")
elif page == "Add Owner Operator":
st.header("Add Owner Operator")
name = st.text_input("Owner Operator Name")
contact_info = st.text_input("Contact Number")
company_percentage = st.number_input("Company Percentage", min_value=0.0, max_value=100.0, format="%f")
if st.button("Save Owner Operator Info"):
if name and contact_info and company_percentage:
# Check if owner operator exists
existing_owner_operator = owner_operators_collection.find_one({"contact_info": contact_info})
if existing_owner_operator:
st.warning(f"Owner Operator with Contact Number {contact_info} already exists.")
else:
owner_operator_data = {
"name": name,
"contact_info": contact_info,
"company_percentage": company_percentage,
"trucks": [],
"loads": []
}
owner_operators_collection.insert_one(owner_operator_data)
st.success(f"Owner Operator {name} added successfully!")
else:
st.error("Please fill in all fields.")
elif page == "Delete Owner Operator":
st.header("Delete Owner Operator")
owner_operators = list(owner_operators_collection.find({}, {"name": 1}))
owner_operator_names = [o.get('name', '') for o in owner_operators]
selected_owner_operator = st.selectbox("Select Owner Operator to Delete", owner_operator_names)
if st.button("Delete Owner Operator"):
if selected_owner_operator:
owner_operators_collection.delete_one({"name": selected_owner_operator})
st.success(f"Owner Operator {selected_owner_operator} deleted successfully!")
else:
st.error("Please select an owner operator to delete.")
elif page == "Assign Truck":
st.header("Assign Truck to Driver or Owner Operator")
drivers = list(drivers_collection.find({}, {"first_name": 1, "last_name": 1}))
owner_operators = list(owner_operators_collection.find({}, {"name": 1}))
driver_names = [f"Driver: {d.get('first_name', '')} {d.get('last_name', '')}" for d in drivers]
owner_operator_names = [f"Owner Operator: {o.get('name', '')}" for o in owner_operators]
all_names = driver_names + owner_operator_names
selected_name = st.selectbox("Select Driver or Owner Operator", all_names)
trucks = list(trucks_collection.find({}, {"truck_name": 1, "truck_plate_number": 1}))
truck_names = [f"{t.get('truck_name', '')} ({t.get('truck_plate_number', '')})" for t in trucks]
selected_truck_name = st.selectbox("Select Truck", truck_names)
if st.button("Assign Truck"):
if selected_name and selected_truck_name:
selected_truck = next(t for t in trucks if f"{t.get('truck_name', '')} ({t.get('truck_plate_number', '')})" == selected_truck_name)
truck_plate_number = selected_truck["truck_plate_number"]
if selected_name.startswith("Driver:"):
selected_driver = next(d for d in drivers if f"Driver: {d.get('first_name', '')} {d.get('last_name', '')}" == selected_name)
# Update driver record to include truck
drivers_collection.update_one(
{"_id": selected_driver["_id"]},
{"$addToSet": {"trucks": truck_plate_number}}
)
st.success(f"Truck {selected_truck_name} assigned to Driver {selected_name.split(': ')[1]} successfully!")
elif selected_name.startswith("Owner Operator:"):
selected_owner_operator = next(o for o in owner_operators if f"Owner Operator: {o.get('name', '')}" == selected_name)
# Update owner operator record to include truck
owner_operators_collection.update_one(
{"_id": selected_owner_operator["_id"]},
{"$addToSet": {"trucks": truck_plate_number}}
)
st.success(f"Truck {selected_truck_name} assigned to Owner Operator {selected_name.split(': ')[1]} successfully!")
else:
st.error("Please select both a driver/owner operator and a truck.")
elif page == "Add Load":
st.header("Add Load Information")
# Fetch drivers, owner operators, and trucks
drivers = list(drivers_collection.find({}, {"first_name": 1, "last_name": 1, "trucks": 1, "contact_info": 1, "pay_per_mile": 1}))
owner_operators = list(owner_operators_collection.find({}, {"name": 1, "trucks": 1, "contact_info": 1, "company_percentage": 1}))
driver_names = [f"Driver: {d.get('first_name', '')} {d.get('last_name', '')}" for d in drivers]
owner_operator_names = [f"Owner Operator: {o.get('name', '')}" for o in owner_operators]
all_names = driver_names + owner_operator_names
selected_names = st.multiselect("Select Driver(s) or Owner Operator(s)", all_names, max_selections=2)
if selected_names:
involved_drivers = []
involved_owner_operators = []
truck_numbers = []
for selected_name in selected_names:
if selected_name.startswith("Driver:"):
selected_driver = next(d for d in drivers if f"Driver: {d.get('first_name', '')} {d.get('last_name', '')}" == selected_name)
involved_drivers.append(selected_driver)
truck_numbers.extend(selected_driver.get('trucks', []))
elif selected_name.startswith("Owner Operator:"):
selected_owner_operator = next(o for o in owner_operators if f"Owner Operator: {o.get('name', '')}" == selected_name)
involved_owner_operators.append(selected_owner_operator)
truck_numbers.extend(selected_owner_operator.get('trucks', []))
trucks_display = [
f"{t['truck_name']} ({t['truck_plate_number']})"
for t in trucks_collection.find({"truck_plate_number": {"$in": truck_numbers}})
]
selected_truck_display = st.selectbox("Select Truck for Load", trucks_display)
if selected_truck_display:
selected_truck_no = selected_truck_display.split(" (")[1][:-1]
load_id = st.text_input("Load ID")
date_from = st.date_input("Date From", datetime.now())
date_to = st.date_input("Date To", datetime.now())
original_miles = st.number_input("Original Miles", min_value=0)
bobtail = st.number_input("Bobtail Miles", min_value=0)
payout = st.number_input("Payout", min_value=0.0, format="%f")
fuel_cost = st.number_input("Fuel Cost", min_value=0.0, format="%f")
actual_miles = original_miles - bobtail
driver_payments = []
if involved_drivers:
for driver in involved_drivers:
driver_pay_per_mile = driver.get("pay_per_mile", 0)
driver_pay = actual_miles * driver_pay_per_mile / len(involved_drivers) # Split pay between drivers
driver_payments.append(driver_pay)
if involved_owner_operators:
for owner_operator in involved_owner_operators:
company_percentage = owner_operator.get("company_percentage", 0)
pay_percent = (payout * company_percentage) / 100
driver_payments.append(pay_percent)
# Determine if the load is shared by both drivers or is separate
if len(involved_drivers) == 2:
common_load = set(involved_drivers[0].get('trucks', [])) & set(involved_drivers[1].get('trucks', []))
if common_load:
st.subheader("Shared Load Calculation")
st.write(f"Drivers {involved_drivers[0]['first_name']} {involved_drivers[0]['last_name']} and {involved_drivers[1]['first_name']} {involved_drivers[1]['last_name']} share the load.")
total_driver_pay = sum(driver_payments)
st.write(f"Total Pay for Drivers: ${total_driver_pay:.2f}")
else:
st.subheader("Separate Load Calculation")
st.write("Drivers do not share a load.")
st.write(f"Driver 1 Pay: ${driver_payments[0]:.2f}")
st.write(f"Driver 2 Pay: ${driver_payments[1]:.2f}")
else:
st.subheader("Single Driver or Owner Operator Calculation")
total_driver_pay = sum(driver_payments)
st.write(f"Total Pay for Driver/Owner Operator: ${total_driver_pay:.2f}")
if st.button("Save Load Info"):
# Check if the load ID already exists
existing_load = drivers_collection.find_one({"loads.load_id": load_id}) or owner_operators_collection.find_one({"loads.load_id": load_id})
if existing_load:
st.warning(f"Load ID {load_id} already exists. Please use a unique Load ID.")
else:
load_data = {
"truck_no": selected_truck_no,
"truck_name": selected_truck_display.split(" (")[0],
"load_id": load_id,
"date_from": date_from.strftime("%Y-%m-%d"),
"date_to": date_to.strftime("%Y-%m-%d"),
"original_miles": original_miles,
"bobtail": bobtail,
"actual_miles": actual_miles,
"payout": payout,
"fuel_cost": fuel_cost,
"driver_pay": total_driver_pay
}
# Assign the load to each driver involved
for driver in involved_drivers:
drivers_collection.update_one(
{"_id": driver["_id"]},
{"$push": {"loads": load_data}}
)
# Assign the load to each owner operator involved
for owner_operator in involved_owner_operators:
owner_operators_collection.update_one(
{"_id": owner_operator["_id"]},
{"$push": {"loads": load_data}}
)
st.success(f"Load information added successfully for the selected drivers/owner operators with Truck {selected_truck_display.split(' (')[0]}!")
elif page == "Update/Delete Load":
st.header("Update or Delete Load")
load_id = st.text_input("Enter Load ID to Update or Delete")
if load_id:
# Search for the load across all drivers and owner operators
driver_record = drivers_collection.find_one({"loads.load_id": load_id}, {"loads.$": 1, "first_name": 1, "last_name": 1})
owner_operator_record = owner_operators_collection.find_one({"loads.load_id": load_id}, {"loads.$": 1, "name": 1})
if driver_record:
load_data = driver_record["loads"][0]
entity_type = "driver"
entity_name = f"{driver_record['first_name']} {driver_record['last_name']}"
elif owner_operator_record:
load_data = owner_operator_record["loads"][0]
entity_type = "owner_operator"
entity_name = owner_operator_record["name"]
else:
load_data = None
if load_data:
st.write(f"Load found for {entity_type.capitalize()}: {entity_name}")
# Display load information
load_data_display = {
"Truck No": load_data["truck_no"],
"Load ID": load_data["load_id"],
"Date From": load_data["date_from"],
"Date To": load_data["date_to"],
"Original Miles": load_data["original_miles"],
"Bobtail Miles": load_data["bobtail"],
"Payout": load_data["payout"],
"Fuel Cost": load_data["fuel_cost"],
"Driver/Owner Operator Pay": load_data["driver_pay"]
}
for key, value in load_data_display.items():
st.write(f"{key}: {value}")
# Input fields for updating the load
st.subheader("Update Load Information")
date_from = st.date_input("Date From", datetime.strptime(load_data["date_from"], "%Y-%m-%d"))
date_to = st.date_input("Date To", datetime.strptime(load_data["date_to"], "%Y-%m-%d"))
original_miles = st.number_input("Original Miles", value=load_data["original_miles"], min_value=0)
bobtail = st.number_input("Bobtail Miles", value=load_data["bobtail"], min_value=0)
payout = st.number_input("Payout", value=load_data["payout"], min_value=0.0, format="%f")
fuel_cost = st.number_input("Fuel Cost", value=load_data["fuel_cost"], min_value=0.0, format="%f")
actual_miles = original_miles - bobtail
if entity_type == "driver":
driver_pay = actual_miles * driver_record["loads"][0]["driver_pay"] / load_data["actual_miles"]
elif entity_type == "owner_operator":
company_percentage = owner_operator_record["loads"][0].get("company_percentage", 0)
pay_percent = (payout * company_percentage) / 100
driver_pay = pay_percent
if st.button("Update Load"):
# Update the load with new values
updated_load_data = {
"loads.$.date_from": date_from.strftime("%Y-%m-%d"),
"loads.$.date_to": date_to.strftime("%Y-%m-%d"),
"loads.$.original_miles": original_miles,
"loads.$.bobtail": bobtail,
"loads.$.actual_miles": actual_miles,
"loads.$.payout": payout,
"loads.$.fuel_cost": fuel_cost,
"loads.$.driver_pay": driver_pay
}
if entity_type == "driver":
drivers_collection.update_one(
{"_id": driver_record["_id"], "loads.load_id": load_id},
{"$set": updated_load_data}
)
st.success(f"Load ID {load_id} updated successfully!")
elif entity_type == "owner_operator":
owner_operators_collection.update_one(
{"_id": owner_operator_record["_id"], "loads.load_id": load_id},
{"$set": updated_load_data}
)
st.success(f"Load ID {load_id} updated successfully!")
if st.button("Delete Load"):
# Remove the load from the driver's or owner operator's record
if entity_type == "driver":
drivers_collection.update_one(
{"_id": driver_record["_id"]},
{"$pull": {"loads": {"load_id": load_id}}}
)
st.success(f"Load ID {load_id} deleted successfully!")
elif entity_type == "owner_operator":
owner_operators_collection.update_one(
{"_id": owner_operator_record["_id"]},
{"$pull": {"loads": {"load_id": load_id}}}
)
st.success(f"Load ID {load_id} deleted successfully!")
else:
st.warning(f"No load found with Load ID {load_id}.")
elif page == "View Owner Operator Info":
st.header("View Owner Operator Information")
owner_operators = list(owner_operators_collection.find({}))
owner_operator_names = [o.get('name', '') for o in owner_operators]
selected_owner_operator_name = st.selectbox("Select Owner Operator to View", owner_operator_names)
if selected_owner_operator_name:
owner_operator_record = owner_operators_collection.find_one({"name": selected_owner_operator_name})
if owner_operator_record:
st.write(f"**Name**: {owner_operator_record['name']}")
st.write(f"**Contact Number**: {owner_operator_record['contact_info']}")
st.write(f"**Company Percentage**: {owner_operator_record['company_percentage']}%")
st.write(f"**Trucks Assigned**: {', '.join(owner_operator_record['trucks']) if owner_operator_record['trucks'] else 'None'}")
# Date range input
date_from = st.date_input("Date From", datetime.now())
date_to = st.date_input("Date To", datetime.now())
# Convert the date input from Streamlit to Pandas datetime
date_from = pd.to_datetime(date_from)
date_to = pd.to_datetime(date_to)
# Display loads associated with this owner operator within the date range
if "loads" in owner_operator_record and owner_operator_record["loads"]:
load_df = pd.DataFrame(owner_operator_record["loads"])
load_df["date_from"] = pd.to_datetime(load_df["date_from"])
load_df["date_to"] = pd.to_datetime(load_df["date_to"])
# Filter by date range
filtered_load_df = load_df[(load_df["date_from"] >= date_from) & (load_df["date_to"] <= date_to)]
if not filtered_load_df.empty:
st.subheader("Load Information")
st.dataframe(filtered_load_df)
# Total Calculations
total_payout = filtered_load_df["payout"].sum()
total_company_fee = filtered_load_df["driver_pay"].sum()
total_fuel_cost = filtered_load_df["fuel_cost"].sum()
st.subheader("Total Calculations")
st.write(f"Total Payout: ${total_payout:.2f}")
st.write(f"Total Company Fee: ${total_company_fee:.2f}")
st.write(f"Total Fuel Cost: ${total_fuel_cost:.2f}")
else:
st.write("No loads found in the selected date range.")
else:
st.write("No loads associated with this Owner Operator.")
elif page == "View Information":
st.header("View Load Information by Truck Name")
# Fetch truck details
trucks = list(trucks_collection.find({}, {"truck_name": 1, "truck_plate_number": 1}))
truck_names = [f"{t.get('truck_name', '')} ({t.get('truck_plate_number', '')})" for t in trucks]
# Dropdown to select a truck
selected_truck_name_view = st.selectbox("Select Truck Name to View Records", truck_names)
if selected_truck_name_view:
truck_record = trucks_collection.find_one({"truck_name": selected_truck_name_view.split(" ")[0]})
if truck_record:
truck_plate_number = truck_record["truck_plate_number"]
# Date range inputs
date_from = st.date_input("Date From", datetime.now())
date_to = st.date_input("Date To", datetime.now())
# Convert date input to strings for MongoDB query
date_from_str = date_from.strftime("%Y-%m-%d")
date_to_str = date_to.strftime("%Y-%m-%d")
# Aggregate load records for the selected truck and date range from both drivers and owner_operators collections
driver_loads = list(drivers_collection.aggregate([
{"$match": {"trucks": truck_plate_number}},
{"$unwind": "$loads"},
{"$match": {
"loads.truck_no": truck_plate_number,
"loads.date_from": {"$gte": date_from_str},
"loads.date_to": {"$lte": date_to_str}
}},
{"$group": {
"_id": "$loads.load_id",
"load_data": {"$first": "$loads"},
"drivers": {"$push": {"first_name": "$first_name", "last_name": "$last_name"}}
}}
]))
# Check if there are loads shared between drivers
if driver_loads:
# Prepare data for table display
table_data = []
for load in driver_loads:
load_data = load['load_data']
driver_names = [f"{d['first_name']} {d['last_name']}" for d in load['drivers']]
table_data.append({
"Load ID": load['_id'],
"Truck No": load_data['truck_no'],
"Truck Name": load_data['truck_name'],
"Date From": load_data['date_from'],
"Date To": load_data['date_to'],
"Original Miles": load_data['original_miles'],
"Bobtail Miles": load_data['bobtail'],
"Actual Miles": load_data['actual_miles'],
"Payout": f"${load_data['payout']:.2f}",
"Fuel Cost": f"${load_data['fuel_cost']:.2f}",
"Driver Pay": f"${load_data['driver_pay']:.2f}",
"Drivers Involved": ', '.join(driver_names)
})
# Display the table using a DataFrame
df = pd.DataFrame(table_data)
st.subheader("Load Information")
st.dataframe(df) # Use st.table(df) for a static table
else:
st.warning(f"No load records found for Truck {selected_truck_name_view.split(' ')[0]} between {date_from} and {date_to}.")