Update app.py

app.py

@@ -5,10 +5,13 @@ from util import get_questions
 
 QUESTION_FILE_PATH = "data/gaia_validation_20.jsonl"
 
-def invoke(level, question, file_name, ground_truth, google_api_key):
+def invoke(level, question, file_name, ground_truth, openai_api_key, 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.")
 
@@ -21,12 +24,14 @@ def invoke(level, question, file_name, ground_truth, google_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,6 +43,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) 🤖🤝🤖",