Update app.py

app.py

@@ -1,20 +1,11 @@
-
 # app.py
+
 import streamlit as st
 import os
 from pathlib import Path
 from typing import Dict, List, Optional, Any
 from datetime import datetime
 from threading import Lock
-from utils.storage_init import initialize_storage_structure, verify_storage_access
-
-# Components imports
-from components.chat import display_chat_interface
-from components.document_preview import enhanced_document_preview
-from components.collection_manager import (
-    display_enhanced_collections,
-    display_collection_content
-)
 
 from utils.database import (
     # Base database operations
@@ -36,20 +27,8 @@ from utils.database import (
 
 from components.chat import display_chat_interface
 
-# Message types
-from langchain_core.messages import (
-    HumanMessage,
-    AIMessage,
-    SystemMessage,
-    BaseMessage
-)
-
-import time
-from threading import Lock
-
 # Create locks for thread-safe operations
 conn_lock = Lock()
-vectorstore_lock = Lock()
 
 def initialize_database():
     """Initialize database with persistent storage."""
@@ -213,296 +192,46 @@ def display_collection_dialog():
                 st.session_state.show_collection_dialog = False
                 st.rerun()
 
-def display_chat_area():
-    """Display the main chat interface."""
-    if not st.session_state.chat_ready:
-        st.title("🤖 Welcome to SYNAPTYX")
-        st.markdown("### Your AI-powered RFP Analysis Assistant")
-        
-        col1, col2 = st.columns(2)
-        with col1:
-            st.markdown("""
-            #### Getting Started:
-            1. Select or create a collection in the sidebar
-            2. Upload your RFP documents
-            3. Start analyzing with AI!
-            """)
-        
-        with col2:
-            st.markdown("#### Example Questions:")
-            examples = [
-                "📊 Summarize the main points of the document",
-                "📝 Draft a 'Why Us' section based on the document",
-                "🎯 Extract key success metrics and outcomes",
-                "💡 What are the innovative solutions mentioned?",
-                "🤝 Analyze the partnership benefits described"
-            ]
-            for example in examples:
-                st.markdown(f"• {example}")
-    else:
-        display_chat_interface()
-
-# app.py
-
-
-
-def display_collections_tab():
-    """Display the collections management interface."""
-    col1, col2 = st.columns([1, 2])
+def display_welcome_screen():
+    """Display the welcome screen with getting started information."""
+    st.title("🤖 Welcome to SYNAPTYX")
+    st.markdown("### Your AI-powered RFP Analysis Assistant")
     
+    col1, col2 = st.columns(2)
     with col1:
-        st.header("Collections")
-        
-        # New collection button
-        if st.button("+ New Collection", use_container_width=True):
-            st.session_state.show_collection_dialog = True
+        st.markdown("""
+        #### Getting Started:
+        1. Upload your RFP documents using the sidebar
+        2. Organize documents into collections
+        3. Start analyzing with AI!
         
-        # Collections list
-        collections = get_collections(st.session_state.db_conn)
-        for collection in collections:
-            collection_button = st.button(
-                f"📁 {collection['name']} ({collection['doc_count']})",
-                key=f"col_{collection['id']}",
-                use_container_width=True
-            )
-            if collection_button:
-                st.session_state.selected_collection = collection
-                st.rerun()
+        You can:
+        - Create multiple collections
+        - Upload documents to specific collections
+        - Search across all documents
+        - Get AI-powered insights and summaries
+        """)
     
     with col2:
-        if st.session_state.get('selected_collection'):
-            display_collection_details(st.session_state.selected_collection)
-
-def show_collection_dialog():
-    """Show dialog for creating/editing collections."""
-    with st.sidebar:
-        st.header("Collection Details")
-        
-        name = st.text_input("Collection Name")
-        description = st.text_area("Description")
-        
-        col1, col2 = st.columns(2)
-        with col1:
-            if st.button("Save", type="primary"):
-                if name:
-                    create_collection(
-                        st.session_state.db_conn,
-                        name,
-                        description
-                    )
-                    st.session_state.show_collection_dialog = False
-                    st.rerun()
-                else:
-                    st.error("Please enter a collection name")
-        
-        with col2:
-            if st.button("Cancel"):
-                st.session_state.show_collection_dialog = False
-                st.rerun()
-                
-def display_collection_details(collection):
-    """Display details and documents for a selected collection."""
-    st.header(collection['name'])
-    st.write(f"Created: {collection['created_at']}")
-    st.write(f"Description: {collection.get('description', 'No description')}")
-    
-    # Documents in collection
-    st.subheader("Documents")
-    documents = get_collection_documents(st.session_state.db_conn, collection['id'])
-    
-    for doc in documents:
-        with st.expander(f"📄 {doc['name']}", expanded=False):
-            col1, col2 = st.columns([3, 1])
-            with col1:
-                st.write(f"Uploaded: {doc['upload_date']}")
-                st.write(f"Size: {len(doc['content'])/1024:.1f} KB")
-            with col2:
-                st.button("Preview", key=f"preview_{doc['id']}", 
-                         on_click=lambda: preview_document(doc))
-                st.button("Remove", key=f"remove_{doc['id']}", 
-                         on_click=lambda: remove_from_collection(doc['id'], collection['id']))
+        st.markdown("#### Example Questions:")
+        examples = [
+            "📊 Summarize the main points of the document",
+            "📝 Draft a 'Why Us' section based on the document",
+            "🎯 Extract key success metrics and outcomes",
+            "💡 What are the innovative solutions mentioned?",
+            "🤝 Analyze the partnership benefits described",
+            "📈 Compare requirements across documents",
+            "🔍 Find similar sections across RFPs"
+        ]
+        for example in examples:
+            st.markdown(f"• {example}")
 
-def display_documents_tab():
-    """Display all documents with metadata and collection management."""
-    st.header("Document Library")
-    
-    # Upload section
-    with st.expander("Upload New Documents", expanded=False):
-        uploaded_files = st.file_uploader(
-            "Upload PDF documents",
-            type=['pdf'],
-            accept_multiple_files=True,
-            help="Limit 200MB per file • PDF"
-        )
-        
-        if uploaded_files:
-            st.subheader("Add to Collection")
-            collections = get_collections(st.session_state.db_conn)
-            collection_options = ["Create New Collection"] + [c['name'] for c in collections]
-            selected_option = st.selectbox("Select Collection", collection_options)
-            
-            if selected_option == "Create New Collection":
-                col1, col2 = st.columns([2, 1])
-                with col1:
-                    new_collection_name = st.text_input("Collection Name")
-                    new_collection_desc = st.text_area("Description")
-                with col2:
-                    if st.button("Create & Upload") and new_collection_name:
-                        collection_id = create_collection(
-                            st.session_state.db_conn,
-                            new_collection_name,
-                            new_collection_desc
-                        )
-                        if collection_id:
-                            handle_document_upload(uploaded_files, collection_id=collection_id)
-            else:
-                if st.button("Upload to Collection"):
-                    collection_id = next(
-                        c['id'] for c in collections if c['name'] == selected_option
-                    )
-                    handle_document_upload(uploaded_files, collection_id=collection_id)
-    
-    # Document list
-    documents = get_all_documents(st.session_state.db_conn)
-    for doc in documents:
-        with st.expander(f"📄 {doc['name']}", expanded=False):
-            col1, col2, col3 = st.columns([2, 1, 1])
-            
-            with col1:
-                st.write("**Metadata:**")
-                st.write(f"• Upload Date: {doc['upload_date']}")
-                st.write(f"• Size: {len(doc['content'])/1024:.1f} KB")
-                st.write(f"• Collections: {', '.join(doc['collections'])}")
-            
-            with col2:
-                st.write("**Actions:**")
-                st.button("Preview", key=f"doc_preview_{doc['id']}")
-                collections = get_collections(st.session_state.db_conn)
-                collection_options = [c['name'] for c in collections]
-                selected_collection = st.selectbox(
-                    "Add to Collection",
-                    collection_options,
-                    key=f"add_to_col_{doc['id']}"
-                )
-                if st.button("Add", key=f"add_btn_{doc['id']}"):
-                    collection_id = next(
-                        c['id'] for c in collections if c['name'] == selected_collection
-                    )
-                    add_document_to_collection(
-                        st.session_state.db_conn,
-                        doc['id'],
-                        collection_id
-                    )
-            
-            with col3:
-                st.write("**Preview:**")
-                if st.button("Show Content", key=f"show_{doc['id']}"):
-                    st.text_area("Content Preview",
-                                doc['content'][:500] + "...",
-                                height=200)
-
-def display_chat_interface_with_sources():
-    """Display chat interface with source references."""
+def display_chat_area():
+    """Display the main chat interface or welcome screen."""
     if not st.session_state.chat_ready:
         display_welcome_screen()
-        return
-        
-    # Display chat history with sources
-    for message in st.session_state.messages:
-        if isinstance(message, HumanMessage):
-            st.chat_message("user").write(message.content)
-        elif isinstance(message, AIMessage):
-            with st.chat_message("assistant"):
-                st.write(message.content)
-                if hasattr(message, 'metadata') and message.metadata.get('sources'):
-                    st.divider()
-                    st.caption("**Sources:**")
-                    for source in message.metadata['sources']:
-                        st.caption(f"• {source}")
-    
-    # Chat input
-    if prompt := st.chat_input("Ask about your documents..."):
-        with st.spinner("Analyzing..."):
-            sources, response = get_response_with_sources(
-                prompt,
-                st.session_state.qa_system
-            )
-            
-            # Add messages to history with metadata
-            human_msg = HumanMessage(content=prompt)
-            ai_msg = AIMessage(
-                content=response,
-                metadata={'sources': sources}
-            )
-            
-            st.session_state.messages.extend([human_msg, ai_msg])
-            st.rerun()
-
-def get_response_with_sources(prompt, qa_system):
-    """Get response from QA system with source documents."""
-    response = qa_system.invoke({
-        "input": prompt,
-        "chat_history": st.session_state.messages
-    })
-    
-    # Extract sources from the chunks used
-    sources = set()
-    if hasattr(response, 'metadata'):
-        for source in response.metadata.get('source_documents', []):
-            sources.add(source.metadata.get('source', 'Unknown source'))
-    
-    return list(sources), response.content
-    
-def display_documents_tab():
-    """Enhanced documents tab with search and preview."""
-    st.header("Document Library")
-    
-    # Search and filter section
-    col1, col2, col3 = st.columns([2, 1, 1])
-    with col1:
-        search_query = st.text_input("🔍 Search documents")
-    with col2:
-        date_filter = st.date_input("Date Filter")
-    with col3:
-        sort_by = st.selectbox("Sort By", ["Name", "Date", "Size"])
-    
-    # Advanced filters
-    with st.expander("Advanced Filters"):
-        filter_cols = st.columns(3)
-        with filter_cols[0]:
-            collections_filter = st.multiselect(
-                "Collections",
-                options=[c['name'] for c in get_collections(st.session_state.db_conn)]
-            )
-        with filter_cols[1]:
-            date_range = st.date_input("Date Range", value=())
-        with filter_cols[2]:
-            st.checkbox("Show only uncategorized")
-    
-    # Build filters dictionary
-    filters = {
-        'collections': collections_filter,
-        'date_range': date_range if len(date_range) == 2 else None
-    }
-    
-    # Get and display documents
-    if search_query:
-        documents = search_documents(
-            st.session_state.db_conn,
-            search_query,
-            filters=filters
-        )
-        if documents:
-            st.success(f"Found {len(documents)} matching documents")
-        else:
-            st.info("No documents found matching your search")
     else:
-        documents = get_all_documents(st.session_state.db_conn)
-    
-    # Display documents
-    for doc in documents:
-        with st.expander(f"📄 {doc['name']}", expanded=False):
-            enhanced_document_preview(doc)
+        display_chat_interface()
 
 def main():
     st.set_page_config(
@@ -511,57 +240,25 @@ def main():
         initial_sidebar_state="expanded"
     )
     
-     # Debug section to understand storage status
-    with st.expander("Storage Debug Information", expanded=True):
-        st.write("Current working directory:", os.getcwd())
-        st.write("Contents of current directory:", os.listdir())
-        if os.path.exists('/data'):
-            st.write("Contents of /data:", os.listdir('/data'))
-        else:
-            st.write("/data directory not found")
-        
-        # Initialize storage
-        base_path = initialize_storage_structure()
-        if base_path:
-            st.write("Storage base path:", str(base_path))
-            if verify_storage_access(base_path):
-                st.success("Storage system verified and ready!")
-            else:
-                st.error("Storage system verification failed!")
-    
-    # Store base_path in session state for other components to use
-    if base_path:
-        st.session_state['storage_path'] = base_path
-    
-    # Initialize database using verified storage path
+    # Initialize database and session state
     if not initialize_database():
         st.error("Failed to initialize database. Please check storage system.")
         return
-        
-    if not initialize_database():
-        st.error("Failed to initialize database. Please contact support.")
-        return
 
     initialize_session_state()
-    display_top_bar()
-
-    # Main tabs
-    tab1, tab2, tab3 = st.tabs(["Chat", "Collections", "Documents"])
     
-    with tab1:
-        display_chat_interface_with_sources()
-    
-    with tab2:
-        display_enhanced_collections()
-    
-    with tab3:
-        display_documents_tab()
+    # Display the top navigation bar
+    display_top_bar()
 
-    # Handle collection dialog if needed
-    if st.session_state.get('show_collection_dialog'):
-        show_collection_dialog()
+    # Display the collection sidebar
+    display_collection_sidebar()
 
+    # Show collection creation dialog if triggered
+    if st.session_state.get('show_collection_dialog', False):
+        display_collection_dialog()
 
+    # Display chat area
+    display_chat_area()
 
 if __name__ == "__main__":
-    main()
+    main()