Update app.py

app.py

@@ -1,212 +1,399 @@
 import gradio as gr
 import pandas as pd
-from datetime import datetime, timedelta
 import os
+from datetime import datetime, timedelta
 
 
-# ===============================
-# FILE SETTINGS
-# ===============================
-DATA_FILE = "permissions.xlsx"
+# ---------------- CONFIG ---------------- #
+
 STAFF_FILE = "staff_details.xlsx"
+PERMISSION_FILE = "permissions.xlsx"
+
+ADMIN_USER = "admin"
+ADMIN_PASS = "admin123"
 
 
-# ===============================
-# CREATE FILES IF NOT EXISTS
-# ===============================
-def create_files():
+# ---------------- INIT ---------------- #
 
-    if not os.path.exists(DATA_FILE):
-        df = pd.DataFrame(columns=[
-            "Staff ID",
-            "Name",
-            "Department",
-            "Date",
-            "From Time",
-            "To Time",
-            "Reason",
-            "Applied On"
-        ])
-        df.to_excel(DATA_FILE, index=False)
+if not os.path.exists(PERMISSION_FILE):
+    df = pd.DataFrame(columns=[
+        "StaffID",
+        "Name",
+        "Department",
+        "Date",
+        "FromTime",
+        "ToTime",
+        "Reason",
+        "SubmittedOn"
+    ])
+    df.to_excel(PERMISSION_FILE, index=False)
 
-    if not os.path.exists(STAFF_FILE):
-        staff = pd.DataFrame({
-            "Staff ID": ["SRC001", "SRC002"],
-            "Name": ["Ramesh", "Suresh"],
-            "Department": ["CSE", "IT"]
-        })
-        staff.to_excel(STAFF_FILE, index=False)
 
+# ---------------- LOADERS ---------------- #
 
-create_files()
+def load_staff():
+    return pd.read_excel(STAFF_FILE)
 
 
-# ===============================
-# TIME CONVERTER
-# ===============================
-def convert_time(time_str):
-    return datetime.strptime(time_str.strip(), "%I:%M %p")
+def load_permissions():
+    return pd.read_excel(PERMISSION_FILE)
 
 
-# ===============================
-# VALIDATION FUNCTION
-# ===============================
-def validate_and_save(staff_id, name, dept, date, from_time, to_time, reason):
+# ---------------- VALIDATORS ---------------- #
 
-    # -------------------------------
-    # Empty Validation
-    # -------------------------------
-    if not all([staff_id, name, dept, date, from_time, to_time, reason]):
-        return "❌ All fields are required"
+# Date: Only Today / Tomorrow / Day after Tomorrow
+def validate_date_range(date_str):
 
+    try:
+        d = datetime.strptime(date_str, "%d/%m/%Y").date()
+    except:
+        return False, "❌ Date format must be DD/MM/YYYY"
 
-    # -------------------------------
-    # Date Validation
-    # -------------------------------
     today = datetime.today().date()
-    selected_date = datetime.strptime(date, "%Y-%m-%d").date()
+    max_day = today + timedelta(days=2)
 
-    allowed_dates = [
-        today,
-        today + timedelta(days=1),
-        today + timedelta(days=2)
-    ]
+    if d < today:
+        return False, "❌ Past date not allowed"
+
+    if d > max_day:
+        return False, "❌ Apply only within 2 days"
 
-    if selected_date not in allowed_dates:
-        return "❌ You can apply only for Today / Tomorrow / Day After Tomorrow"
+    return True, ""
 
 
-    # -------------------------------
-    # Time Validation
-    # -------------------------------
+# Time: 08:45 - 17:20 and exactly 1 hour
+def validate_time_range(f, t):
+
     try:
-        start = convert_time(from_time)
-        end = convert_time(to_time)
+        f_time = datetime.strptime(f, "%H:%M")
+        t_time = datetime.strptime(t, "%H:%M")
     except:
-        return "❌ Time must be in format: HH:MM AM/PM (Ex: 10:00 AM)"
+        return False, "❌ Time format must be HH:MM (24-hour)"
+
+    min_time = datetime.strptime("08:45", "%H:%M")
+    max_time = datetime.strptime("17:20", "%H:%M")
 
+    # Check range
+    if f_time < min_time or t_time > max_time:
+        return False, "❌ Time allowed: 08:45 to 17:20 only"
 
-    # -------------------------------
-    # Office Time Limits
-    # -------------------------------
-    office_start = datetime.strptime("08:45 AM", "%I:%M %p")
-    office_end = datetime.strptime("05:20 PM", "%I:%M %p")
+    # Check 1 hour duration
+    diff = (t_time - f_time).total_seconds() / 3600
 
+    if diff != 1:
+        return False, "❌ Permission must be exactly 1 hour"
 
