format

app.py

@@ -1,13 +1,17 @@
+# app.py
+# Staff Permission Management System (HF + Gradio Ready)
+
 import gradio as gr
 import pandas as pd
 import os
-from datetime import datetime, timedelta
+from datetime import datetime
 
 
 # ---------------- CONFIG ---------------- #
 
 STAFF_FILE = "staff_details.xlsx"
 PERMISSION_FILE = "permissions.xlsx"
+LOGO_FILE = "logo.jpg"
 
 ADMIN_USER = "admin"
 ADMIN_PASS = "admin123"
@@ -16,6 +20,7 @@ ADMIN_PASS = "admin123"
 # ---------------- INIT ---------------- #
 
 if not os.path.exists(PERMISSION_FILE):
+
     df = pd.DataFrame(columns=[
         "StaffID",
         "Name",
@@ -26,6 +31,7 @@ if not os.path.exists(PERMISSION_FILE):
         "Reason",
         "SubmittedOn"
     ])
+
     df.to_excel(PERMISSION_FILE, index=False)
 
 
@@ -41,49 +47,22 @@ def load_permissions():
 
 # ---------------- VALIDATORS ---------------- #
 
-# Date: Only Today / Tomorrow / Day after Tomorrow
-def validate_date_range(date_str):
+def validate_date(d):
 
     try:
-        d = datetime.strptime(date_str, "%d/%m/%Y").date()
+        datetime.strptime(d, "%d/%m/%Y")
+        return True
     except:
-        return False, "❌ Date format must be DD/MM/YYYY"
-
-    today = datetime.today().date()
-    max_day = today + timedelta(days=2)
-
-    if d < today:
-        return False, "❌ Past date not allowed"
-
-    if d > max_day:
-        return False, "❌ Apply only within 2 days"
-
-    return True, ""
+        return False
 
 
-# Time: 08:45 - 17:20 and exactly 1 hour
-def validate_time_range(f, t):
+def validate_time(t):
 
     try:
-        f_time = datetime.strptime(f, "%H:%M")
-        t_time = datetime.strptime(t, "%H:%M")
+        datetime.strptime(t, "%H:%M")
+        return True
     except:
-        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"
-
-    # Check 1 hour duration
-    diff = (t_time - f_time).total_seconds() / 3600
-
-    if diff != 1:
-        return False, "❌ Permission must be exactly 1 hour"
-
-    return True, ""
+        return False
 
 
 # ---------------- STAFF FUNCTIONS ---------------- #
@@ -91,6 +70,7 @@ def validate_time_range(f, t):
 def fetch_staff(staff_id):
 
     df = load_staff()
+
     row = df[df["StaffID"] == staff_id]
 
     if row.empty:
@@ -100,6 +80,7 @@ def fetch_staff(staff_id):
     dept = row.iloc[0]["Department"]
 
     perms = load_permissions()
+
     count = len(perms[perms["StaffID"] == staff_id])
 
     return name, dept, str(count)
@@ -107,70 +88,58 @@ def fetch_staff(staff_id):
 
 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 not validate_date(date):
+        return "❌ Date format: DD/MM/YYYY"
 
-    if f.strip() == "" or t.strip() == "":
-        return "❌ Time Required"
+    if not validate_time(f) or not validate_time(t):
+        return "❌ Time format: HH:MM"
 
     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"
 
-
     perms = load_permissions()
 
 
-    # Same day check (max 2)
+    # ---------- RULE 1: SAME DAY ---------- #
+
     same_day = perms[
         (perms["StaffID"] == staff_id) &
         (perms["Date"] == date)
     ]
 
     if len(same_day) == 1:
-        return "⚠️ Second permission today → Apply Leave"
+        return "⚠️ Apply Leave (Second permission same day)"
 
     if len(same_day) >= 2:
         return "❌ Daily limit reached"
 
 
-    # Monthly limit (2 only)
-    m = date.split("/")[1]
-    y = date.split("/")[2]
+    # ---------- RULE 2: MONTHLY LIMIT ---------- #
+
+    month = date.split("/")[1]
+    year = date.split("/")[2]
 
     same_month = perms[
         (perms["StaffID"] == staff_id) &
-        (perms["Date"].str.contains(f"/{m}/{y}", na=False))
+        (perms["Date"].str.contains(f"/{month}/{year}", na=False))
     ]
 
     if len(same_month) >= 2:
-        return "❌ Monthly limit reached (2 only)"
+        return "❌ Monthly limit (Only 2 allowed)"
 
 
-    # Save
+    # ---------- SAVE ---------- #
+
     name = row.iloc[0]["Name"]
     dept = row.iloc[0]["Department"]
 
@@ -187,7 +156,11 @@ def submit_permission(staff_id, date, f, t, reason):
         "SubmittedOn": now
     }
 
-    perms = pd.concat([perms, pd.DataFrame([new_row])], ignore_index=True)
+    perms = pd.concat(
+        [perms, pd.DataFrame([new_row])],
+        ignore_index=True
+    )
+
     perms.to_excel(PERMISSION_FILE, index=False)
 
     return "✅ Permission Submitted Successfully"
@@ -196,6 +169,7 @@ def submit_permission(staff_id, date, f, t, reason):
 def delete_permission(staff_id, date):
 
     perms = load_permissions()
