Update app.py

app.py

@@ -212,33 +212,210 @@ def display_chat_area():
     else:
         display_chat_interface()
 
+# app.py
+
+
+
+def display_collections_tab():
+    """Display the collections management interface."""
+    col1, col2 = st.columns([1, 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
+        
+        # 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()
+    
+    with col2:
+        if st.session_state.get('selected_collection'):
+            display_collection_details(st.session_state.selected_collection)
+
+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']))
+
+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."""
+    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 main():
-    # Configure the page
-    st.set_page_config(
-        layout="wide",
-        page_title="SYNAPTYX - RFP Analysis Agent",
-        initial_sidebar_state="expanded"
-    )
-
-    # Initialize database and session state
+    st.set_page_config(layout="wide", page_title="SYNAPTYX - RFP Analysis Agent")
+    
     if not initialize_database():
         st.error("Failed to initialize database. Please contact support.")
         return
 
     initialize_session_state()
-
-    # Display the top navigation bar
     display_top_bar()
 
-    # 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()
-
-    # Main chat area
-    display_chat_area()
+    # Main navigation tabs
+    tab1, tab2, tab3 = st.tabs(["Chat", "Collections", "Documents"])
 
+    with tab1:
+        display_chat_interface_with_sources()
+    
+    with tab2:
+        display_collections_tab()
+    
+    with tab3:
+        display_documents_tab()
 if __name__ == "__main__":
     main()