Update app.py

app.py

@@ -1,55 +1,56 @@
 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
+import random
+
+# Hostels and rooms
+hostels = ["Hostel A", "Hostel B", "Hostel C"]
+rooms = list(range(1, 21))  # 20 rooms per hostel
+
+# DataFrame to store allotments
+allotments = pd.DataFrame(columns=["Name", "Roll No", "Father's Name", "Hostel", "Room No"])
+
+# Function to allot hostel and room
+def allot_hostel(name, roll_no, father_name):
+    global allotments
+
+    # Check if already allotted
+    existing = allotments[allotments["Roll No"] == roll_no]
+    if not existing.empty:
+        return f"Already allotted: {existing.iloc[0]['Hostel']} - Room {existing.iloc[0]['Room No']}"
+
+    # Random hostel & room
+    hostel = random.choice(hostels)
+    room = random.choice(rooms)
+
+    # Save in DataFrame
+    new_row = pd.DataFrame([[name, roll_no, father_name, hostel, room]],
+                           columns=allotments.columns)
+    allotments = pd.concat([allotments, new_row], ignore_index=True)
+
+    return f"Allotted: {hostel}, Room {room}"
+
+# Function to show all allotments
+def show_allotments():
+    if allotments.empty:
+        return "No allotments yet."
+    return allotments
+
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🏨 Hostel Allotment System")
+    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")
+        father_name = gr.Textbox(label="Father's Name")
 
-    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")
+    allot_btn = gr.Button("Allot Hostel")
+    output = gr.Textbox(label="Allotment Result")
 
-    submit_btn = gr.Button("Allot Hostel")
-    output = gr.Textbox(label="Allotment Details", lines=10)
+    show_btn = gr.Button("Show Allotted Data")
+    table = gr.Dataframe(label="Allotment Records")
 
-    submit_btn.click(fn=allot_hostel, inputs=[name, roll_no, department, hostel, room_type, bed_no], outputs=output)
+    allot_btn.click(allot_hostel, inputs=[name, roll_no, father_name], outputs=output)
+    show_btn.click(show_allotments, outputs=table)
 
 demo.launch()