Update app.py

app.py

@@ -2,8 +2,15 @@ import gradio as gr
 import os, threading
 
 from agents import run_gaia
+from crew import get_crew
 from helper import get_questions
 
+###
+LLM_MANAGER = "o1"
+LLM_AGENTS = "gpt-4o"
+VERBOSE = False
+###
+
 lock = threading.Lock()
 
 def invoke(level, question, file_name, ground_truth, openai_api_key):
@@ -18,7 +25,8 @@ def invoke(level, question, file_name, ground_truth, openai_api_key):
 
         try:
             os.environ["OPENAI_API_KEY"] = openai_api_key
-            answer = run_gaia(question, file_name)
+            #answer = run_gaia(question, file_name)
+            answer = str(get_crew(LLM_MANAGER, LLM_AGENTS, VERBOSE).kickoff(inputs={"topic": question}))
         except Exception as e:
             raise gr.Error(e)
         finally: