Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from crew import run_crew
 from utils import get_questions
 
-def _run(question, openai_api_key, gemini_api_key, file_name = ""):
+def _run(question, openai_api_key, gemini_api_key, anthropic_api_key, file_name = ""):
     """
     Run GAIA General AI Assistant to answer a question.
 
@@ -11,6 +11,7 @@ def _run(question, openai_api_key, gemini_api_key, file_name = ""):
         question (str): The question to answer
         openai_api_key (str): OpenAI API key
         gemini_api_key (str): Gemini API key
+        anthropic_api_key (str): Anthropic API key
         file_name (str): Optional file name
 
     Returns:
@@ -25,6 +26,9 @@ def _run(question, openai_api_key, gemini_api_key, file_name = ""):
     if not gemini_api_key:
         raise gr.Error("Gemini API Key is required.")
 
+    if not anthropic_api_key:
+        raise gr.Error("Anthropic API Key is required.")
+        
     if file_name:
         file_name = f"data/{file_name}"
     
@@ -36,6 +40,7 @@ def _run(question, openai_api_key, gemini_api_key, file_name = ""):
         try:
             os.environ["OPENAI_API_KEY"] = openai_api_key
             os.environ["GEMINI_API_KEY"] = gemini_api_key
+            os.environ["ANTHROPIC_API_KEY"] = anthropic_api_key
             
             answer = run_crew(question, file_name)
         except Exception as e:
@@ -43,6 +48,7 @@ def _run(question, openai_api_key, gemini_api_key, file_name = ""):
         finally:
             del os.environ["OPENAI_API_KEY"]
             del os.environ["GEMINI_API_KEY"]
+            del os.environ["ANTHROPIC_API_KEY"]
         
         return answer
 
@@ -81,9 +87,13 @@ with gr.Blocks() as gaia:
                     label="Gemini API Key *",
                     type="password"
                 )
+                anthropic_api_key = gr.Textbox(
+                    label="Anthropic API Key *",
+                    type="password"
+                )
             with gr.Row():
                 clear_btn = gr.ClearButton(
-                    components=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key]
+                    components=[question, level, ground_truth, file_name]
                 )
                 submit_btn = gr.Button("Submit", variant="primary")
         with gr.Column(scale=1):
@@ -95,7 +105,7 @@ with gr.Blocks() as gaia:
 
     submit_btn.click(
         fn=_run,
-        inputs=[question, openai_api_key, gemini_api_key, file_name],
+        inputs=[question, openai_api_key, gemini_api_key, anthropic_api_key, file_name],
         outputs=answer
     )
 
@@ -104,7 +114,7 @@ with gr.Blocks() as gaia:
     gr.Examples(
         label="Level 1",
         examples=get_questions(QUESTION_FILE_PATH, 1),
-        inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
+        inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
         outputs=answer,
         cache_examples=False
     )
@@ -112,7 +122,7 @@ with gr.Blocks() as gaia:
     gr.Examples(
         label="Level 2",
         examples=get_questions(QUESTION_FILE_PATH, 2),
-        inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
+        inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
         outputs=answer,
         cache_examples=False
     )
@@ -120,7 +130,7 @@ with gr.Blocks() as gaia:
     gr.Examples(
         label="Level 3",
         examples=get_questions(QUESTION_FILE_PATH, 3),
-        inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key],
+        inputs=[question, level, ground_truth, file_name, openai_api_key, gemini_api_key, anthropic_api_key],
         outputs=answer,
         cache_examples=False
     )