Update page_files/categorized/Backend/PDF_DataExtraction.py

page_files/categorized/Backend/PDF_DataExtraction.py

@@ -151,13 +151,8 @@ def get_gemini_model(api_key: str) -> str:
         _run_startup_probes(_cached_gemini_model)
         return _cached_gemini_model
 
-
-GEMINI_MODEL = get_gemini_model(GEMINI_API_KEY)
-GEMINI_API_URL = (
-    f"https://generativelanguage.googleapis.com/v1beta/"
-    f"models/{GEMINI_MODEL}:generateContent"
-    f"?key={GEMINI_API_KEY}"
-)
+GEMINI_MODEL = None
+GEMINI_API_URL = None
 
 # ── OpenAI / GPT-4o ───────────────────────────────────────────────────────────
 GPT_MODEL       = "gpt-4o"
@@ -2023,6 +2018,16 @@ def run_pipeline(
     doi_override:      str = "",
     progress_callback: Any = None,
 ) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame, List[Chunk], List[str], Dict]:
+    global GEMINI_MODEL, GEMINI_API_URL          
+    if not GEMINI_MODEL:       
+        key = os.getenv("GEMINI_API_KEY", "")                  
+        GEMINI_MODEL = get_gemini_model(key)                   
+        GEMINI_API_URL = (                                     
+            f"https://generativelanguage.googleapis.com/v1beta/" 
+            f"models/{GEMINI_MODEL}:generateContent"       
+            f"?key={key}"  
+        )                 
+
     """
     Full dual-LLM consensus pipeline.