Update app.py

Added functionality to generate a new index everytime

app.py

@@ -17,6 +17,7 @@ from langchain_pinecone import PineconeVectorStore
 from pinecone import Pinecone, ServerlessSpec
 from pinecone import PineconeApiException, NotFoundException
 import shutil
+import uuid
 
 from dotenv import load_dotenv
 load_dotenv()
@@ -46,6 +47,16 @@ if 'chat_history' not in st.session_state:
     st.session_state.chat_history = []
 if 'chat_enabled' not in st.session_state:
     st.session_state.chat_enabled = False
+if 'session_id' not in st.session_state:
+    # Generate a unique session ID using UUID
+    st.session_state.session_id = str(uuid.uuid4())[:8]
+
+def get_session_index_name():
+    """Get unique index name for current session"""
+    base_name = "docdb"  # Using a short base name to leave room for the unique identifier
+    unique_id = st.session_state.session_id
+    # Combine base name with unique ID, ensuring total length is under 45 chars
+    return f"{base_name}-{unique_id}"  # This will be like "docdb-12345678"
 
 if not st.session_state.initialized:
     # Clear everything only on first run or page refresh
@@ -57,6 +68,15 @@ if not st.session_state.initialized:
     st.session_state.vectorstore = None
     st.session_state.retriever = None
     st.session_state.initialized = True
+    
+    # Delete any existing index for this session (in case of page refresh)
+    try:
+        pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
+        index_name = get_session_index_name()
+        if index_name in pc.list_indexes().names():
+            pc.delete_index(index_name)
+    except Exception as e:
+        st.error(f"Error cleaning up old index: {str(e)}")
 
 def save_uploaded_file(uploaded_file):
     """Save uploaded file to the data directory"""
@@ -127,7 +147,7 @@ def process_documents(uploaded_files_dict):
             
             # Initialize Pinecone
             pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
-            index_name = os.getenv("VECTORDB_NAME")
+            index_name = get_session_index_name()
 
             try:
                 # Recreate the index
@@ -316,7 +336,7 @@ def process_and_chat():
             # Initialize components for chat
             llm = ChatGroq(temperature=0, model_name="mixtral-8x7b-32768", groq_api_key=os.getenv("GROQ_API_KEY"))
             pc = Pinecone(api_key=os.getenv("PINECONE_API_KEY"))
-            index_name = os.getenv("VECTORDB_NAME")
+            index_name = get_session_index_name()
             pc_index = pc.Index(index_name)
             
             # Create vectorstore