Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,251 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import pandas as pd
+import os
+import sys
+from typing import Dict, List, Optional
+from datetime import datetime
+
+# Add the src directory to Python path for imports
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+# Import our custom modules
+from rag_system import RAGSystem
+from auth_system import AuthSystem
+from document_processor import DocumentProcessor
+
+# Page config
+st.set_page_config(
+    page_title="FinSolve AI Assistant",
+    page_icon="🤖",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# Custom CSS
+st.markdown("""
+<style>
+    .main-header {
+        background: linear-gradient(90deg, #1f4e79 0%, #2d5aa0 100%);
+        padding: 1rem;
+        border-radius: 10px;
+        margin-bottom: 2rem;
+        color: white;
+        text-align: center;
+    }
+    .role-badge {
+        background-color: #28a745;
+        color: white;
+        padding: 0.25rem 0.5rem;
+        border-radius: 15px;
+        font-size: 0.8rem;
+        font-weight: bold;
+    }
+    .chat-message {
+        padding: 1rem;
+        border-radius: 10px;
+        margin-bottom: 1rem;
+        border-left: 4px solid #1f4e79;
+        background-color: #f8f9fa;
+    }
+    .source-info {
+        background-color: #e9ecef;
+        padding: 0.5rem;
+        border-radius: 5px;
+        margin-top: 0.5rem;
+        font-size: 0.8rem;
+        color: #6c757d;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+def initialize_session_state():
+    """Initialize session state variables"""
+    if 'authenticated' not in st.session_state:
+        st.session_state.authenticated = False
+    if 'user_role' not in st.session_state:
+        st.session_state.user_role = None
+    if 'username' not in st.session_state:
+        st.session_state.username = None
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
+    if 'rag_system' not in st.session_state:
+        st.session_state.rag_system = None
+
+def login_page():
+    """Display login page"""
+    st.markdown('<div class="main-header"><h1>🤖 FinSolve AI Assistant</h1><p>Please login to access role-specific insights</p></div>', unsafe_allow_html=True)
+    
+    col1, col2, col3 = st.columns([1, 2, 1])
+    
+    with col2:
+        st.subheader("🔐 Login")
+        
+        with st.form("login_form"):
+            username = st.text_input("Username", placeholder="Enter your username")
+            password = st.text_input("Password", type="password", placeholder="Enter your password")
+            submit_button = st.form_submit_button("Login", use_container_width=True)
+            
+            if submit_button:
+                auth_system = AuthSystem()
+                if auth_system.authenticate(username, password):
+                    st.session_state.authenticated = True
+                    st.session_state.username = username
+                    st.session_state.user_role = auth_system.get_user_role(username)
+                    st.success(f"Welcome {username}! Your role: {st.session_state.user_role}")
+                    st.rerun()
+                else:
+                    st.error("Invalid credentials. Please try again.")
+        
+        # Demo credentials info
+        st.info("""
+        **Demo Credentials:**
+        - **Finance**: tony.finance / password123
+        - **Marketing**: sarah.marketing / password123  
+        - **HR**: mike.hr / password123
+        - **Engineering**: peter.engineering / password123
+        - **C-Level**: ceo.admin / password123
+        - **Employee**: john.employee / password123
+        """)
+
+def main_app():
+    """Main application interface"""
+    # Initialize RAG system if not done
+    if st.session_state.rag_system is None:
+        with st.spinner("Initializing AI system..."):
+            st.session_state.rag_system = RAGSystem()
+            st.session_state.rag_system.initialize_system()
+    
+    # Header
+    col1, col2, col3 = st.columns([2, 1, 1])
+    with col1:
+        st.markdown('<div class="main-header"><h1>🤖 FinSolve AI Assistant</h1></div>', unsafe_allow_html=True)
+    with col2:
+        st.markdown(f"""
+        <div style="text-align: center; margin-top: 1rem;">
+            <strong>Welcome, {st.session_state.username}</strong><br>
+            <span class="role-badge">{st.session_state.user_role}</span>
+        </div>
+        """, unsafe_allow_html=True)
+    with col3:
+        if st.button("Logout", use_container_width=True):
+            for key in list(st.session_state.keys()):
+                del st.session_state[key]
+            st.rerun()
+    
+    # Sidebar with role info and capabilities
+    with st.sidebar:
+        st.header("📋 Your Access Level")
+        
+        role_permissions = {
+            "Finance": ["Financial reports", "Marketing expenses", "Equipment costs", "Reimbursements"],
+            "Marketing": ["Campaign performance", "Customer feedback", "Sales metrics"],
+            "HR": ["Employee data", "Attendance records", "Payroll", "Performance reviews"],
+            "Engineering": ["Technical architecture", "Development processes", "Operational guidelines"],
+            "C-Level": ["Full access to all company data"],
+            "Employee": ["General company information", "Policies", "Events", "FAQs"]
+        }
+        
+        permissions = role_permissions.get(st.session_state.user_role, [])
+        for perm in permissions:
+            st.markdown(f"✅ {perm}")
+        
+        st.markdown("---")
+        st.header("💡 Sample Questions")
+        
+        sample_questions = {
+            "Finance": [
+                "What was our Q4 2024 revenue?",
+                "Show me marketing expenses breakdown",
+                "What are the major cost drivers?"
+            ],
+            "Marketing": [
+                "How did our Q4 campaigns perform?",
+                "What was our customer acquisition cost?",
+                "Show me ROI for digital marketing"
+            ],
+            "HR": [
+                "How many employees do we have?",
+                "What are the leave policies?",
+                "Show me attendance statistics"
+            ],
+            "Engineering": [
+                "What is our system architecture?",
+                "What technologies do we use?",
+                "Show me our deployment process"
+            ],
+            "C-Level": [
+                "Give me a company overview",
+                "What are our growth metrics?",
+                "Show me all department performance"
+            ],
+            "Employee": [
+                "What are the company policies?",
+                "How do I apply for leave?",
+                "What benefits do we have?"
+            ]
+        }
+        
+        questions = sample_questions.get(st.session_state.user_role, [])
+        for q in questions:
+            if st.button(q, key=f"sample_{q}", use_container_width=True):
+                st.session_state.current_query = q
+    
+    # Main chat interface
+    st.header("💬 Ask Me Anything")
+    
+    # Display chat history
+    for i, (query, response, sources) in enumerate(st.session_state.chat_history):
+        st.markdown(f"""
+        <div class="chat-message">
+            <strong>You:</strong> {query}
+        </div>
+        """, unsafe_allow_html=True)
+        
+        st.markdown(f"""
+        <div class="chat-message">
+            <strong>AI Assistant:</strong> {response}
+            {f'<div class="source-info"><strong>Sources:</strong> {", ".join(sources)}</div>' if sources else ""}
+        </div>
+        """, unsafe_allow_html=True)
+    
+    # Query input
+    query = st.text_input("Ask your question:", 
+                         value=st.session_state.get('current_query', ''),
+                         placeholder="Type your question here...",
+                         key="query_input")
+    
+    col1, col2 = st.columns([1, 4])
+    with col1:
+        ask_button = st.button("Ask", use_container_width=True)
+    
+    if ask_button and query:
+        with st.spinner("Thinking..."):
+            try:
+                response, sources = st.session_state.rag_system.query(
+                    query, 
+                    st.session_state.user_role
+                )
+                
+                # Add to chat history
+                st.session_state.chat_history.append((query, response, sources))
+                
+                # Clear the current query
+                if 'current_query' in st.session_state:
+                    del st.session_state.current_query
+                
+                st.rerun()
+                
+            except Exception as e:
+                st.error(f"Error processing query: {str(e)}")
+
+def main():
+    """Main application entry point"""
+    initialize_session_state()
+    
+    if not st.session_state.authenticated:
+        login_page()
+    else:
+        main_app()
 
-"""
-# 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])),
-    ))
+if __name__ == "__main__":
+    main()