Update app.py

app.py

@@ -1,60 +1,28 @@
 import sqlite3
 import pandas as pd
 import gradio as gr
+from database import initialize_database
 
-DB_PATH = "college.db"
+# Initialize the database at startup (creates tables if not exist)
+initialize_database()
 
-def query_students(department):
-    conn = sqlite3.connect(DB_PATH)
-    if department == "All":
+def query_students(dept):
+    conn = sqlite3.connect("college.db")
+    if dept == "All":
         df = pd.read_sql_query("SELECT * FROM student", conn)
     else:
-        df = pd.read_sql_query("SELECT * FROM student WHERE major = ?", conn, params=(department,))
+        df = pd.read_sql_query(f"SELECT * FROM student WHERE major = '{dept}'", conn)
     conn.close()
     return df
 
-def query_employees(department):
-    conn = sqlite3.connect(DB_PATH)
-    if department == "All":
-        df = pd.read_sql_query("SELECT * FROM employee", conn)
-    else:
-        df = pd.read_sql_query("SELECT * FROM employee WHERE department = ?", conn, params=(department,))
-    conn.close()
-    return df
-
-def query_courses(department):
-    conn = sqlite3.connect(DB_PATH)
-    if department == "All":
-        df = pd.read_sql_query("SELECT * FROM course", conn)
-    else:
-        df = pd.read_sql_query("SELECT * FROM course WHERE department = ?", conn, params=(department,))
-    conn.close()
-    return df
-
-departments_students = ["All", "Computer Science", "Mechanical Engineering", "ECE", "Civil", "IT", "CSE", "AI"]
-departments_employees = ["All", "CSE", "ECE", "IT", "Civil", "Mechanical", "AI"]
-departments_courses = ["All", "CSE", "ECE", "IT", "Civil", "Mechanical", "AI"]
-
-with gr.Blocks() as demo:
-    gr.Markdown("# 🎓 College Dashboard")
-
-    with gr.Tab("Students"):
-        dept_stu = gr.Dropdown(choices=departments_students, value="All", label="Filter by Major/Department")
-        stu_table = gr.Dataframe(headers=["student_id","first_name","last_name","date_of_birth","email","phone_number","major","year_of_enrollment"], interactive=False)
-        dept_stu.change(fn=query_students, inputs=dept_stu, outputs=stu_table)
-        # Show all by default
-        stu_table.value = query_students("All")
+# Gradio UI components (simplified example)
+dept_dropdown = gr.Dropdown(choices=["All", "CSE", "ECE", "IT", "AI"], label="Select Department")
+stu_table = gr.Dataframe()
 
-    with gr.Tab("Employees"):
-        dept_emp = gr.Dropdown(choices=departments_employees, value="All", label="Filter by Department")
-        emp_table = gr.Dataframe(headers=["employee_id","first_name","last_name","email","department","position","salary","date_of_joining"], interactive=False)
-        dept_emp.change(fn=query_employees, inputs=dept_emp, outputs=emp_table)
-        emp_table.value = query_employees("All")
+def update_table(dept):
+    return query_students(dept)
 
-    with gr.Tab("Courses"):
-        dept_crs = gr.Dropdown(choices=departments_courses, value="All", label="Filter by Department")
-        crs_table = gr.Dataframe(headers=["course_id","course_name","course_code","instructor_id","department","credits","semester"], interactive=False)
-        dept_crs.change(fn=query_courses, inputs=dept_crs, outputs=crs_table)
-        crs_table.value = query_courses("All")
+demo = gr.Interface(fn=update_table, inputs=dept_dropdown, outputs=stu_table)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()