Update app.py

app.py

@@ -1,242 +1,55 @@
-""" Iqbal Hall Hostel Allotment (Manual Selection) - Gradio App
-
-Save this file as app.py and deploy in a Gradio Space on Hugging Face. Dependencies (requirements.txt): gradio>=4.29.0 pillow pandas
-
-Features added/changed from auto-only version:
-
-Manual selection of Floor, Room and Bed (with availability validation)
-
-Option to Auto-Assign if the selected room/bed is full or user prefers automatic allocation
-
-Occupancy refresh, search, and card generation as before
-
-All data stored in data/ folder inside the Space (students.csv, photos/, cards/)
-
-
-Usage:
-
-Create a Gradio Space (Gradio runtime), upload this file, add requirements.txt and README
-
-The UI allows an admin or user to manually choose floor, room and bed. The app will verify availability. """
-
-
-import gradio as gr import os import pandas as pd from PIL import Image, ImageDraw from datetime import datetime import uuid import io
-
-=== Configuration ===
-
-HOSTEL_NAME = "Iqbal Hall" UNIVERSITY_NAME = "NFC Institute of Engineering & Technology (NFCIET) Multan" TOTAL_ROOMS = 160 FLOORS = 4 CAPACITY_PER_ROOM = 3 BEDS = ["A", "B", "C"]  # 3-person capacity per room
-
-DATA_DIR = "data" PHOTOS_DIR = os.path.join(DATA_DIR, "photos") CARDS_DIR = os.path.join(DATA_DIR, "cards") CSV_PATH = os.path.join(DATA_DIR, "students.csv")
-
-os.makedirs(PHOTOS_DIR, exist_ok=True) os.makedirs(CARDS_DIR, exist_ok=True)
-
-Initialize CSV if not exists
-
-if not os.path.exists(CSV_PATH): df = pd.DataFrame(columns=[ "id","timestamp","name","father_name","nic_bform","district","province","date_of_birth", "floor","room","bed","photo_path","card_path" ]) df.to_csv(CSV_PATH, index=False)
-
-Helper functions
-
-def rooms_per_floor(): return TOTAL_ROOMS // FLOORS
-
-def floor_room_range(floor): """Return list of room numbers for a floor. Floor is 1..FLOORS We use numbering: 101..140, 201..240, etc. """ rpf = rooms_per_floor() start = floor * 100 + 1 return [start + i for i in range(rpf)]
-
-def all_rooms_list(): rooms = [] for fl in range(1, FLOORS+1): rooms.extend(floor_room_range(fl)) return rooms
-
-def read_db(): try: df = pd.read_csv(CSV_PATH) except Exception: df = pd.DataFrame(columns=[ "id","timestamp","name","father_name","nic_bform","district","province","date_of_birth", "floor","room","bed","photo_path","card_path" ]) return df
-
-def current_allocations(): df = read_db() occ = {} for _, row in df.iterrows(): r = str(row["room"]) b = str(row["bed"]) if r not in occ: occ[r] = set() occ[r].add(b) return occ
-
-def available_beds_for_room(room_number): occ = current_allocations() used = occ.get(str(room_number), set()) return [b for b in BEDS if b not in used]
-
-def is_bed_available(room, bed): return bed in available_beds_for_room(room)
-
-def next_available_room_bed(): occ = current_allocations() for r in all_rooms_list(): r_str = str(r) used = occ.get(r_str, set()) for b in BEDS: if b not in used: # derive floor from room number floor = r // 100 return floor, r, b return None, None, None
-
-def draw_hostel_card(record, student_img): W, H = 1000, 600 card = Image.new("RGB", (W, H), (245, 248, 250)) draw = ImageDraw.Draw(card)
-
-header_h = 90
-draw.rectangle([(0,0), (W, header_h)], fill=(30, 64, 175))
-header_text = f"{HOSTEL_NAME} – {UNIVERSITY_NAME}"
-draw.text((24, 20), header_text, fill=(255,255,255))
-
-photo_box = (40, header_h + 30, 40+220, header_h + 30 + 260)
-if student_img.mode != "RGB":
-    student_img = student_img.convert("RGB")
-img = student_img.copy()
-img.thumbnail((photo_box[2]-photo_box[0], photo_box[3]-photo_box[1]))
-px = photo_box[0] + ((photo_box[2]-photo_box[0]) - img.size[0])//2
-py = photo_box[1] + ((photo_box[3]-photo_box[1]) - img.size[1])//2
-draw.rectangle(photo_box, outline=(100,100,100), width=2)
-card.paste(img, (px, py))
-
-info_x = 300
-info_y = header_h + 30
-line_gap = 36
-
-def t(label, value, y):
-    draw.text((info_x, y), f"{label}: {value}", fill=(20,20,20))
-
-t("Name", record["name"], info_y); info_y += line_gap
-t("Father's Name", record["father_name"], info_y); info_y += line_gap
-t("NIC/B-Form", record["nic_bform"], info_y); info_y += line_gap
-t("District", record["district"], info_y); info_y += line_gap
-t("Province", record["province"], info_y); info_y += line_gap
-t("Date of Birth", record["date_of_birth"], info_y); info_y += line_gap
-t("Floor", record.get("floor",""), info_y); info_y += line_gap
-t("Room", record.get("room",""), info_y); info_y += line_gap
-t("Bed", record.get("bed",""), info_y); info_y += line_gap
-t("Allotment ID", record["id"], info_y); info_y += line_gap
-
-footer_text = f"Issue Date: {datetime.now().strftime('%Y-%m-%d')}    Valid for current academic year"
-draw.text((24, H-40), footer_text, fill=(60,60,60))
-
-buf = io.BytesIO()
-card.save(buf, format="PNG")
-buf.seek(0)
-return card, buf
-
-def validate_inputs(name, father_name, nic_bform, district, province, dob, photo): missing = [] if not name: missing.append("Name") if not father_name: missing.append("Father's Name") if not nic_bform: missing.append("NIC/B-Form") if not district: missing.append("District") if not province: missing.append("Province") if not dob: missing.append("Date of Birth") if photo is None: missing.append("Photo") if missing: return False, f"Please provide: {', '.join(missing)}" return True, ""
-
-def save_record(record): try: df = pd.read_csv(CSV_PATH) if os.path.exists(CSV_PATH) else pd.DataFrame() except Exception: df = pd.DataFrame() df = pd.concat([df, pd.DataFrame([record])], ignore_index=True) df.to_csv(CSV_PATH, index=False)
-
---- Gradio callbacks ---
-
-Populate floors for dropdown
-
-FLOOR_OPTIONS = [str(i) for i in range(1, FLOORS+1)] PROVINCES = ["Punjab","Sindh","Khyber Pakhtunkhwa","Balochistan","Gilgit-Baltistan","Azad Kashmir","Islamabad Capital Territory"]
-
-def rooms_for_floor_str(floor_str): if not floor_str: return [] fl = int(floor_str) return [str(r) for r in floor_room_range(fl)]
-
-def beds_for_room_str(room_str): if not room_str: return [] room = int(room_str) return available_beds_for_room(room)
-
-def manual_submit(name, father_name, nic_bform, district, province, dob, photo, floor_sel, room_sel, bed_sel, auto_assign_flag): # Validate fields ok, msg = validate_inputs(name, father_name, nic_bform, district, province, dob, photo) if not ok: return gr.update(value=None), gr.update(visible=False), gr.update(visible=False), gr.update(value=msg)
-
-# If user elected auto_assign_flag (checkbox True), perform automatic assignment
-if auto_assign_flag:
-    floor, room, bed = next_available_room_bed()
-    if room is None:
-        return gr.update(value=None), gr.update(visible=False), gr.update(visible=False), gr.update(value="Hostel is fully occupied. No rooms available.")
-else:
-    # Use manual selection - ensure provided
-    if not floor_sel or not room_sel or not bed_sel:
-        return gr.update(value=None), gr.update(visible=False), gr.update(visible=False), gr.update(value="Please select Floor, Room and Bed, or choose Auto-Assign.")
-    room = int(room_sel)
-    bed = bed_sel
-    floor = int(floor_sel)
-    # Check bed availability
-    if not is_bed_available(room, bed):
-        return gr.update(value=None), gr.update(visible=False), gr.update(visible=False), gr.update(value=f"Selected Bed {bed} in Room {room} is already occupied. Choose another.")
-
-# Create unique ID
-sid = str(uuid.uuid4())[:8].upper()
-
-# Save photo
-photo_path = os.path.join(PHOTOS_DIR, f"{sid}.png")
-if isinstance(photo, str) and os.path.exists(photo):
-    im = Image.open(photo)
-else:
-    im = photo
-if im.mode != "RGB":
-    im = im.convert("RGB")
-im.save(photo_path)
-
-record = {
-    "id": sid,
-    "timestamp": datetime.now().isoformat(timespec="seconds"),
-    "name": name.strip(),
-    "father_name": father_name.strip(),
-    "nic_bform": nic_bform.strip(),
-    "district": district.strip(),
-    "province": province.strip(),
-    "date_of_birth": dob.strip(),
-    "floor": floor,
-    "room": room,
-    "bed": bed,
-    "photo_path": photo_path,
-    "card_path": ""
-}
-
-# Generate card
-card_img, buf = draw_hostel_card(record, im)
-card_path = os.path.join(CARDS_DIR, f"{sid}_card.png")
-card_img.save(card_path)
-record["card_path"] = card_path
-
-# Save record
-save_record(record)
-
-success_msg = f"Allotment successful!\nAssigned Floor: {floor} Room: {room}  Bed: {bed}\nAllotment ID: {sid}"
-return card_path, gr.update(visible=True), gr.update(visible=True), gr.update(value=success_msg)
-
-def load_occupancy(): df = read_db() total_beds = TOTAL_ROOMS * CAPACITY_PER_ROOM used_beds = len(df) free_beds = total_beds - used_beds status = f"Total Capacity: {total_beds} | Allocated: {used_beds} | Available: {free_beds}" # convert some columns to string for display safety if not df.empty: df_disp = df.astype(str) else: df_disp = df return df_disp, status
-
-def search_by_nic(nic): if not nic: return "Enter NIC/B-Form to search.", None, None df = read_db() if df.empty: return "No records yet.", None, None res = df[df["nic_bform"].astype(str).str.strip().str.lower() == nic.strip().lower()] if res.empty: return "No record found.", None, None row = res.iloc[0].to_dict() card_img = row.get("card_path") if isinstance(row.get("card_path"), str) and os.path.exists(row.get("card_path")) else None text = f"Found: {row['name']} | Floor {row.get('floor','')} Room {row['room']} Bed {row['bed']} | ID: {row['id']}" return text, card_img, row.get("card_path")
-
-Build Gradio UI
-
-with gr.Blocks(title=f"{HOSTEL_NAME} Allotment – {UNIVERSITY_NAME}") as demo: gr.Markdown(f"## {HOSTEL_NAME} – {UNIVERSITY_NAME}  \nManual Room Selection + Auto-Assign fallback")
-
-with gr.Tabs():
-    with gr.Tab("New Allotment"):
-        with gr.Row():
-            with gr.Column(scale=1):
-                name = gr.Textbox(label="Student Name", placeholder="Enter full name")
-                father_name = gr.Textbox(label="Father's Name", placeholder="Enter father's name")
-                nic_bform = gr.Textbox(label="NIC / B-Form", placeholder="e.g., 35202-1234567-1")
-                district = gr.Textbox(label="District")
-                province = gr.Dropdown(choices=PROVINCES, label="Province")
-                dob = gr.Textbox(label="Date of Birth", placeholder="YYYY-MM-DD")
-            with gr.Column(scale=1):
-                photo = gr.Image(type="pil", label="Student Photo (Passport size preferred)")
-                gr.Markdown("**Manual Room Selection (leave blank for Auto-Assign)**")
-                floor_dd = gr.Dropdown(choices=FLOOR_OPTIONS, label="Floor (1-4)")
-                room_dd = gr.Dropdown(choices=[], label="Room Number")
-                bed_dd = gr.Dropdown(choices=[], label="Bed (A/B/C)")
-                auto_assign_checkbox = gr.Checkbox(label="Auto-Assign instead of manual selection", value=False)
-                submit_btn = gr.Button("Allocate Room & Generate Card", variant="primary")
-                status_box = gr.Textbox(label="Status", interactive=False)
-
-        with gr.Row():
-            card_out = gr.Image(label="Generated Hostel Card (PNG)", visible=False)
-        with gr.Row():
-            download_card = gr.File(label="Download Hostel Card", visible=False)
-
-        # Populate rooms when floor changes
-        floor_dd.change(lambda f: rooms_for_floor_str(f), inputs=[floor_dd], outputs=[room_dd])
-        # Populate available beds when room changes
-        room_dd.change(lambda r: beds_for_room_str(r), inputs=[room_dd], outputs=[bed_dd])
-
-        submit_btn.click(
-            manual_submit,
-            inputs=[name, father_name, nic_bform, district, province, dob, photo, floor_dd, room_dd, bed_dd, auto_assign_checkbox],
-            outputs=[card_out, download_card, card_out, status_box]
-        )
-
-    with gr.Tab("Occupancy"):
-        refresh_btn = gr.Button("Refresh Occupancy")
-        occupancy_status = gr.Markdown("")
-        table = gr.Dataframe(wrap=True, interactive=False)
-
-        def _refresh():
-            df, status = load_occupancy()
-            return status, df
-
-        refresh_btn.click(_refresh, inputs=None, outputs=[occupancy_status, table])
-        occupancy_status.value, table.value = load_occupancy()
-
-    with gr.Tab("Find Student"):
-        query_nic = gr.Textbox(label="NIC / B-Form", placeholder="Enter exact NIC/B-Form to search")
-        search_btn = gr.Button("Search")
-        result_text = gr.Markdown("")
-        found_card = gr.Image(label="Hostel Card", visible=False)
-        found_file = gr.File(label="Download Card", visible=False)
-
-        def _search(nic):
-            text, img_path, file_path = search_by_nic(nic)
-            return text, gr.update(visible=bool(img_path), value=img_path), gr.update(visible=bool(file_path), value=file_path)
-
-        search_btn.click(_search, inputs=[query_nic], outputs=[result_text, found_card, found_file])
-
-gr.Markdown("— Built with ❤️ using Gradio. Data is stored locally in the `data/` folder inside this Space.")
-
-if name == "main": demo.launch()
+import gradio as gr
+import os
+import pandas as pd
+from datetime import datetime
+import uuid
+
+# File to store allotment data
+CSV_FILE = "hostel_allotments.csv"
+
+# Create CSV if not exists
+if not os.path.exists(CSV_FILE):
+    df = pd.DataFrame(columns=["Allotment_ID", "Name", "Roll_No", "Department", "Hostel", "Room_Type", "Bed_No", "Date_Time"])
+    df.to_csv(CSV_FILE, index=False)
+
+# Function for hostel allotment
+def allot_hostel(name, roll_no, department, hostel, room_type, bed_no):
+    allotment_id = str(uuid.uuid4())[:8]  # unique short ID
+    date_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+    # Save to CSV
+    df = pd.read_csv(CSV_FILE)
+    new_entry = {
+        "Allotment_ID": allotment_id,
+        "Name": name,
+        "Roll_No": roll_no,
+        "Department": department,
+        "Hostel": hostel,
+        "Room_Type": room_type,
+        "Bed_No": bed_no,
+        "Date_Time": date_time
+    }
+    df = pd.concat([df, pd.DataFrame([new_entry])], ignore_index=True)
+    df.to_csv(CSV_FILE, index=False)
+
+    return f"✅ Hostel Allotted Successfully!\n\nAllotment ID: {allotment_id}\nName: {name}\nRoll No: {roll_no}\nDepartment: {department}\nHostel: {hostel}\nRoom Type: {room_type}\nBed No: {bed_no}\nDate & Time: {date_time}"
+
+# UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🏨 Hostel Allotment System")
+
+    with gr.Row():
+        name = gr.Textbox(label="Student Name")
+        roll_no = gr.Textbox(label="Roll Number")
+        department = gr.Textbox(label="Department")
+
+    hostel = gr.Dropdown(["Hostel A", "Hostel B", "Hostel C"], label="Select Hostel")
+    room_type = gr.Dropdown(["Single", "Double", "Triple"], label="Select Room Type")
+    bed_no = gr.Dropdown([f"Bed {i}" for i in range(1, 11)], label="Select Bed No")
+
+    submit_btn = gr.Button("Allot Hostel")
+    output = gr.Textbox(label="Allotment Details", lines=10)
+
+    submit_btn.click(fn=allot_hostel, inputs=[name, roll_no, department, hostel, room_type, bed_no], outputs=output)
+
+demo.launch()