+
     before = len(perms)
 
     perms = perms[
@@ -208,7 +182,7 @@ def delete_permission(staff_id, date):
     if len(perms) == before:
         return "❌ No Record Found"
     else:
-        return "✅ Deleted"
+        return "✅ Deleted Successfully"
 
 
 def clear_form():
@@ -217,10 +191,10 @@ def clear_form():
 
 # ---------------- ADMIN ---------------- #
 
-def admin_login(user, pwd):
+def admin_login(u, p):
 
-    if user == ADMIN_USER and pwd == ADMIN_PASS:
-        return gr.update(visible=True), "✅ Login Successful"
+    if u == ADMIN_USER and p == ADMIN_PASS:
+        return gr.update(visible=True), "✅ Login Success"
 
     return gr.update(visible=False), "❌ Invalid Login"
 
@@ -233,54 +207,72 @@ def download_excel():
     return PERMISSION_FILE
 
 
-def monthly_report(month, year):
+def monthly_report(m, y):
 
     perms = load_permissions()
 
     return perms[
-        perms["Date"].str.contains(f"/{month}/{year}", na=False)
+        perms["Date"].str.contains(f"/{m}/{y}", na=False)
     ]
 
 
-def dept_report(dept):
+def dept_report(d):
 
     perms = load_permissions()
 
-    return perms[perms["Department"] == dept]
+    return perms[perms["Department"] == d]
 
 
 def reset_data():
 
     df = pd.DataFrame(columns=[
-        "StaffID",
-        "Name",
-        "Department",
-        "Date",
-        "FromTime",
-        "ToTime",
-        "Reason",
-        "SubmittedOn"
+        "StaffID", "Name", "Department",
+        "Date", "FromTime", "ToTime",
+        "Reason", "SubmittedOn"
     ])
 
     df.to_excel(PERMISSION_FILE, index=False)
 
-    return "✅ All Data Cleared"
+    return "✅ All Data Reset"
 
 
 # ================= UI ================= #
 
 with gr.Blocks() as app:
 
+
+    # ---------- CSS ---------- #
+
     gr.HTML("""
-    <center>
-        <h2>SRC, SASTRA</h2>
-        <h3>Staff Permission Management System</h3>
-        <hr>
-    </center>
+    <style>
+    .center-text {
+        text-align: center;
+    }
+    </style>
     """)
 
 
-    # ---------------- STAFF TAB ---------------- #
+    # ---------- HEADER ---------- #
+
+    with gr.Column():
+
+        gr.Image(
+            LOGO_FILE,
+            show_label=False,
+            height=120,
+            container=True
+        )
+
+        gr.Markdown(
+            "## SRC, SASTRA\n"
+            "### Staff Permission Management System",
+            elem_classes=["center-text"]
+        )
+
+    gr.Markdown("---")
+
+
+    # ================= STAFF TAB ================= #
 
     with gr.Tab("Staff Panel"):
 
@@ -300,8 +292,8 @@ with gr.Blocks() as app:
             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)")
+                f = gr.Textbox(label="From Time (HH:MM)")
+                t = gr.Textbox(label="To Time (HH:MM)")
 
                 reason = gr.Textbox(label="Reason", lines=3)
 
@@ -313,7 +305,8 @@ with gr.Blocks() as app:
                 status = gr.Textbox(label="Status", interactive=False)
 
 
-    # ---------------- ADMIN TAB ---------------- #
+
+    # ================= ADMIN TAB ================= #
 
     with gr.Tab("Admin Panel"):
 
@@ -326,13 +319,14 @@ with gr.Blocks() as app:
 
         admin_box = gr.Column(visible=False)
 
+
         with admin_box:
 
             gr.Markdown("### 📊 All Records")
 
             table = gr.Dataframe(interactive=False)
 
-            refresh = gr.Button("Refresh")
+            refresh = gr.Button("Refresh Table")
 
 
             gr.Markdown("### 📅 Monthly Report")
@@ -361,14 +355,15 @@ with gr.Blocks() as app:
             file = gr.File()
 
 
-            gr.Markdown("### ⚠️ Reset")
+            gr.Markdown("### ⚠️ Reset System")
 
             reset = gr.Button("Reset All", variant="stop")
 
             reset_msg = gr.Textbox(interactive=False)
 
 
-    # ---------------- EVENTS ---------------- #
+
+    # ================= EVENTS ================= #
 
     fetch.click(fetch_staff, staff_id, [name, dept, count])
 
@@ -383,7 +378,11 @@ with gr.Blocks() as app:
     clear.click(clear_form, outputs=[date, f, t, reason, status])
 
 
-    login.click(admin_login, [admin_user, admin_pwd], [admin_box, msg])
+    login.click(
+        admin_login,
+        [admin_user, admin_pwd],
+        [admin_box, msg]
+    )
 
     refresh.click(get_all_data, outputs=table)
 
@@ -396,4 +395,9 @@ with gr.Blocks() as app:
     reset.click(reset_data, outputs=reset_msg)
 
 
-app.launch()
+
+# ================= LAUNCH ================= #
+
+if __name__ == "__main__":
+
+    app.launch()