Update app.py

app.py

@@ -4,6 +4,7 @@ from utils.document_processor import DocumentProcessor
 from utils.case_manager import CaseManager
 from utils.legal_notebook_interface import LegalNotebookInterface
 from utils.case_manager_interface import CaseManagerInterface
+from utils.admin_interface import AdminInterface
 from datetime import datetime
 import os
 from pathlib import Path
@@ -119,102 +120,71 @@ def main():
         # Initialize components
         case_manager, vector_store, doc_processor = init_components()
 
-        # Sidebar navigation (clean and minimal)
-        tab = st.sidebar.radio(
-            "Navigation",
-            ["📁 Case Manager", "📄 Document Analysis", "🤖 Legal Assistant"]
-        )
+        # Session state for user authentication
+        if 'user_authenticated' not in st.session_state:
+            st.session_state.user_authenticated = False
+            st.session_state.current_user = None
 
-        if tab == "📁 Case Manager":
-            st.title("📁 Case Management")
+        # Sidebar navigation with admin access
+        if st.session_state.user_authenticated:
+            is_admin = st.session_state.current_user == 'admin'
             
-            # Case creation form
-            with st.expander("➕ Create New Case", expanded=True):
-                with st.form("create_case_form"):
-                    title = st.text_input("Case Title")
-                    description = st.text_area("Description", placeholder="Enter case details...")
-                    case_type = st.selectbox(
-                        "Case Type", 
-                        ["Judgment", "Contract", "MOU", "Will", "Other"]
-                    )
-                    create_case = st.form_submit_button("Create Case")
-
-                    if create_case and title:
-                        try:
-                            case_id = case_manager.create_case(title, description, case_type)
-                            st.success(f"Case '{title}' created successfully!")
-                        except Exception as e:
-                            st.error(f"Error creating case: {str(e)}")
-
-            # List cases
-            st.subheader("📜 Cases")
-            try:
-                cases = case_manager.get_all_cases()
-                if cases:
-                    for case in cases:
-                        with st.expander(f"📂 {case['title']}", expanded=False):
-                            st.markdown(f"**Type:** {case['case_type']}")
-                            st.markdown(f"**Description:** {case['description']}")
-                            st.markdown(f"**Created:** {case['created_at']}")
-
-                            # List documents
-                            documents = case_manager.list_documents(case["id"])
-                            if documents:
-                                st.markdown("### Documents")
-                                for doc in documents:
-                                    st.markdown(f"- 📄 {doc['title']} (Added: {doc['added_at']})")
-                            
-                            # Document upload
-                            uploaded_file = st.file_uploader(
-                                "Upload Document",
-                                key=f"upload_{case['id']}"
-                            )
-                            
-                            if uploaded_file:
-                                with st.status("Processing document...", expanded=True) as status:
-                                    try:
-                                        # Process document
-                                        text, chunks, metadata = doc_processor.process_and_tag_document(uploaded_file)
-                                        
-                                        # Add to vector store
-                                        for chunk in chunks:
-                                            vector_store.add_document(
-                                                doc_id=metadata['doc_id'],
-                                                text=chunk['text'],
-                                                metadata={
-                                                    **metadata,
-                                                    'chunk_index': chunk.get('chunk_id', 0),
-                                                    'total_chunks': len(chunks)
-                                                }
-                                            )
-                                        
-                                        # Add to case
-                                        doc_data = {
-                                            "id": metadata['doc_id'],
-                                            "title": uploaded_file.name,
-                                            "text": text,
-                                            "metadata": metadata,
-                                            "chunks": chunks
-                                        }
-                                        case_manager.add_document(case["id"], doc_data)
-                                        
-                                        st.success(f"Document '{uploaded_file.name}' added successfully!")
-                                        
-                                    except Exception as e:
-                                        st.error(f"Error processing document: {str(e)}")
-                else:
-                    st.info("No cases yet. Create your first case using the form above.")
-                    
-            except Exception as e:
-                st.error(f"Error loading cases: {str(e)}")
-
-        elif tab == "📄 Document Analysis":
-            notebook = LegalNotebookInterface(case_manager, vector_store, doc_processor)
-            notebook.render()
-
-        elif tab == "🤖 Legal Assistant":
-            notebook = LegalNotebookInterface(case_manager, vector_store, doc_processor)
-            notebook.render()
+            navigation_options = ["📁 Case Manager", "📄 Document Analysis", "🤖 Legal Assistant"]
+            if is_admin:
+                navigation_options.append("⚙️ Admin")
+            
+            tab = st.sidebar.radio("Navigation", navigation_options)
+            
+            # Add logout button
+            if st.sidebar.button("Logout"):
+                st.session_state.user_authenticated = False
+                st.session_state.current_user = None
+                st.rerun()
+
+            if tab == "📁 Case Manager":
+                case_manager_interface = CaseManagerInterface(case_manager, vector_store, doc_processor)
+                case_manager_interface.render()
+
+            elif tab == "📄 Document Analysis":
+                notebook = LegalNotebookInterface(case_manager, vector_store, doc_processor)
+                notebook.render()
+
+            elif tab == "🤖 Legal Assistant":
+                notebook = LegalNotebookInterface(case_manager, vector_store, doc_processor)
+                notebook.render()
+
+            elif tab == "⚙️ Admin" and is_admin:
+                admin_interface = AdminInterface(case_manager, vector_store, doc_processor)
+                admin_interface.render()
+
+        else:
+            # Login form
+            st.sidebar.title("Login")
+            with st.sidebar.form("login_form"):
+                username = st.text_input("Username")
+                password = st.text_input("Password", type="password")
+                submit = st.form_submit_button("Login")
+
+                if submit:
+                    if username == "admin" and password == "demo":
+                        st.session_state.user_authenticated = True
+                        st.session_state.current_user = "admin"
+                        st.rerun()
+                    # Here you can add more user authentication logic
+                    else:
+                        st.error("Invalid credentials")
+
+            # Show welcome message for non-authenticated users
+            st.markdown("""
+            ## Welcome to SuoMoto.AI
+            Please login to access the legal analysis tools.
+            
+            ### Available Features:
+            - 📁 Case Management
+            - 📄 Document Analysis
+            - 🤖 Legal Assistant
+            - ⚙️ Admin Panel (for administrators)
+            """)
 
     except Exception as e:
         st.error("An unexpected error occurred")