Upload 8 files

src/pages/auth.py

@@ -0,0 +1,122 @@
+import streamlit as st
+import requests
+from src.utils.history import retrieve_history  
+from src.schemas.schemas import UserModel
+
+# List of possible user roles for registration dropdown
+roles = ["engineering", "marketing", "finance", "hr", "general", "executives"] 
+
+
+# Login page UI and logic
+def login_page():
+    st.title("Login Page")
+
+    # User inputs for credentials
+    username = st.text_input("Username") 
+    password = st.text_input("Password", type="password")
+    api_key = st.text_input("Openai API Key", type="password") 
+
+    # Center the buttons using three columns
+    col1, _, _ = st.columns([2, 2, 1])
+
+    # Place both buttons in the left column (col1)
+    with col1:
+        login_clicked = st.button("Login")
+        register_clicked = st.button("Don't have an account? Register")
+
+    if login_clicked:
+        # Check if API key is provided
+        if not api_key.strip():
+            st.warning("⚠️ Please enter your OpenAI API Key before logging in.")
+        else:
+            try:
+                # Send POST request to FastAPI login endpoint with credentials
+                response = requests.post(
+                    "https://finsolve-chatbot-backend.onrender.com/api/v1/auth/login",
+                    json={"username": username, "password": password, "api_key": api_key}
+                )
+
+                if response.status_code == 200:
+                    user_resp = response.json()
+                    role = "employee" if user_resp['user']['role'] == "general" else user_resp['user']['role']
+                    user_resp['user']['role'] = role
+                    user_model = UserModel(**user_resp)
+
+                    st.success(f"✅ Login successful, Welcome, {user_model.user.username}, (Role: {user_model.user.role})")
+                    st.session_state.token = user_model.token
+                    st.session_state.user = user_model.user
+                    retrieve_history()
+                    st.rerun()
+                else:
+                    st.error(f"❌ Login failed: {response.json().get('msg', 'Invalid credentials')}")
+
+            except Exception as e:
+                st.error(f"⚠️ Error connecting to server: {e}")
+
+
+    if register_clicked:
+        # Switch UI to registration page and rerun app
+        st.session_state.is_login = False
+        st.rerun()
+
+
+# Registration page UI and logic
+def registration_page():
+    st.title("Registration Page")
+
+    # User inputs for new account creation
+    username = st.text_input("Username")
+    role = st.selectbox("Select Role", roles) 
+    password = st.text_input("Password", type="password")
+
+    role = "general" if role == "employee" else role
+
+    # Center the buttons using three columns
+    col1, _, _ = st.columns([2, 2, 1])
+
+    # Place both buttons in the left column (col1)
+    with col1:
+        register_clicked = st.button("Register")
+        login_clicked = st.button("Already have an account? Login")
+
+    if register_clicked:
+        try:
+            # Send POST request to FastAPI registration endpoint with user data
+            response = requests.post(
+                "https://finsolve-chatbot-backend.onrender.com/api/v1/auth/register",
+                json={"username": username, "password": password, "role": role}
+            )
+
+            if response.status_code == 200:
+
+                # Switch UI back to login page
+                st.session_state.is_login = True
+
+                # Refresh app to show login UI
+                st.rerun()
+
+                # Show success message on successful registration
+                st.success(f"✅ Registration of {username} successful!")
+
+            else:
+                # Show API error message on failure
+                st.error(f"❌ Registration failed: {response.json().get('msg', 'Unknown error')}")
+
+        except Exception as e:
+            # Handle unexpected errors gracefully
+            st.error(f"⚠️ Error connecting to server: {e}")
+    
+    if login_clicked:
+        # Switch UI back to login page and rerun
+        st.session_state.is_login = True
+        st.rerun()
+
+
+# Function to handle user logout
+def logout():
+    # Clear session data such as auth token and messages
+    st.session_state.token = None 
+    st.session_state.messages = []
+
+    # Refresh app to redirect to login page
+    st.rerun()

src/pages/chat.py

@@ -0,0 +1,79 @@
+import streamlit as st
+import requests
+from .auth import logout
+from src.schemas.schemas import User
+
+# Main chat UI function
+def chat_ui():
+    st.title("FinSolve Technologies Chatbot")
+
+    user:User = st.session_state.user
+
+    # Sidebar section for user info and logout button
+    with st.sidebar:
+        st.write(f"Welcome: {user.username}")
+        st.write(user.role.value.capitalize())
+        
+        # Logout button
+        if st.button("🔒 Logout"):
+            logout()
+
+        # Display clickable link for resources
+        st.markdown(
+            "[View resources](https://github.com/guavacoderepo/Role-based-chatbot/tree/main/backend/resources)",
+            unsafe_allow_html=True
+        )
+
+    # Display existing chat history
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    # Prompt user for new chat input
+    if prompt := st.chat_input("What is up?"):
+        # Store user input in session state
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        
+        # Display user input in the chat interface
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        # Make POST request to backend API
+        try:
+            response = requests.post(
+                "https://finsolve-chatbot-backend.onrender.com/api/v1/chat/start",
+                json={"prompt": prompt},
+                headers={
+                    "Authorization": f"Bearer {st.session_state.token}"
+                }
+            )
+            response.raise_for_status()
+
+            # Check the response content type
+            content_type = response.headers.get("content-type", "")
+
+            # If it's a PDF, show download button
+            if "application/pdf" in content_type:
+                st.success("📄 PDF is ready to download!")
+                st.download_button(
+                    label="📥 Click to download PDF",
+                    data=response.content,
+                    file_name=f"{user.username}_{user.role.value}_summary.pdf",
+                    mime="application/pdf"
+                )
+                assistant_text = "✅ I've generated a PDF summary for you!"
+            else:
+                # If it's text response, parse and display
+                assistant_reply = response.json()
+                assistant_text = assistant_reply.get('response', 'No response received.')
+
+        except Exception as e:
+            # Handle any errors during request
+            assistant_text = f"❌ Error: {str(e)}"
+
+        # Display assistant response
+        with st.chat_message("assistant"):
+            st.markdown(assistant_text)
+
+        # Save assistant response to session
+        st.session_state.messages.append({"role": "assistant", "content": assistant_text})

src/schemas/schemas.py

@@ -0,0 +1,21 @@
+from pydantic import BaseModel
+from enum import Enum
+from typing import Optional
+
+class Roles(Enum):
+    Engineering = "engineering"
+    Employee = "employee"
+    Finance = "finance"
+    Executives = "executives"
+    Marketing = "marketing"
+    HR = "hr"
+
+class User(BaseModel):
+    id:int
+    username:str
+    role: Roles
+
+class UserModel(BaseModel):
+    status: bool
+    user: User
+    token: Optional[str] = None

src/utils/history.py

@@ -0,0 +1,30 @@
+import streamlit as st
+import requests
+
+# Function to retrieve chat history from backend
+def retrieve_history():
+    # Send GET request to backend `/history` endpoint with user credentials
+    response = requests.get(
+        "https://finsolve-chatbot-backend.onrender.com/api/v1/chat/history",
+        headers={
+            "Authorization": f"Bearer {st.session_state.token}"
+        }
+    )
+
+    # Parse JSON response containing previous chat entries
+    mesages = response.json()
+
+    # Initialize the messages list in session state if not already present
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    # Loop through each entry and append user and assistant messages to the session state
+    for entry in mesages:
+        st.session_state.messages.append({
+            "role": "user", 
+            "content": entry["prompt"]
+        })
+        st.session_state.messages.append({
+            "role": "assistant", 
+            "content": entry["response"]
+        })