Update util.py

util.py

@@ -1,7 +1,6 @@
 import os
 import pandas as pd
-#from google import genai
-from openai import OpenAI
+from google import genai
 
 def get_questions(file_path):
     df = pd.read_json(file_path, lines=True)
@@ -13,28 +12,6 @@ def get_questions(file_path):
 
     return result
 
-#def get_img_b64(file_path):
-#    with open(file_path, "rb") as file:
-#        return base64.b64encode(file.read()).decode("utf-8")
-
-#def get_imgs_b64(file_path):
-#    video = cv2.VideoCapture(file_path)
-    
-#    result = []
-    
-#    while video.isOpened():
-#        success, frame = video.read()
-
-#        if not success:
-#            break
-        
-#        _, buffer = cv2.imencode(".png", frame)
-#        result.append(base64.b64encode(buffer).decode("utf-8"))
-
-#    video.release()
-    
-#    return result
-
 def get_final_answer(model, question, answer):
     prompt_template = """
         You are an expert question answering assistant. Given a question and an initial answer, your task is to provide the final answer.
@@ -62,20 +39,11 @@ def get_final_answer(model, question, answer):
 
         """
 
-    client = OpenAI()
-    
-    completion = client.chat.completions.create(
-        model=model,
-        messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}]
-    )
-
-    #client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+    client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
 
-    #response = client.models.generate_content(
-    #    model=model, 
-    #    contents=[prompt_template]
-    #)
-    
-    return completion.choices[0].message.content
+    response = client.models.generate_content(
+        model=model, 
+        contents=[prompt_template]
+    )
     
-    #return response.text
+    return response.text