Update app.py

app.py

@@ -5,15 +5,15 @@ from util import get_questions
 
 QUESTION_FILE_PATH = "data/gaia_validation_20.jsonl"
 
-def invoke(level, question, file_name, ground_truth, openai_api_key, google_api_key):
+def invoke(level, question, file_name, ground_truth, openai_api_key, gemini_api_key):
     if not question:
         raise gr.Error("Question is required.")
 
     if not openai_api_key:
         raise gr.Error("OpenAI API Key is required.")
     
-    if not google_api_key:
-        raise gr.Error("Google API Key is required.")
+    if not gemini_api_key:
+        raise gr.Error("Gemini API Key is required.")
 
     if file_name:
         file_name = f"data/{file_name}"
@@ -25,14 +25,14 @@ def invoke(level, question, file_name, ground_truth, openai_api_key, google_api_
 
         try:
             os.environ["OPENAI_API_KEY"] = openai_api_key
-            os.environ["GOOGLE_API_KEY"] = google_api_key
+            os.environ["GEMINI_API_KEY"] = gemini_api_key
             
             answer = run_crew(question, file_name)
         except Exception as e:
             raise gr.Error(e)
         finally:
             del os.environ["OPENAI_API_KEY"]
-            del os.environ["GOOGLE_API_KEY"]
+            del os.environ["GEMINI_API_KEY"]
         
         return answer
 
@@ -44,7 +44,7 @@ demo = gr.Interface(fn=invoke,
                             gr.Textbox(label="File Name"),
                             gr.Textbox(label="Ground Truth"),
                             gr.Textbox(label="OpenAI API Key", type="password"),
-                            gr.Textbox(label="Google API Key", type="password")],
+                            gr.Textbox(label="Gemini API Key", type="password")],
                     outputs=[gr.Textbox(label="Answer", lines=1, interactive=False)],
                     title="General AI Assistant (GAIA) 🤖🤝🤖",
                     description=os.environ["DESCRIPTION"],