Fix langchain dependency for HF Space

app.py

@@ -12,6 +12,14 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
+# Initialize session state immediately
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+if 'rag_system' not in st.session_state:
+    st.session_state.rag_system = None
+if 'initialized' not in st.session_state:
+    st.session_state.initialized = False
+
 # Custom CSS for better styling
 st.markdown("""
 <style>
@@ -50,14 +58,6 @@ st.markdown("""
 </style>
 """, unsafe_allow_html=True)
 
-# Initialize session state
-if 'messages' not in st.session_state:
-    st.session_state.messages = []
-if 'rag_system' not in st.session_state:
-    st.session_state.rag_system = None
-if 'initialized' not in st.session_state:
-    st.session_state.initialized = False
-
 # RAG System Functions
 def initialize_rag_system(api_key):
     """Initialize the RAG system with all components"""
@@ -71,8 +71,6 @@ def initialize_rag_system(api_key):
             import chromadb
             from chromadb.config import Settings
             import google.generativeai as genai
-            from datasets import load_dataset
-            from tqdm import tqdm
             import re
         except ImportError as e:
             st.error(f"Import error: {e}")
@@ -93,7 +91,7 @@ def initialize_rag_system(api_key):
         except:
             collection = chroma_client.create_collection(
                 name=collection_name,
-                metadata={"description": "ML/AI knowledge base from The Pile dataset"}
+                metadata={"description": "ML/AI knowledge base"}
             )
         
         # Check if collection already has data
@@ -149,7 +147,7 @@ def initialize_rag_system(api_key):
                 metadatas=chunk_metadatas
             )
         
-        # Initialize Gemini using direct API instead of LangChain
+        # Initialize Gemini using direct API
         genai.configure(api_key=api_key)
         
         return {