Update page_files/categorized/Backend/Pdf_ImageExtraction.py

page_files/categorized/Backend/Pdf_ImageExtraction.py

@@ -62,12 +62,8 @@ def get_available_model(api_key: str) -> str:
         _cached_model = "gemini-2.0-flash"
         return _cached_model
  
-GEMINI_MODEL = get_available_model(API_KEY)
-API_URL = (
-    f"https://generativelanguage.googleapis.com/v1beta/"
-    f"models/{GEMINI_MODEL}:generateContent"
-    f"?key={API_KEY}"
-)
+GEMINI_MODEL = None
+API_URL = None
 
 
 
@@ -351,10 +347,11 @@ def main():
                 with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp:
                     tmp.write(uploaded_file.getbuffer())
                     st.session_state.temp_pdf = tmp.name
-
-                # model = init_gemini(api_key)
-                st.session_state.raw_data = get_plot_data_from_llm(GEMINI_MODEL, st.session_state.temp_pdf)
-                st.success(f"Detected {len(st.session_state.raw_data)} candidate plot figures (LLM).")
+    
+                genai.configure(api_key=api_key)                                     
+                model_name = get_available_model(api_key)                           
+                active_model = genai.GenerativeModel(model_name)                    
+                st.session_state.raw_data = get_plot_data_from_llm(active_model, st.session_state.temp_pdf)
 
     # Results Display
     if st.session_state.raw_data and st.session_state.temp_pdf: