Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from PyPDF2 import PdfReader
 from io import BytesIO
+import os
+
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_google_genai import GoogleGenerativeAIEmbeddings
 from langchain_community.vectorstores import FAISS
@@ -8,17 +10,8 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains.question_answering import load_qa_chain
 from langchain.prompts import PromptTemplate
 
-# Hardcoded Gemini API keys (for fallback)
-API_KEYS = [
-    "AIzaSyBYbyC4qCJoKxKR-r0oIn4SVqj4CfSdx4s",
-    "AIzaSyBfvYURYVijTurxvFUyV3vZkCojpOAAnFk"
-]
-current_key_index = -1
-
-def switch_api_key():
-    global current_key_index
-    current_key_index = (current_key_index + 1) % len(API_KEYS)
-    return API_KEYS[current_key_index]
+# --- Get API key from environment variable (set in Hugging Face Secrets or .env file) ---
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY", "")
 
 def get_pdf_text(pdf_docs):
     text = ""
@@ -67,21 +60,28 @@ def main():
     st.header("Retrieval-Augmented Generation - Gemini 2.0")
     st.markdown("---")
 
-    # STEP 1: Enter API key
+    # STEP 1: Use API key from env or ask user
     if "api_entered" not in st.session_state:
         st.session_state["api_entered"] = False
     if "pdf_processed" not in st.session_state:
         st.session_state["pdf_processed"] = False
 
+    api_key = GOOGLE_API_KEY
+
     if not st.session_state["api_entered"]:
-        user_api_key = st.text_input("Enter your Gemini API key", type="password")
-        if st.button("Continue") and user_api_key:
-            st.session_state["user_api_key"] = user_api_key
+        if not api_key:
+            user_api_key = st.text_input("Enter your Gemini API key", type="password")
+            if st.button("Continue") and user_api_key:
+                st.session_state["user_api_key"] = user_api_key
+                st.session_state["api_entered"] = True
+                st.experimental_rerun()
+            st.stop()
+        else:
+            st.session_state["user_api_key"] = api_key
             st.session_state["api_entered"] = True
             st.experimental_rerun()
-        st.stop()
 
-    api_key = st.session_state.get("user_api_key", switch_api_key())
+    api_key = st.session_state.get("user_api_key", "")
 
     # STEP 2: Upload PDF(s)
     if not st.session_state["pdf_processed"]: