Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,78 +1,49 @@
 import streamlit as st
 import pandas as pd
 import sqlite3
+import os
 from database import initialize_database
 
-# Initialize DB on start
+# Initialize DB
 initialize_database()
 
-# Function to fetch filtered students
-def get_students(department, year):
-    conn = sqlite3.connect("college.db")
-    query = "SELECT * FROM student WHERE 1=1"
-    if department != "All":
-        query += f" AND major = '{department}'"
-    if year != "All":
-        query += f" AND year_of_enrollment = {year}"
-    df = pd.read_sql_query(query, conn)
-    conn.close()
-    return df
+# Path
+db_path = os.path.join(os.path.dirname(__file__), "../college.db")
 
-# Function to fetch filtered courses
-def get_courses(department):
-    conn = sqlite3.connect("college.db")
-    query = "SELECT * FROM course"
-    if department != "All":
-        query += f" WHERE department = '{department}'"
-    df = pd.read_sql_query(query, conn)
-    conn.close()
-    return df
+# Title
+st.title("📊 College Dashboard")
 
-# Prepare filter options
-departments = [
-    "All",
-    "Computer Science",
-    "Mechanical Engineering",
-    "ECE",
-    "Civil",
-    "IT",
-    "CSE",
-    "AI"
-]
+# Sidebar filter
+table = st.sidebar.selectbox("Select Table", ["student", "employee", "course"])
 
-def get_years():
-    conn = sqlite3.connect("college.db")
-    df = pd.read_sql_query("SELECT DISTINCT year_of_enrollment FROM student ORDER BY year_of_enrollment", conn)
+# Load data
+def load_data(table):
+    conn = sqlite3.connect(db_path)
+    df = pd.read_sql_query(f"SELECT * FROM {table}", conn)
     conn.close()
-    years = df["year_of_enrollment"].astype(str).tolist()
-    years.insert(0, "All")
-    return years
-
-# Streamlit app UI
-
-st.title("College Database Dashboard")
-
-tab1, tab2 = st.tabs(["Students", "Courses"])
-
-with tab1:
-    st.header("Student Records")
-    dept_filter = st.selectbox("Select Department", departments)
-    year_filter = st.selectbox("Select Year of Enrollment", get_years())
-    
-    if st.button("Show Students"):
-        students_df = get_students(dept_filter, year_filter)
-        if students_df.empty:
-            st.info("No students found with the selected filters.")
-        else:
-            st.dataframe(students_df)
+    return df
 
-with tab2:
-    st.header("Course Records")
-    course_dept_filter = st.selectbox("Select Department", departments, key="course_dept")
-    
-    if st.button("Show Courses"):
-        courses_df = get_courses(course_dept_filter)
-        if courses_df.empty:
-            st.info("No courses found with the selected department.")
-        else:
-            st.dataframe(courses_df)
+# Filter logic
+def filter_data(df):
+    col = st.sidebar.selectbox("Select column to filter", df.columns)
+    unique_values = df[col].unique().tolist()
+    selected = st.sidebar.multiselect(f"Filter {col}", unique_values, default=unique_values)
+    return df[df[col].isin(selected)]
+
+# Display
+df = load_data(table)
+st.subheader(f"{table.capitalize()} Table")
+filtered_df = filter_data(df)
+st.dataframe(filtered_df)
+
+# Show SQL Query Results
+st.markdown("### 🔍 Run Custom SQL Query")
+sql_query = st.text_area("Enter your SQL Query below:")
+if st.button("Execute Query"):
+    try:
+        conn = sqlite3.connect(db_path)
+        result_df = pd.read_sql_query(sql_query, conn)
+        st.dataframe(result_df)
+        conn.close()
+    except Exception as e:
+        st.error(f"Error: {e}")