Update app.py

app.py

@@ -1,9 +1,13 @@
 from openai import OpenAI
-from google.colab import userdata
+# from google.colab import userdata
 import os
 
-openai_key = userdata.get('OPENAI_API_KEY')
-SD_API_KEY = userdata.get('SD_API_KEY')
+# openai_key = userdata.get('OPENAI_API_KEY')
+# SD_API_KEY = userdata.get('SD_API_KEY')
+
+# API 키를 환경 변수에서 읽어오기
+openai_key = os.environ.get('OPENAI_API_KEY')
+SD_API_KEY = os.environ.get('SD_API_KEY')
 
 os.environ["OPENAI_API_KEY"] = openai_key
 # 자동 프롬프트 생성