Add import os

main.py

@@ -1,3 +1,4 @@
+import os
 import openai
 
 openai.api_key = os.environ.get("openai_api_key")