Update util.py

util.py

@@ -1,7 +1,7 @@
 #import base64, cv2
 import pandas as pd
-#from google import genai
-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)
@@ -46,6 +46,7 @@ def get_final_answer(model, question, answer):
         If the final answer is a word, start with an uppercase character.
         If the final answer is a comma-separated list of numbers, use a space character after each comma.
         If the final answer is a comma-separated list of strings, start with a lowercase character.
+        Do not add any content to the final answer that is not in the initial answer.
 
         **Question:** """ + question + """
         
@@ -61,20 +62,20 @@ def get_final_answer(model, question, answer):
 
         """
 
-    client = OpenAI()
+    #client = OpenAI()
     
-    completion = client.chat.completions.create(
-        model=model,
-        messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}]
-    )
+    #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"])
+    client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
 
-    #response = client.models.generate_content(
-    #    model=model, 
-    #    contents=[prompt_template]
-    #)
+    response = client.models.generate_content(
+        model=model, 
+        contents=[prompt_template]
+    )
     
-    return completion.choices[0].message.content
+    #return completion.choices[0].message.content
     
-    #return response.text
+    return response.text