Update app.py

app.py

@@ -142,37 +142,6 @@ 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 document management sidebar."""
@@ -327,11 +296,12 @@ def main():
     display_collection_sidebar()
     display_top_bar()
 
-    # Main content area with tabs
-    tab1, tab2 = st.tabs(["Chat", "Document Store"])
+     tab1, tab2 = st.tabs(["Chat", "Document Store"])
     
     with tab1:
         if st.session_state.chat_ready:
+            # Use the chat component from components/chat.py
+            from components.chat import display_chat_interface
             display_chat_interface()
         else:
             display_welcome_screen()