Update app.py

app.py

@@ -6,7 +6,6 @@ from pathlib import Path
 from typing import Dict, List, Optional, Any
 from datetime import datetime
 from threading import Lock
-from components.styled_response import display_styled_chat_message
 
 from utils.database import (
    # Base database operations
@@ -142,11 +141,52 @@ def display_top_bar():
         if st.session_state.current_collection:
             st.info(f"📁 Active Collection: {st.session_state.current_collection['name']}")
 
+def display_chat_interface():
+    """Display the chat interface with formatted responses."""
+    # Initialize chat history if needed
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    # Display chat history
+    for message in st.session_state.messages:
+        is_user = isinstance(message, HumanMessage)
+        display_styled_chat_message(message, is_user=is_user)
+
+    # Chat input
+    if prompt := st.chat_input("Ask about your documents..."):
+        # Add user message
+        st.session_state.messages.append(HumanMessage(content=prompt))
+        display_styled_chat_message(prompt, is_user=True)
+
+        with st.spinner("Thinking..."):
+            # Get AI response
+            response = st.session_state.qa_system.invoke({
+                "input": prompt,
+                "chat_history": st.session_state.messages
+            })
+
+            # Add AI message with metadata
+            ai_message = AIMessage(
+                content=response.content,
+                metadata={'sources': response.metadata.get('sources', [])}
+            )
+            st.session_state.messages.append(ai_message)
+            display_styled_chat_message(ai_message)
+
 def display_collection_sidebar():
-    """Display the collection management sidebar."""
+    """Display the document management sidebar."""
     with st.sidebar:
         st.title("📚 Document Manager")
 
+        # Upload Section
+        st.header("Upload Documents", anchor=False)
+        uploaded_files = st.file_uploader(
+            "Upload PDF documents",
+            type=['pdf'],
+            accept_multiple_files=True,
+            help="Limit 200MB per file • PDF"
+        )
+
         # Collection Selection
         st.header("Collections")
         collections = get_collections(st.session_state.db_conn)
@@ -173,15 +213,6 @@ def display_collection_sidebar():
         else:
             st.session_state.current_collection = None
 
-        # Upload Section
-        st.header("Upload Documents")
-        uploaded_files = st.file_uploader(
-            "Upload PDF documents",
-            type=['pdf'],
-            accept_multiple_files=True,
-            help="Limit 200MB per file • PDF"
-        )
-
         # Process uploads automatically when files are selected
         if uploaded_files:
             if 'processed_files' not in st.session_state or uploaded_files != st.session_state.processed_files:
@@ -291,6 +322,8 @@ def main():
         return
 
     initialize_session_state()
+        # Display sidebar first
+    display_collection_sidebar()
     display_top_bar()
 
     # Main content area with tabs