-    if start < office_start or end > office_end:
-        return "❌ Time must be between 8:45 AM and 5:20 PM"
+    return True, ""
 
 
-    # -------------------------------
-    # One Hour Validation
-    # -------------------------------
-    diff = (end - start).seconds / 3600
+# ---------------- STAFF FUNCTIONS ---------------- #
+
+def fetch_staff(staff_id):
+
+    df = load_staff()
+    row = df[df["StaffID"] == staff_id]
+
+    if row.empty:
+        return "Not Found", "Not Found", "0"
+
+    name = row.iloc[0]["Name"]
+    dept = row.iloc[0]["Department"]
+
+    perms = load_permissions()
+    count = len(perms[perms["StaffID"] == staff_id])
+
+    return name, dept, str(count)
+
+
+def submit_permission(staff_id, date, f, t, reason):
+
+    # Basic validation
+    if staff_id.strip() == "":
+        return "❌ Staff ID Required"
+
+    if date.strip() == "":
+        return "❌ Date Required"
+
+    if f.strip() == "" or t.strip() == "":
+        return "❌ Time Required"
+
+    if reason.strip() == "":
+        return "❌ Reason Required"
+
+
+    # Date validation
+    ok, msg = validate_date_range(date)
+    if not ok:
+        return msg
+
+
+    # Time validation
+    ok, msg = validate_time_range(f, t)
+    if not ok:
+        return msg
+
+
+    # Staff validation
+    staff = load_staff()
+    row = staff[staff["StaffID"] == staff_id]
+
+    if row.empty:
+        return "❌ Invalid Staff ID"
 
-    if diff != 1:
-        return "❌ Permission allowed for exactly 1 hour only"
 
+    perms = load_permissions()
 
-    # -------------------------------
-    # Save to Excel
-    # -------------------------------
-    df = pd.read_excel(DATA_FILE)
 
