Update app.py

app.py

@@ -14,9 +14,16 @@ def invoke(level, question, file_name, ground_truth, openai_api_key):
         raise gr.Error("OpenAI API Key is required.")
 
     with lock:
-        os.environ["OPENAI_API_KEY"] = openai_api_key
-        answer = run_agent(question, file_name)
-        del os.environ["OPENAI_API_KEY"]
+        answer = ""
+
+        try:
+            os.environ["OPENAI_API_KEY"] = openai_api_key
+            answer = run_agent(question, file_name)
+        except Exception as e:
+            raise gr.Error(e)
+        finally:
+            del os.environ["OPENAI_API_KEY"]
+        
         return answer
 
 gr.close_all()