Nechba commited on
Commit
5af38ac
·
verified ·
1 Parent(s): 94eeb49

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +10 -13
utils.py CHANGED
@@ -1,7 +1,7 @@
1
- import google.generativeai as genai
 
2
 
3
-
4
- genai.configure(api_key="AIzaSyAP85jSUKncrIGOAhm3Gvo-TYra_e1wmEA")
5
  import os
6
  import pandas as pd
7
  import io
@@ -94,22 +94,19 @@ def analyze_pdf_images_with_gemini(pdf_bytes):
94
  # Initialize Gemini client
95
 
96
  images = pdf_to_images(pdf_bytes)
97
-
98
  # Prepare image parts
99
  contents = [question]
100
  for img in images:
101
  img_bytes = pil_image_to_bytes(img)
102
- contents.append({
103
- 'mime_type': 'image/jpeg',
104
- 'data': img_bytes
105
- })
106
- #contents.append(genai.types.Part.from_bytes(data=img_bytes, mime_type="image/jpeg"))
107
 
108
  # Generate content using Gemini
109
- model = genai.GenerativeModel('gemini-2.0-flash') # Updated model name
110
-
111
- # Send both the prompt and image to Gemini
112
- response = model.generate_content(contents)
 
113
 
114
  return response
115
 
 
1
+ from google import genai
2
+ from google.genai import types
3
 
4
+ #genai.configure(api_key="AIzaSyAP85jSUKncrIGOAhm3Gvo-TYra_e1wmEA")
 
5
  import os
6
  import pandas as pd
7
  import io
 
94
  # Initialize Gemini client
95
 
96
  images = pdf_to_images(pdf_bytes)
97
+ client = genai.Client(api_key="AIzaSyAP85jSUKncrIGOAhm3Gvo-TYra_e1wmEA")
98
  # Prepare image parts
99
  contents = [question]
100
  for img in images:
101
  img_bytes = pil_image_to_bytes(img)
102
+ contents.append(types.Part.from_bytes(data=img_bytes, mime_type="image/jpeg"))
 
 
 
 
103
 
104
  # Generate content using Gemini
105
+ response = client.models.generate_content(
106
+ model="gemini-2.0-flash",
107
+ contents=contents
108
+ )
109
+
110
 
111
  return response
112