Update app.py

app.py

@@ -5,12 +5,12 @@ from util import get_questions
 
 QUESTION_FILE_PATH = "data/gaia_validation_20.jsonl"
 
-def invoke(level, question, file_name, ground_truth, openai_api_key):
+def invoke(level, question, file_name, ground_truth, google_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 file_name:
         file_name = f"data/{file_name}"
@@ -21,13 +21,13 @@ def invoke(level, question, file_name, ground_truth, openai_api_key):
         answer = ""
 
         try:
-            os.environ["OPENAI_API_KEY"] = openai_api_key
+            os.environ["GOOGLE_API_KEY"] = google_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"]
         
         return answer
 
@@ -38,7 +38,7 @@ demo = gr.Interface(fn=invoke,
                             gr.Markdown(label="Question"),
                             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")],
                     outputs=[gr.Textbox(label="Answer", lines=1, interactive=False)],
                     title="General AI Assistant (GAIA) 🤖🤝🤖",
                     description=os.environ["DESCRIPTION"],