Update app.py

app.py

@@ -13,8 +13,9 @@ from transformers import AutoProcessor, AutoModelForCausalLM, pipeline
 # 설치 과정은 생략 가능하며 필요한 경우에만 실행
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
-# OpenAI 클라이언트 초기화
-client = OpenAI()
+# OpenAI 클라이언트 초기화 (API 키는 환경 변수에서 가져옴)
+openai_api_key = os.getenv("OPENAI_API")
+client = OpenAI(api_key=openai_api_key)
 
 # 번역 모델 추가
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
@@ -338,7 +339,7 @@ class OpenAIGenerationNode:
             # 모델 매핑 - OpenAI는 모델이 다르므로 적절히 대체
             model_mapping = {
 
-                "gpt-4.1": "gpt-4.1-mini"
+                "OpenAI": "gpt-4.1-mini"
             }
             
             openai_model = model_mapping.get(model, "gpt-4.1-mini")  # 기본값은 gpt-4.1-mini