Testing

app.py

@@ -1,6 +1,7 @@
 import os, threading
 import gradio as gr
 from crew import run_parallel_crew
+from crew import run_crew
 from utils import get_questions
 
 
@@ -43,7 +44,8 @@ def ask(question, openai_api_key, gemini_api_key, anthropic_api_key, file_name =
             os.environ["GEMINI_API_KEY"] = gemini_api_key
             os.environ["MODEL_API_KEY"] = anthropic_api_key
             
-            answer = run_parallel_crew(question, file_name)
+            #answer = run_parallel_crew(question, file_name)
+            answer = run_crew(question, file_name)
         except Exception as e:
             raise gr.Error(e)
         finally:

crew.py

@@ -144,7 +144,43 @@ class GAIACrew():
             verbose=True
         )
 
-import concurrent.futures
+def run_crew(question, file_path):
+    """
+    Orchestrates the GAIA crew to answer a question, optionally with a file.
+
+    Args:
+        question (str): The user's question.
+        file_path (str): Optional path to a data file to include in the prompt.
+
+    Returns:
+        str: The final answer from the manager agent.
+    """
+    # Build the final prompt, including file JSON if needed
+    final_question = question
+    if file_path:
+        if is_ext(file_path, ".csv") or is_ext(file_path, ".xls") \
+           or is_ext(file_path, ".xlsx") or is_ext(file_path, ".json") \
+           or is_ext(file_path, ".jsonl"):
+            json_data = read_file_json(file_path)
+            final_question = f"{question} JSON data:\n{json_data}."
+        else:
+            final_question = f"{question} File path: {file_path}."
+    
+    # Instantiate the crew and kick off the workflow
+    crew_instance = GAIACrew()
+    crew = crew_instance.get_crew()
+    answer = crew.kickoff(inputs={"question": final_question})
+    
+    # Post-process through the final-answer model
+    final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(answer))
+
+    # Debug logging
+    print(f"=> Initial question: {question}")
+    print(f"=> Final question: {final_question}")
+    print(f"=> Initial answer: {answer}")
+    print(f"=> Final answer: {final_answer}")
+    
+    return final_answer
 
 import concurrent.futures