Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,320 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import requests
+import json
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# Backend URL
+BASE_URL = "https://testys-clearance-sys.hf.space"
+
+# Session state for auth
+if 'token' not in st.session_state:
+    st.session_state.token = None
+if 'user' not in st.session_state:
+    st.session_state.user = None
+
+def login(username, password):
+    response = requests.post(f"{BASE_URL}/token", data={"username": username, "password": password})
+    if response.status_code == 200:
+        data = response.json()
+        st.session_state.token = data['access_token']
+        st.session_state.user = username
+        st.success("Logged in successfully!")
+    else:
+        st.error("Login failed!")
+
+def logout():
+    st.session_state.token = None
+    st.session_state.user = None
+    st.success("Logged out!")
+
+def api_call(method, endpoint, data=None, headers=None):
+    url = f"{BASE_URL}{endpoint}"
+    if headers is None:
+        headers = {}
+    if st.session_state.token:
+        headers['Authorization'] = f"Bearer {st.session_state.token}"
+    if method == 'GET':
+        response = requests.get(url, headers=headers)
+    elif method == 'POST':
+        response = requests.post(url, json=data, headers=headers)
+    elif method == 'PUT':
+        response = requests.put(url, json=data, headers=headers)
+    elif method == 'DELETE':
+        response = requests.delete(url, headers=headers)
+    return response
+
+st.title("Clearance System Frontend")
+
+if st.session_state.token is None:
+    st.header("Login")
+    username = st.text_input("Username")
+    password = st.text_input("Password", type="password")
+    if st.button("Login"):
+        login(username, password)
+else:
+    st.write(f"Logged in as: {st.session_state.user}")
+    if st.button("Logout"):
+        logout()
+
+    # Navigation
+    page = st.sidebar.selectbox("Select Page", [
+        "Students", "Users", "Tags", "Devices", "Clearance", "RFID", "Scanners", "Profile"
+    ])
+
+    if page == "Students":
+        st.header("Student Management")
+        action = st.selectbox("Action", ["List", "Create", "Lookup", "Read Single", "Update", "Delete"])
+        if action == "List":
+            skip = st.number_input("Skip", min_value=0, value=0)
+            limit = st.number_input("Limit", min_value=1, value=100)
+            if st.button("Get Students"):
+                response = api_call('GET', f'/admin/students/?skip={skip}&limit={limit}')
+                if response.status_code == 200:
+                    students = response.json()
+                    st.json(students)
+                else:
+                    st.error("Failed to fetch students")
+        elif action == "Create":
+            full_name = st.text_input("Full Name")
+            matric_no = st.text_input("Matric No")
+            email = st.text_input("Email")
+            department = st.selectbox("Department", ["Computer Science", "Engineering", "Business Administration", "Law", "Medicine"])
+            password = st.text_input("Password", type="password")
+            if st.button("Create Student"):
+                data = {"full_name": full_name, "matric_no": matric_no, "email": email, "department": department, "password": password}
+                response = api_call('POST', '/admin/students/', data)
+                if response.status_code == 201:
+                    st.success("Student created!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to create student")
+        elif action == "Lookup":
+            lookup_type = st.selectbox("Lookup by", ["Matric No", "Tag ID"])
+            if lookup_type == "Matric No":
+                matric_no = st.text_input("Matric No")
+                if st.button("Lookup"):
+                    response = api_call('GET', f'/admin/students/lookup?matric_no={matric_no}')
+                    if response.status_code == 200:
+                        st.json(response.json())
+                    else:
+                        st.error("Student not found")
+            else:
+                tag_id = st.text_input("Tag ID")
+                if st.button("Lookup"):
+                    response = api_call('GET', f'/admin/students/lookup?tag_id={tag_id}')
+                    if response.status_code == 200:
+                        st.json(response.json())
+                    else:
+                        st.error("Student not found")
+        elif action == "Read Single":
+            student_id = st.number_input("Student ID", min_value=1)
+            if st.button("Read"):
+                response = api_call('GET', f'/admin/students/{student_id}')
+                if response.status_code == 200:
+                    st.json(response.json())
+                else:
+                    st.error("Student not found")
+        elif action == "Update":
+            student_id = st.number_input("Student ID", min_value=1)
+            full_name = st.text_input("Full Name")
+            department = st.selectbox("Department", ["Computer Science", "Engineering", "Business Administration", "Law", "Medicine"])
+            email = st.text_input("Email")
+            if st.button("Update"):
+                data = {"full_name": full_name, "department": department, "email": email}
+                response = api_call('PUT', f'/admin/students/{student_id}', data)
+                if response.status_code == 200:
+                    st.success("Student updated!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to update student")
+        elif action == "Delete":
+            student_id = st.number_input("Student ID", min_value=1)
+            if st.button("Delete"):
+                response = api_call('DELETE', f'/admin/students/{student_id}')
+                if response.status_code == 200:
+                    st.success("Student deleted!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to delete student")
+
+    elif page == "Users":
+        st.header("User Management")
+        action = st.selectbox("Action", ["Create", "List", "Lookup", "Update", "Delete"])
+        if action == "Create":
+            username = st.text_input("Username")
+            password = st.text_input("Password", type="password")
+            email = st.text_input("Email")
+            full_name = st.text_input("Full Name")
+            role = st.selectbox("Role", ["admin", "staff"])
+            department = st.selectbox("Department", ["Computer Science", "Engineering", "Business Administration", "Law", "Medicine"])
+            if st.button("Create User"):
+                data = {"username": username, "password": password, "email": email, "full_name": full_name, "role": role, "department": department}
+                response = api_call('POST', '/admin/users/', data)
+                if response.status_code == 201:
+                    st.success("User created!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to create user")
+        elif action == "List":
+            if st.button("Get Users"):
+                response = api_call('GET', '/admin/users/')
+                if response.status_code == 200:
+                    users = response.json()
+                    st.json(users)
+                else:
+                    st.error("Failed to fetch users")
+        elif action == "Lookup":
+            lookup_type = st.selectbox("Lookup by", ["Username", "Tag ID"])
+            if lookup_type == "Username":
+                username = st.text_input("Username")
+                if st.button("Lookup"):
+                    response = api_call('GET', f'/admin/users/lookup?username={username}')
+                    if response.status_code == 200:
+                        st.json(response.json())
+                    else:
+                        st.error("User not found")
+            else:
+                tag_id = st.text_input("Tag ID")
+                if st.button("Lookup"):
+                    response = api_call('GET', f'/admin/users/lookup?tag_id={tag_id}')
+                    if response.status_code == 200:
+                        st.json(response.json())
+                    else:
+                        st.error("User not found")
+        elif action == "Update":
+            user_id = st.number_input("User ID", min_value=1)
+            username = st.text_input("Username")
+            email = st.text_input("Email")
+            full_name = st.text_input("Full Name")
+            role = st.selectbox("Role", ["admin", "staff"])
+            department = st.selectbox("Department", ["Computer Science", "Engineering", "Business Administration", "Law", "Medicine"])
+            password = st.text_input("Password", type="password")
+            if st.button("Update"):
+                data = {"username": username, "email": email, "full_name": full_name, "role": role, "department": department, "password": password}
+                response = api_call('PUT', f'/admin/users/{user_id}', data)
+                if response.status_code == 200:
+                    st.success("User updated!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to update user")
+        elif action == "Delete":
+            user_id = st.number_input("User ID", min_value=1)
+            if st.button("Delete"):
+                response = api_call('DELETE', f'/admin/users/{user_id}')
+                if response.status_code == 200:
+                    st.success("User deleted!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to delete user")
+
+    elif page == "Tags":
+        st.header("Tag Management")
+        action = st.selectbox("Action", ["Link", "Unlink"])
+        if action == "Link":
+            tag_id = st.text_input("Tag ID")
+            matric_no = st.text_input("Matric No")
+            username = st.text_input("Username")
+            if st.button("Link Tag"):
+                data = {"tag_id": tag_id, "matric_no": matric_no, "username": username}
+                response = api_call('POST', '/admin/tags/link', data)
+                if response.status_code == 200:
+                    st.success("Tag linked!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to link tag")
+        elif action == "Unlink":
+            tag_id = st.text_input("Tag ID")
+            if st.button("Unlink Tag"):
+                response = api_call('DELETE', f'/admin/tags/{tag_id}/unlink')
+                if response.status_code == 200:
+                    st.success("Tag unlinked!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to unlink tag")
+
+    elif page == "Devices":
+        st.header("Device Management")
+        action = st.selectbox("Action", ["Create", "List", "Delete"])
+        if action == "Create":
+            device_name = st.text_input("Device Name")
+            location = st.text_input("Location")
+            department = st.selectbox("Department", ["Computer Science", "Engineering", "Business Administration", "Law", "Medicine"])
+            if st.button("Create Device"):
+                data = {"device_name": device_name, "location": location, "department": department}
+                response = api_call('POST', '/admin/devices/', data)
+                if response.status_code == 201:
+                    st.success("Device created!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to create device")
+        elif action == "List":
+            if st.button("Get Devices"):
+                response = api_call('GET', '/admin/devices/')
+                if response.status_code == 200:
+                    devices = response.json()
+                    st.json(devices)
+                else:
+                    st.error("Failed to fetch devices")
+        elif action == "Delete":
+            device_id = st.number_input("Device ID", min_value=1)
+            if st.button("Delete"):
+                response = api_call('DELETE', f'/admin/devices/{device_id}')
+                if response.status_code == 200:
+                    st.success("Device deleted!")
+                    st.json(response.json())
+                else:
+                    st.error("Failed to delete device")
+
+    elif page == "Clearance":
+        st.header("Clearance Management")
+        matric_no = st.text_input("Matric No")
+        department = st.selectbox("Department", ["Library", "Student Affairs", "Bursary", "Academic Affairs", "Health Center"])
+        status = st.selectbox("Status", ["pending", "approved", "rejected"])
+        remarks = st.text_area("Remarks")
+        if st.button("Update Clearance"):
+            data = {"matric_no": matric_no, "department": department, "status": status, "remarks": remarks}
+            response = api_call('PUT', '/clearance/update', data)
+            if response.status_code == 200:
+                st.success("Clearance updated!")
+                st.json(response.json())
+            else:
+                st.error("Failed to update clearance")
+
+    elif page == "RFID":
+        st.header("RFID Check Status")
+        tag_id = st.text_input("Tag ID")
+        if st.button("Check Status"):
+            data = {"tag_id": tag_id}
+            response = api_call('POST', '/rfid/check-status', data)
+            if response.status_code == 200:
+                st.json(response.json())
+            else:
+                st.error("Failed to check status")
+
+    elif page == "Scanners":
+        st.header("Scanner Management")
+        action = st.selectbox("Action", ["Activate", "Retrieve"])
+        if action == "Activate":
+            api_key = st.text_input("Device API Key")
+            if st.button("Activate Scanner"):
+                data = {"api_key": api_key}
+                response = api_call('POST', '/rfid/scanners/activate', data)
+                if response.status_code == 204:
+                    st.success("Scanner activated!")
+                else:
+                    st.error("Failed to activate")
+        elif action == "Retrieve":
+            if st.button("Retrieve Tag"):
+                response = api_call('GET', '/rfid/scanners/retrieve')
+                if response.status_code == 200:
+                    st.json(response.json())
+                else:
+                    st.error("No tag scanned yet")
+
+    elif page == "Profile":
+        st.header("My Profile")
+        if st.button("Get My Profile"):
+            response = api_call('GET', '/users/me')
+            if response.status_code == 200:
+                st.json(response.json())
+            else:
+                st.error("Failed to fetch profile")