Update app.py

app.py

@@ -115,10 +115,14 @@ def initialize_session_state():
         st.session_state.chat_ready = False
     if 'current_collection' not in st.session_state:
         st.session_state.current_collection = None
+    if 'current_document' not in st.session_state:
+        st.session_state.current_document = None
     if 'processed_files' not in st.session_state:
         st.session_state.processed_files = []
     if 'vector_stores' not in st.session_state:
         st.session_state.vector_stores = {}
+    if 'selected_collection' not in st.session_state:
+        st.session_state.selected_collection = None
 
 def display_top_bar():
     """Display the application's top navigation bar."""
@@ -280,25 +284,29 @@ def main():
         initial_sidebar_state="expanded"
     )
     
-    # Initialize database and session state
+    # Initialize database silently
     if not initialize_database():
-        st.error("Failed to initialize database. Please check storage system.")
+        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()
+    # Main content area with tabs
+    tab1, tab2 = st.tabs(["Chat", "Document Store"])
+    
+    with tab1:
+        if st.session_state.chat_ready:
+            display_chat_interface()
+        else:
+            display_welcome_screen()
+    
+    with tab2:
+        display_documents_tab()
 
-    # Show collection creation dialog if triggered
-    if st.session_state.get('show_collection_dialog', False):
+    # Show collection creation dialog if needed
+    if st.session_state.get('show_collection_dialog'):
         display_collection_dialog()
 
-    # Display chat area
-    display_chat_area()
-
 if __name__ == "__main__":
     main()