Update src/streamlit_app.py

src/streamlit_app.py

@@ -30,27 +30,16 @@ with st.container():
     st.markdown("Ask financial questions about ITC Ltd. based on transcript data, powered by AI.")
     st.markdown('</div>', unsafe_allow_html=True)
 
-# Sidebar
-with st.sidebar:
-    st.header("⚙️ Settings")
-    
-    # Load API key securely from Hugging Face secrets
-    GOOGLE_API_KEY = st.secrets.get("GOOGLE_API_KEY", None)
-    
-    if GOOGLE_API_KEY:
-        st.success("✅ Google API Key loaded successfully!")
-    else:
-        st.error("❌ Google API Key not found in Hugging Face Secrets. Please set 'GOOGLE_API_KEY'.")
 
-    # Theme toggle
-    theme = st.selectbox("Theme", ["Light", "Dark"], index=0)
-    if theme == "Dark":
-        st.markdown("<style>body { background-color: #343a40; color: white; } .main { background-color: #495057; }</style>", unsafe_allow_html=True)
+    
+# Load API key securely from Hugging Face secrets
+GOOGLE_API_KEY = st.secrets.get("GOOGLE_API_KEY")
+   
 
 # Initialize Chroma DB
 @st.cache_resource
-def initialize_vectorstore(_api_key):
-    embedding = GoogleGenerativeAIEmbeddings(model="models/embedding-001", google_api_key=_api_key)
+def initialize_vectorstore(GOOGLE_API_KEY):
+    embedding = GoogleGenerativeAIEmbeddings(model="models/embedding-001", google_api_key=GOOGLE_API_KEY)
     zip_path = "./chroma_db1.zip"
     extract_dir = "chroma_db2"
     if os.path.exists(zip_path):
@@ -71,10 +60,10 @@ def initialize_vectorstore(_api_key):
 
 # Initialize vectorstore and retriever
 retriever = None
-if GOOGLE_API_KEY:
-    vectorstore = initialize_vectorstore(GOOGLE_API_KEY)
-    if vectorstore:
-        retriever = vectorstore.as_retriever(search_type="mmr", search_kwargs={"k": 3, "lambda_mult": 1})
+
+vectorstore = initialize_vectorstore(GOOGLE_API_KEY)
+if vectorstore:
+    retriever = vectorstore.as_retriever(search_type="mmr", search_kwargs={"k": 3, "lambda_mult": 1})
 
 # Prompt template
 prompt = ChatPromptTemplate.from_messages([