Update app.py

app.py

@@ -0,0 +1,128 @@
+import streamlit as st
+import pandas as pd
+import sqlite3
+from datetime import datetime
+
+# Database connection
+conn = sqlite3.connect('cafm.db', check_same_thread=False)
+c = conn.cursor()
+
+# Database initialization
+def create_tables():
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS assets (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            site_name TEXT,
+            asset_name TEXT,
+            asset_type TEXT,
+            maintenance_history TEXT,
+            warranty TEXT,
+            current_condition TEXT
+        )
+    ''')
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS work_orders (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            site_name TEXT,
+            work_order_description TEXT,
+            assigned_to TEXT,
+            status TEXT,
+            created_at TEXT
+        )
+    ''')
+    c.execute('''
+        CREATE TABLE IF NOT EXISTS preventive_maintenance (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            site_name TEXT,
+            task_name TEXT,
+            schedule_date TEXT,
+            status TEXT
+        )
+    ''')
+
+create_tables()
+
+# Sidebar Navigation
+st.sidebar.title("CAFM System")
+menu = st.sidebar.selectbox("Menu", ["Asset Management", "Work Order Management", "Preventive Maintenance", "Reporting & Analytics"])
+
+# Asset Management
+if menu == "Asset Management":
+    st.title("Asset Management")
+    site_name = st.text_input("Site Name")
+    asset_name = st.text_input("Asset Name")
+    asset_type = st.text_input("Asset Type")
+    maintenance_history = st.text_area("Maintenance History")
+    warranty = st.text_input("Warranty (e.g., 1 year)")
+    current_condition = st.selectbox("Current Condition", ["Good", "Needs Repair", "Critical"])
+
+    if st.button("Add Asset"):
+        c.execute('''
+            INSERT INTO assets (site_name, asset_name, asset_type, maintenance_history, warranty, current_condition)
+            VALUES (?, ?, ?, ?, ?, ?)
+        ''', (site_name, asset_name, asset_type, maintenance_history, warranty, current_condition))
+        conn.commit()
+        st.success("Asset added successfully!")
+
+    st.subheader("Asset List")
+    assets = pd.read_sql_query("SELECT * FROM assets", conn)
+    st.dataframe(assets)
+
+# Work Order Management
+elif menu == "Work Order Management":
+    st.title("Work Order Management")
+    site_name = st.text_input("Site Name")
+    work_order_description = st.text_area("Work Order Description")
+    assigned_to = st.text_input("Assigned To")
+    status = st.selectbox("Status", ["Pending", "In Progress", "Completed"])
+
+    if st.button("Create Work Order"):
+        c.execute('''
+            INSERT INTO work_orders (site_name, work_order_description, assigned_to, status, created_at)
+            VALUES (?, ?, ?, ?, ?)
+        ''', (site_name, work_order_description, assigned_to, status, datetime.now().strftime("%Y-%m-%d %H:%M:%S")))
+        conn.commit()
+        st.success("Work order created successfully!")
+
+    st.subheader("Work Orders")
+    work_orders = pd.read_sql_query("SELECT * FROM work_orders", conn)
+    st.dataframe(work_orders)
+
+# Preventive Maintenance
+elif menu == "Preventive Maintenance":
+    st.title("Preventive Maintenance")
+    site_name = st.text_input("Site Name")
+    task_name = st.text_input("Task Name")
+    schedule_date = st.date_input("Schedule Date")
+    status = st.selectbox("Status", ["Scheduled", "Completed"])
+
+    if st.button("Schedule Task"):
+        c.execute('''
+            INSERT INTO preventive_maintenance (site_name, task_name, schedule_date, status)
+            VALUES (?, ?, ?, ?)
+        ''', (site_name, task_name, schedule_date, status))
+        conn.commit()
+        st.success("Task scheduled successfully!")
+
+    st.subheader("Scheduled Maintenance")
+    maintenance = pd.read_sql_query("SELECT * FROM preventive_maintenance", conn)
+    st.dataframe(maintenance)
+
+# Reporting & Analytics
+elif menu == "Reporting & Analytics":
+    st.title("Reporting & Analytics")
+    st.subheader("Asset Overview")
+    asset_data = pd.read_sql_query("SELECT site_name, COUNT(*) as total_assets FROM assets GROUP BY site_name", conn)
+    st.dataframe(asset_data)
+
+    st.subheader("Work Order Summary")
+    work_order_data = pd.read_sql_query("SELECT site_name, COUNT(*) as total_work_orders FROM work_orders GROUP BY site_name", conn)
+    st.dataframe(work_order_data)
+
+    st.subheader("Maintenance Tasks Overview")
+    maintenance_data = pd.read_sql_query("SELECT site_name, COUNT(*) as total_tasks FROM preventive_maintenance GROUP BY site_name", conn)
+    st.dataframe(maintenance_data)
+
+# Close database connection
+conn.close()
+