Update app.py

app.py

@@ -1,20 +1,57 @@
 import streamlit as st
+import sqlite3
+from datetime import datetime
 from reportlab.lib.pagesizes import A4
 from reportlab.pdfgen import canvas
-import datetime
 
-# Function to create PDF Work Permit
+# Initialize Database
+conn = sqlite3.connect("work_permits.db")
+cursor = conn.cursor()
+cursor.execute("""
+    CREATE TABLE IF NOT EXISTS permits (
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        permit_number TEXT UNIQUE,
+        permit_type TEXT,
+        job_description TEXT,
+        location TEXT,
+        start_date TEXT,
+        end_date TEXT,
+        equipment TEXT,
+        worker_name TEXT,
+        supervisor_name TEXT,
+        contact_number TEXT,
+        safety_checks TEXT,
+        generated_on TEXT
+    )
+""")
+conn.commit()
+
+# Function to generate Permit Number
+def generate_permit_number():
+    today = datetime.now().strftime("%Y%m%d")
+    cursor.execute("SELECT COUNT(*) FROM permits WHERE generated_on LIKE ?", (f"%{today}%",))
+    count = cursor.fetchone()[0] + 1
+    return f"WP-{today}-{str(count).zfill(3)}"
+
+# Function to save permit to Database
+def save_to_db(data):
+    cursor.execute("""
+        INSERT INTO permits (permit_number, permit_type, job_description, location, start_date, end_date, 
+                             equipment, worker_name, supervisor_name, contact_number, safety_checks, generated_on) 
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+    """, data)
+    conn.commit()
+
+# Function to create PDF
 def generate_pdf(data):
-    pdf_filename = "work_permit.pdf"
+    pdf_filename = f"{data[0]}.pdf"
     c = canvas.Canvas(pdf_filename, pagesize=A4)
     c.setFont("Helvetica", 12)
 
-    # Title
-    c.drawString(200, 800, "WORK PERMIT FORM")
-
-    # Permit Details
-    y = 780
-    for key, value in data.items():
+    y = 800
+    for key, value in zip(["Permit Number", "Permit Type", "Job Description", "Location", "Start Date", 
+                           "End Date", "Equipment", "Worker Name", "Supervisor Name", "Contact Number", 
+                           "Safety Checks", "Generated On"], data):
         c.drawString(50, y, f"{key}: {value}")
         y -= 20
 
@@ -22,14 +59,14 @@ def generate_pdf(data):
     return pdf_filename
 
 # Streamlit UI
-st.title("🏗️ Work Permit System")
+st.title("🏗️ Work Permit Management System")
 
-# Select permit type
+# Select Permit Type
 permit_types = [
-    "Hot Work (Red)", "Cold Work (Blue)", "Electrical Work (Orange/Yellow)", 
-    "Confined Space Entry (Green)", "Working at Heights (Yellow)", 
-    "Lockout/Tagout (White/Red)", "Excavation/Trenching (Brown)", 
-    "Radiation Work (Purple)", "Chemical Handling (Orange/Yellow)", 
+    "Hot Work (Red)", "Cold Work (Blue)", "Electrical Work (Orange/Yellow)",
+    "Confined Space Entry (Green)", "Working at Heights (Yellow)",
+    "Lockout/Tagout (White/Red)", "Excavation/Trenching (Brown)",
+    "Radiation Work (Purple)", "Chemical Handling (Orange/Yellow)",
     "Marine Work (Blue/Green)"
 ]
 permit_type = st.selectbox("Select Work Permit Type:", permit_types)
@@ -48,37 +85,56 @@ contact_number = st.text_input("Emergency Contact Number")
 
 # Safety Checklist
 st.subheader("✅ Safety Checklist")
-ppe = st.checkbox("PPE Worn")
-hazard_inspection = st.checkbox("Area Inspected for Hazards")
-fire_extinguisher = st.checkbox("Fire Extinguisher Available")
-loto = st.checkbox("Lockout/Tagout Applied (if needed)")
-atmospheric_testing = st.checkbox("Atmospheric Testing Completed (for confined spaces)")
-
-# Submit Button
-if st.button("Generate Work Permit PDF"):
-    permit_data = {
-        "Permit Type": permit_type,
-        "Job Description": job_description,
-        "Work Location": location,
-        "Start Date": start_date,
-        "End Date": end_date,
-        "Equipment Used": equipment,
-        "Worker Name": worker_name,
-        "Supervisor Name": supervisor_name,
-        "Emergency Contact": contact_number,
-        "PPE Worn": "Yes" if ppe else "No",
-        "Hazard Inspection": "Yes" if hazard_inspection else "No",
-        "Fire Extinguisher Available": "Yes" if fire_extinguisher else "No",
-        "Lockout/Tagout Applied": "Yes" if loto else "No",
-        "Atmospheric Testing Done": "Yes" if atmospheric_testing else "No",
-        "Generated On": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    }
+safety_checks = []
+if st.checkbox("PPE Worn"):
+    safety_checks.append("PPE Worn")
+if st.checkbox("Hazard Inspection Done"):
+    safety_checks.append("Hazard Inspection Done")
+if st.checkbox("Fire Extinguisher Available"):
+    safety_checks.append("Fire Extinguisher Available")
+if st.checkbox("Lockout/Tagout Applied"):
+    safety_checks.append("Lockout/Tagout Applied")
+if st.checkbox("Atmospheric Testing Completed"):
+    safety_checks.append("Atmospheric Testing Completed")
+
+# Generate Work Permit
+if st.button("Generate Work Permit"):
+    permit_number = generate_permit_number()
+    generated_on = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+    # Save to database
+    permit_data = (permit_number, permit_type, job_description, location, str(start_date), str(end_date),
+                   equipment, worker_name, supervisor_name, contact_number, ", ".join(safety_checks), generated_on)
+    save_to_db(permit_data)
 
     # Generate PDF
     pdf_file = generate_pdf(permit_data)
-    
+
     # Provide Download Link
     with open(pdf_file, "rb") as file:
-        st.download_button(label="📥 Download Work Permit", data=file, file_name="work_permit.pdf", mime="application/pdf")
+        st.download_button(label="📥 Download Work Permit", data=file, file_name=f"{permit_number}.pdf", mime="application/pdf")
+
+    st.success(f"✅ Work Permit {permit_number} generated successfully!")
+
+# View Saved Permits
+st.subheader("📋 View Saved Work Permits")
+search_type = st.radio("Search by:", ("Date", "Permit Number"))
+if search_type == "Date":
+    search_date = st.date_input("Select Date")
+    cursor.execute("SELECT * FROM permits WHERE generated_on LIKE ?", (f"%{search_date}%",))
+elif search_type == "Permit Number":
+    permit_no = st.text_input("Enter Permit Number")
+    cursor.execute("SELECT * FROM permits WHERE permit_number = ?", (permit_no,))
+    
+results = cursor.fetchall()
+if results:
+    for permit in results:
+        st.write(f"📄 **Permit Number:** {permit[1]}")
+        st.write(f"📌 **Type:** {permit[2]}")
+        st.write(f"📍 **Location:** {permit[4]}")
+        st.write(f"📅 **Date:** {permit[11]}")
+        st.write("---")
+else:
+    st.warning("No permits found.")
 
-st.success("Work Permit System Ready! ✅")
+st.success("Work Permit System Ready ✅")