Update app.py

app.py

@@ -519,6 +519,30 @@ def create_app():
             "상품 이미지를 업로드하고, 제품명, 배경 옵션, 추가 요청사항을 입력하면 Gemini API를 통해 영어 프롬프트를 생성하고, 해당 프롬프트로 이미지가 생성됩니다.\n\n"
             "단일 이미지 생성과 4장 이미지 생성 모두 가능합니다."
         )
+        
+        with gr.Row():
+            # API 키 입력 필드 추가
+            gemini_api_key = gr.Textbox(
+                label="Gemini API 키", 
+                type="password", 
+                placeholder="API 키를 입력하세요",
+                interactive=True
+            )
+
+            # API 키 설정 버튼 추가
+            api_key_btn = gr.Button("API 키 설정")
+            api_key_status = gr.Textbox(label="API 키 상태", interactive=False)
+
+        def set_api_key(api_key):
+            os.environ["GEMINI_API_KEY"] = api_key
+            return "API 키가 성공적으로 설정되었습니다."
+
+        api_key_btn.click(
+            fn=set_api_key, 
+            inputs=[gemini_api_key], 
+            outputs=[api_key_status]
+        )
+
         with gr.Row():
             with gr.Column(scale=1):
                 product_name = gr.Textbox(label="상품명 (한국어 입력)", placeholder="예: 스킨케어 튜브, 스마트워치, 향수, 운동화 등", interactive=True)
@@ -603,8 +627,7 @@ def create_app():
                         tech_dropdown: gr.update(visible=(bg_type == "테크놀로지 배경")),
                         colorful_dropdown: gr.update(visible=(bg_type == "컬러풀 패턴 배경")),
                         abstract_dropdown: gr.update(visible=(bg_type == "추상/특수 배경")),
-                        jewelry_dropdown: gr.update(visible=(bg_type == "쥬얼리 배경"))  # 쥬얼리 드롭다운 추가
-                        
+                        jewelry_dropdown: gr.update(visible=(bg_type == "쥬얼리 배경"))
                     }
                 background_type.change(
                     fn=update_dropdowns,