Update util.py

util.py

@@ -1,6 +1,7 @@
 #import base64, cv2
 import pandas as pd
-from openai import OpenAI
+from google import genai
+#from openai import OpenAI
 
 def get_questions(file_path):
     df = pd.read_json(file_path, lines=True)
@@ -60,11 +61,20 @@ def get_final_answer(model, question, answer):
 
         """
 
-    client = OpenAI()
+    #client = OpenAI()
     
-    completion = client.chat.completions.create(
-        messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}],
-        model=model
-    )
+    #completion = client.chat.completions.create(
+    #    model=model,
+    #    messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}]
+    #)
+
+    client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
-    return completion.choices[0].message.content
+    response = client.models.generate_content(
+        model=model, 
+        contents=[prompt_template]
+    )
+    
+    #return completion.choices[0].message.content
+    
+    return response.text