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 typing import Dict, Any, Optional
 from dataclasses import dataclass
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.vectorstores import FAISS
@@ -307,8 +306,8 @@ def initialize_chat_from_collection():
             if documents:
                 embeddings = get_embeddings_model()
                 text_splitter = RecursiveCharacterTextSplitter(
-                    chunk_size=1000,
-                    chunk_overlap=200,
+                    chunk_size=700,
+                    chunk_overlap=100,
                     length_function=len,
                     separators=["\n\n", "\n", " ", ""]
                 )
@@ -403,10 +402,14 @@ def display_collection_dialog():
                 st.rerun()
 
 def display_welcome_screen():
-    """Display the welcome screen with getting started information."""
+    """Display the welcome screen with getting started information and enhanced features."""
     st.title("🤖 Welcome to SYNAPTYX")
     st.markdown("### Your AI-powered RFP Analysis Assistant")
     
+    # Check for existing documents
+    documents = get_all_documents(st.session_state.db_conn)
+    collections = get_collections(st.session_state.db_conn)
+    
     col1, col2 = st.columns(2)
     with col1:
         st.markdown("""
@@ -421,6 +424,20 @@ def display_welcome_screen():
         - Search across all documents
         - Get AI-powered insights and summaries
         """)
+        
+        # Add action buttons if documents exist
+        if documents:
+            st.success(f"🎉 You have {len(documents)} documents ready for analysis!")
+            col_a, col_b = st.columns(2)
+            with col_a:
+                if st.button("Start Chatting", use_container_width=True):
+                    if initialize_chat_from_collection():
+                        st.session_state.chat_ready = True
+                        st.rerun()
+            with col_b:
+                if st.button("View Documents", use_container_width=True):
+                    st.session_state.show_document_store = True
+                    st.rerun()
     
     with col2:
         st.markdown("#### Example Questions:")
@@ -435,13 +452,71 @@ def display_welcome_screen():
         ]
         for example in examples:
             st.markdown(f"• {example}")
+        
+        # Add collection stats if they exist
+        if collections:
+            st.markdown("---")
+            st.markdown("#### Your Collections:")
+            for collection in collections[:3]:  # Show top 3 collections
+                with st.container():
+                    st.markdown(f"""
+                    🗂️ **{collection['name']}**  
+                    Documents: {collection['doc_count']}
+                    """)
+            if len(collections) > 3:
+                st.caption("... and more collections")
 
 def display_chat_area():
-    """Display the main chat interface or welcome screen."""
+    """Display the main chat interface or welcome screen with proper state management."""
     if not st.session_state.chat_ready:
         display_welcome_screen()
     else:
-        display_chat_interface()
+        col1, col2 = st.columns([3, 1])
+        with col1:
+            display_chat_interface()
+        with col2:
+            # Document context panel
+            st.markdown("### Current Context")
+            if st.session_state.current_collection:
+                st.info(f"📁 Using Collection: {st.session_state.current_collection['name']}")
+            else:
+                st.info("📚 Using All Documents")
+            
+            # Quick actions
+            st.markdown("### Quick Actions")
+            if st.button("🔄 New Chat", use_container_width=True):
+                reset_chat_state()
+                st.rerun()
+            if st.button("📋 Export Chat", use_container_width=True):
+                export_chat_history()
+            
+            # Example questions for quick reference
+            with st.expander("💡 Question Examples", expanded=False):
+                examples = [
+                    "Summarize main points",
+                    "Extract requirements",
+                    "Compare solutions",
+                    "Analyze pricing"
+                ]
+                for example in examples:
+                    if st.button(example, key=f"example_{example}"):
+                        # Set the example as the current question
+                        st.session_state.current_question = example
+                        st.rerun()
+
+def export_chat_history():
+    """Export current chat history."""
+    if st.session_state.messages:
+        chat_text = "\n\n".join([
+            f"{'User' if isinstance(m, HumanMessage) else 'Assistant'}: {m.content}"
+            for m in st.session_state.messages
+        ])
+        st.download_button(
+            "Download Chat History",
+            chat_text,
+            file_name=f"chat_export_{datetime.now().strftime('%Y%m%d_%H%M%S')}.txt",
+            mime="text/plain"
+        )
 
 def main():
     """Main application function."""