-    new_data = {
-        "Staff ID": staff_id,
+    # Same day check (max 2)
+    same_day = perms[
+        (perms["StaffID"] == staff_id) &
+        (perms["Date"] == date)
+    ]
+
+    if len(same_day) == 1:
+        return "⚠️ Second permission today → Apply Leave"
+
+    if len(same_day) >= 2:
+        return "❌ Daily limit reached"
+
+
+    # Monthly limit (2 only)
+    m = date.split("/")[1]
+    y = date.split("/")[2]
+
+    same_month = perms[
+        (perms["StaffID"] == staff_id) &
+        (perms["Date"].str.contains(f"/{m}/{y}", na=False))
+    ]
+
+    if len(same_month) >= 2:
+        return "❌ Monthly limit reached (2 only)"
+
+
+    # Save
+    name = row.iloc[0]["Name"]
+    dept = row.iloc[0]["Department"]
+
+    now = datetime.now().strftime("%d/%m/%Y %H:%M")
+
+    new_row = {
+        "StaffID": staff_id,
         "Name": name,
         "Department": dept,
         "Date": date,
-        "From Time": from_time,
-        "To Time": to_time,
+        "FromTime": f,
+        "ToTime": t,
         "Reason": reason,
-        "Applied On": datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        "SubmittedOn": now
     }
 
-    df = pd.concat([df, pd.DataFrame([new_data])], ignore_index=True)
-    df.to_excel(DATA_FILE, index=False)
+    perms = pd.concat([perms, pd.DataFrame([new_row])], ignore_index=True)
+    perms.to_excel(PERMISSION_FILE, index=False)
 
-    return "✅ Permission Applied Successfully"
+    return "✅ Permission Submitted Successfully"
 
 
-# ===============================
-# GRADIO UI
-# ===============================
-with gr.Blocks(theme=gr.themes.Soft()) as app:
+def delete_permission(staff_id, date):
+
+    perms = load_permissions()
+    before = len(perms)
+
+    perms = perms[
+        ~((perms["StaffID"] == staff_id) &
+          (perms["Date"] == date))
+    ]
+
+    perms.to_excel(PERMISSION_FILE, index=False)
+
+    if len(perms) == before:
+        return "❌ No Record Found"
+    else:
+        return "✅ Deleted"
+
+
+def clear_form():
+    return "", "", "", "", ""
+
+
+# ---------------- ADMIN ---------------- #
+
+def admin_login(user, pwd):
+
+    if user == ADMIN_USER and pwd == ADMIN_PASS:
+        return gr.update(visible=True), "✅ Login Successful"
+
+    return gr.update(visible=False), "❌ Invalid Login"
+
+
+def get_all_data():
+    return load_permissions()
+
+
+def download_excel():
+    return PERMISSION_FILE
 
-    # ---------- STYLE ----------
-    gr.HTML("""
-    <style>
-        .center {
-            text-align: center;
-        }
-    </style>
-    """)
 
-    # ---------- HEADER ----------
+def monthly_report(month, year):
+
+    perms = load_permissions()
+
+    return perms[
+        perms["Date"].str.contains(f"/{month}/{year}", na=False)
+    ]
+
+
+def dept_report(dept):
+
+    perms = load_permissions()
+
+    return perms[perms["Department"] == dept]
+
+
+def reset_data():
+
+    df = pd.DataFrame(columns=[
+        "StaffID",
+        "Name",
+        "Department",
+        "Date",
+        "FromTime",
+        "ToTime",
+        "Reason",
+        "SubmittedOn"
+    ])
+
+    df.to_excel(PERMISSION_FILE, index=False)
+
+    return "✅ All Data Cleared"
+
+
+# ================= UI ================= #
+
+with gr.Blocks() as app:
+
     gr.HTML("""
-    <div class="center">
-        <img src="file/logo.jpg" width="120"><br><br>
+    <center>
         <h2>SRC, SASTRA</h2>
         <h3>Staff Permission Management System</h3>
         <hr>
-    </div>
+    </center>
     """)
 
 
-    # ---------- FORM ----------
-    with gr.Row():
-        with gr.Column(scale=1):
+    # ---------------- STAFF TAB ---------------- #
+
+    with gr.Tab("Staff Panel"):
+
+        with gr.Row():
+
+            with gr.Column():
+
+                staff_id = gr.Textbox(label="Staff ID")
+
+                name = gr.Textbox(label="Name", interactive=False)
+                dept = gr.Textbox(label="Department", interactive=False)
+                count = gr.Textbox(label="Permission Count", interactive=False)
+
+                fetch = gr.Button("Fetch Details")
+
+
+            with gr.Column():
+
+                date = gr.Textbox(label="Date (DD/MM/YYYY)")
+                f = gr.Textbox(label="From (HH:MM 24-hr)")
+                t = gr.Textbox(label="To (HH:MM 24-hr)")
+
+                reason = gr.Textbox(label="Reason", lines=3)
+
+                with gr.Row():
+                    submit = gr.Button("Submit", variant="primary")
+                    delete = gr.Button("Delete", variant="stop")
+                    clear = gr.Button("Clear")
+
+                status = gr.Textbox(label="Status", interactive=False)
+
+
+    # ---------------- ADMIN TAB ---------------- #
+
+    with gr.Tab("Admin Panel"):
 
-            staff_id = gr.Textbox(label="Staff ID")
-            name = gr.Textbox(label="Name")
-            dept = gr.Textbox(label="Department")
+        admin_user = gr.Textbox(label="Username")
+        admin_pwd = gr.Textbox(label="Password", type="password")
 
-            date = gr.Textbox(
-                label="Date (YYYY-MM-DD)",
-                placeholder="2026-02-15"
-            )
+        login = gr.Button("Login")
 
-            from_time = gr.Textbox(
-                label="From Time (HH:MM AM/PM)",
-                placeholder="10:00 AM"
-            )
+        msg = gr.Textbox(interactive=False)
 
-            to_time = gr.Textbox(
-                label="To Time (HH:MM AM/PM)",
-                placeholder="11:00 AM"
-            )
+        admin_box = gr.Column(visible=False)
 
-            reason = gr.Textbox(
-                label="Reason",
-                lines=3
-            )
+        with admin_box:
 
-            submit = gr.Button("Apply Permission")
+            gr.Markdown("### 📊 All Records")
 
-            result = gr.Textbox(
-                label="Status",
-                interactive=False
-            )
+            table = gr.Dataframe(interactive=False)
 
+            refresh = gr.Button("Refresh")
+
+
+            gr.Markdown("### 📅 Monthly Report")
+
+            month = gr.Textbox(label="Month (MM)")
+            year = gr.Textbox(label="Year (YYYY)")
+
+            month_btn = gr.Button("Generate")
+
+            month_table = gr.Dataframe()
+
+
+            gr.Markdown("### 🏢 Department Report")
+
+            dept_name = gr.Textbox(label="Department")
+
+            dept_btn = gr.Button("Generate")
+
+            dept_table = gr.Dataframe()
+
+
+            gr.Markdown("### 📥 Download")
+
+            down = gr.Button("Download Excel")
+
+            file = gr.File()
+
+
+            gr.Markdown("### ⚠️ Reset")
+
+            reset = gr.Button("Reset All", variant="stop")
+
+            reset_msg = gr.Textbox(interactive=False)
+
+
+    # ---------------- EVENTS ---------------- #
+
+    fetch.click(fetch_staff, staff_id, [name, dept, count])
 
-    # ---------- BUTTON ----------
     submit.click(
-        validate_and_save,
-        inputs=[
-            staff_id,
-            name,
-            dept,
-            date,
-            from_time,
-            to_time,
-            reason
-        ],
-        outputs=result
+        submit_permission,
+        [staff_id, date, f, t, reason],
+        status
     )
 
+    delete.click(delete_permission, [staff_id, date], status)
+
+    clear.click(clear_form, outputs=[date, f, t, reason, status])
+
+
+    login.click(admin_login, [admin_user, admin_pwd], [admin_box, msg])
+
+    refresh.click(get_all_data, outputs=table)
+
+    month_btn.click(monthly_report, [month, year], month_table)
+
+    dept_btn.click(dept_report, dept_name, dept_table)
+
+    down.click(download_excel, outputs=file)
+
+    reset.click(reset_data, outputs=reset_msg)
+
 
-# ===============================
-# LAUNCH
-# ===============================
-if __name__ == "__main__":
-    app.launch()
+app.launch()