Update app.py

app.py

@@ -1,4 +1,5 @@
 import importlib.util
+import os
 import sys
 import traceback
 
@@ -20,7 +21,7 @@ def import_module_from_string(name: str, source: str):
     sys.modules[name] = module
     globals()[name] = module
 
-def main(code: str, key: str):
+def main(code: str, key: str = os.getenv("OPENAI_API_KEY")):
     success = False
     while not success:
         try: