Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os, threading
 
-from agent import run_agent
+from agent import run_agent, get_final_answer
 from helper import get_questions
 
 LEVEL = 1
@@ -23,9 +23,10 @@ def invoke(level, question, file_name, ground_truth, openai_api_key):
 
     with lock:
         os.environ["OPENAI_API_KEY"] = openai_api_key
-        result, matches_ground_truth = run_agent(level, question, file_name, ground_truth)
+        answer, matches_ground_truth = run_agent(level, question, file_name, ground_truth)
+        final_answer = get_final_answer(question, answer)
         del os.environ["OPENAI_API_KEY"]
-        return result, matches_ground_truth
+        return final_answer, matches_ground_truth
 
 gr.close_all()