Update app.py

app.py

@@ -24,6 +24,15 @@ import google.generativeai as genai
 from datetime import datetime
 
 
+def load_environment():
+    load_dotenv()
+    # Ensure HF_TOKEN is available
+    if "HUGGINGFACEHUB_API_TOKEN" not in os.environ and "HF_TOKEN" in os.environ:
+        os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.environ["HF_TOKEN"] 
+    if "GOOGLE_API_KEY" not in os.environ:
+        raise ValueError("GOOGLE_API_KEY not found in environment variables.")
+    genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+    
 class GeminiLLM(Runnable):
     def __init__(self, model_name="models/gemini-1.5-pro-latest", api_key=None):
         self.api_key = api_key or os.environ["GOOGLE_API_KEY"]
@@ -77,12 +86,7 @@ vectorstore_global = None
 if "feedback_log" not in st.session_state:
     st.session_state.feedback_log = []
 
-def load_environment():
-    load_dotenv()
-    # Ensure HF_TOKEN is available
-    if "HUGGINGFACEHUB_API_TOKEN" not in os.environ and "HF_TOKEN" in os.environ:
-        os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.environ["HF_TOKEN"] 
-    genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+
 
 from keybert import KeyBERT
 from sentence_transformers import CrossEncoder