Update crew.py

crew.py

@@ -418,4 +418,40 @@ def run_crew(question, file_path):
     print(f"Final answer: {final_answer}")
     print("###")
     
-    return final_answer
+    return final_answer
+
+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.
+        Your final answer must be a number and/or string 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 USD, $, percent, or % unless specified otherwise.
+        If you are asked for a string, don't use articles, neither abbreviations (for example 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.
+        If the final answer is a number, use a number not a word.
+        If the final answer is a string, 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, use a space character after each comma and start with a lowercase character.
+        Do not add any content to the final answer that is not in the initial answer.
+
+        **Question:** """ + question + """
+        
+        **Initial answer:** """ + answer + """
+        
+        **Example 1:** What is the biggest city in California? Los Angeles
+        **Example 2:** How many 'r's are in strawberry? 3
+        **Example 3:** What is the opposite of black? White
+        **Example 4:** What are the first 5 numbers in the Fibonacci sequence? 0, 1, 1, 2, 3
+        **Example 5:** What is the opposite of bad, worse, worst? good, better, best
+        
+        **Final answer:** 
+
+        """
+
+    client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
+
+    response = client.models.generate_content(
+        model=model, 
+        contents=[prompt_template]
+    )
+    
+    return response.text