Update page_files/categorized/Backend/Pdf_ImageExtraction.py

page_files/categorized/Backend/Pdf_ImageExtraction.py

@@ -89,9 +89,8 @@ def robust_json_load(text: str):
     return parsed
 
 
-def get_plot_data_from_llm(GEMINI_MODEL, pdf_path: str):
-    sample_file = genai.upload_file(path=pdf_path)
-
+def get_plot_data_from_llm(client, model_name: str, pdf_path: str):
+    sample_file = client.files.upload(path=pdf_path)
     prompt = prompt = """
 Analyze this PDF and identify ONLY true data plots:
 (Line, Scatter, Bar, Histogram, Heatmap, Box, Violin).
@@ -144,11 +143,11 @@ IMPORTANT: Only include items where figure_kind="plot".
     }
 
     # Some environments accept response_mime_type; if yours errors, remove it.
-    response = GEMINI_MODEL.generate_content(
-        [sample_file, prompt],
-        generation_config=generation_config
+    response = client.models.generate_content(
+        model=model_name,
+        contents=[sample_file, prompt],
+        config={"temperature": 0.0, "max_output_tokens": 4096}
     )
-
     data = robust_json_load(response.text)
 
     # Hard filter if model still returns nonplots
@@ -348,10 +347,9 @@ def main():
                     tmp.write(uploaded_file.getbuffer())
                     st.session_state.temp_pdf = tmp.name
     
-                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)
+                client = genai.Client(api_key=api_key)
+                model_name = get_available_model(api_key)
+                st.session_state.raw_data = get_plot_data_from_llm(client, model_name, st.session_state.temp_pdf)
 
     # Results Display
     if st.session_state.raw_data and st.session_state.temp_pdf: