Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,106 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+from transformers import pipeline
+from backend import db_connection # Import db_connection
 
-"""
-# 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])),
-    ))
+# -----------------------------
+# Summarizer model
+# -----------------------------
+@st.cache_resource
+def load_summarizer():
+    return pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
+
+summarizer = load_summarizer()
+
+# -----------------------------
+# Functions
+# -----------------------------
+def signup():
+    st.title("📝 Signup Page")
+    # name = st.text_input("Enter your Name") # Removed name input
+    email = st.text_input("Enter your Email")
+    username = st.text_input("Choose a Username")
+    password = st.text_input("Choose a Password", type="password")
+
+    if st.button("Signup"):
+        if email and username and password: # Removed 'name'
+            # Use db_connection to create user
+            if db_connection.create_user(email, username, password): # Removed 'name' argument
+                st.success("✅ Account created! Please log in now.")
+                st.session_state.page = "login"
+                st.rerun()
+            else:
+                st.error("⚠️ An account with this username or email already exists. Please choose another.")
+        else:
+            st.error("⚠️ Please fill in all fields.")
+
+def login():
+    st.title("🔒 Login Page")
+    username = st.text_input("Username")
+    password = st.text_input("Password", type="password")
+
+    if st.button("Login"):
+        # Use db_connection to validate user
+        user = db_connection.validate_user(username, password)
+        if user:
+            st.session_state.logged_in = True
+            st.session_state.username = user['username'] # Store username in session
+            st.session_state.user_role = user['role'] # Store user role in session
+            st.success("✅ Login successful! Redirecting...")
+            st.session_state.page = "dashboard"
+            st.rerun()
+        else:
+            st.error("❌ Invalid username or password")
+
+    if st.button("Go to Signup"):
+        st.session_state.page = "signup"
+        st.rerun()
+
+def summarizer_page():
+    st.title("📄 Text Summarizer")
+
+    text = st.text_area("Enter the text you want to summarize:", height=200)
+
+    if st.button("Summarize"):
+        if text.strip():
+            summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
+            st.subheader("✨ Summary:")
+            st.write(summary[0]['summary_text'])
+        else:
+            st.error("⚠️ Please enter some text to summarize.")
+
+    if st.button("Logout"):
+        st.session_state.logged_in = False
+        st.session_state.page = "login" # Ensure it redirects to login after logout
+        st.rerun()
+
+# -----------------------------
+# Main App
+# -----------------------------
+def main():
+    if "logged_in" not in st.session_state:
+        st.session_state.logged_in = False
+    if "page" not in st.session_state:
+        st.session_state.page = "login"
+
+    if st.session_state.page == "signup":
+        signup()
+    elif st.session_state.page == "login":
+        login()
+    elif st.session_state.page == "dashboard" and st.session_state.logged_in: # New dashboard page
+        st.title("Welcome to Dashboard!") # Placeholder for dashboard content
+        # In a real app, you'd import and call a dashboard function here
+        if st.button("Go to Summarizer"): # Example navigation to summarizer
+            st.session_state.page = "summarizer"
+            st.rerun()
+        if st.button("Logout"): # Logout from dashboard
+            st.session_state.logged_in = False
+            st.session_state.page = "login"
+            st.rerun()
+    elif st.session_state.page == "summarizer" and st.session_state.logged_in:
+        summarizer_page()
+    else:
+        st.session_state.page = "login"
+        login()
+
+if __name__ == "__main__":
+    main()