Update util.py

util.py

@@ -9,4 +9,33 @@ def get_questions():
     for index, row in df.iterrows():
         result.append([row["Level"], row["Question"], row["file_name"], row["Final answer"]])
 
-    return result
+    return result
+
+def get_final_answer(question, initial_answer):
+    prompt_template = """
+        You are given a question and an initial answer.
+        Your final answer must be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+        If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
+        If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
+        If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+        **Question:** """ + question + """
+        **Initial answer:** """ + initial_answer + """
+        **Example:** What is the opposite of white? Black
+        **Final answer:** 
+        """
+
+    client = OpenAI()
+    completion = client.chat.completions.create(
+        messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}],
+        model="gpt-4o"
+    )
+
+    final_answer = completion.choices[0].message.content
+    
+    print("###")
+    print(question)
+    print(initial_answer)
+    print(final_answer)
+    print("###")
+    
+    return final_answer