Update app.py

app.py

@@ -11,11 +11,18 @@ from PIL import Image
 from transformers import AutoProcessor, AutoModelForCausalLM, pipeline
 
 # 설치 과정은 생략 가능하며 필요한 경우에만 실행
-subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+try:
+    subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
+except:
+    print("Flash attention installation skipped")
 
 # OpenAI 클라이언트 초기화 (API 키는 환경 변수에서 가져옴)
 openai_api_key = os.getenv("OPENAI_API")
-client = OpenAI(api_key=openai_api_key)
+if openai_api_key:
+    client = OpenAI(api_key=openai_api_key)
+else:
+    print("Warning: OPENAI_API key not found in environment variables")
+    client = None
 
 # 번역 모델 추가
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
@@ -57,8 +64,12 @@ def florence_caption(image):
 # JSON 파일 로드 함수
 def load_json_file(file_name):
     file_path = os.path.join("data", file_name)
-    with open(file_path, "r") as file:
-        return json.load(file)
+    try:
+        with open(file_path, "r") as file:
+            return json.load(file)
+    except:
+        print(f"Warning: Could not load {file_name}. Using empty list.")
+        return []
 
 # JSON 데이터 로드
 ARTFORM = load_json_file("artform.json")
@@ -317,7 +328,7 @@ class PromptGenerator:
 
 class OpenAIGenerationNode:
     def __init__(self):
-        self.client = OpenAI()
+        self.client = client
         self.prompts_dir = "./prompts"
         os.makedirs(self.prompts_dir, exist_ok=True)
 
@@ -336,6 +347,9 @@ class OpenAIGenerationNode:
 
     def generate(self, input_text, happy_talk, compress, compression_level, poster, custom_base_prompt=""):
         try:
+            if not self.client:
+                return "Error: OpenAI API key not found. Please set OPENAI_API environment variable."
+                
             # 고정 모델: gpt-4.1-mini
             openai_model = "gpt-4.1-mini"
 
@@ -458,6 +472,14 @@ footer {
     font-size: 1.1em !important;
     padding: 12px 24px !important;
 }
+.section-header {
+    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
+    color: white;
+    padding: 10px 15px;
+    border-radius: 8px;
+    margin-bottom: 15px;
+    font-weight: bold;
+}
 """
 
 def create_interface():
@@ -489,105 +511,96 @@ def create_interface():
         with gr.Row():
             # 왼쪽 패널 - 설정
             with gr.Column(scale=2):
-                with gr.Box(elem_classes="settings-container"):
-                    gr.Markdown("### 🎯 기본 설정")
-                    seed = gr.Slider(0, 30000, label='시드값', step=1, value=random.randint(0,30000))
-                    custom = gr.Textbox(label="✏️ 커스텀 프롬프트 (한글 가능)", placeholder="원하는 설명을 입력하세요...")
-                    subject = gr.Textbox(label="🎭 주제 (선택사항)", placeholder="예: 아름다운 여성, 귀여운 고양이 등...")
-                    
-                    # 전역 옵션 설정
-                    gr.Markdown("### ⚡ 빠른 설정")
-                    global_option = gr.Radio(
-                        ["Disabled", "Random", "No Figure Rand"],
-                        label="모든 옵션 일괄 설정:",
-                        value="Disabled",
-                        info="한 번에 모든 설정을 변경합니다"
-                    )
+                gr.HTML('<div class="section-header">🎯 기본 설정</div>')
+                seed = gr.Slider(0, 30000, label='시드값', step=1, value=random.randint(0,30000))
+                custom = gr.Textbox(label="✏️ 커스텀 프롬프트 (한글 가능)", placeholder="원하는 설명을 입력하세요...")
+                subject = gr.Textbox(label="🎭 주제 (선택사항)", placeholder="예: 아름다운 여성, 귀여운 고양이 등...")
+                
+                # 전역 옵션 설정
+                gr.HTML('<div class="section-header">⚡ 빠른 설정</div>')
+                global_option = gr.Radio(
+                    ["Disabled", "Random", "No Figure Rand"],
+                    label="모든 옵션 일괄 설정:",
+                    value="Disabled",
+                    info="한 번에 모든 설정을 변경합니다"
+                )
                 
                 # 상세 설정들
                 with gr.Accordion("🎨 아트폼 & 사진 타입", open=False):
-                    with gr.Box(elem_classes="settings-container"):
-                        artform = gr.Dropdown(["disabled", "random"] + ARTFORM, label="아트폼", value="disabled")
-                        photo_type = gr.Dropdown(["disabled", "random"] + PHOTO_TYPE, label="사진 타입", value="disabled")
+                    artform = gr.Dropdown(["disabled", "random"] + ARTFORM, label="아트폼", value="disabled")
+                    photo_type = gr.Dropdown(["disabled", "random"] + PHOTO_TYPE, label="��진 타입", value="disabled")
             
                 with gr.Accordion("👤 캐릭터 설정", open=False):
-                    with gr.Box(elem_classes="settings-container"):
-                        body_types = gr.Dropdown(["disabled", "random"] + BODY_TYPES, label="체형", value="disabled")
-                        default_tags = gr.Dropdown(["disabled", "random"] + DEFAULT_TAGS, label="기본 태그", value="disabled")
-                        roles = gr.Dropdown(["disabled", "random"] + ROLES, label="역할", value="disabled")
-                        hairstyles = gr.Dropdown(["disabled", "random"] + HAIRSTYLES, label="헤어스타일", value="disabled")
-                        clothing = gr.Dropdown(["disabled", "random"] + CLOTHING, label="의상", value="disabled")
+                    body_types = gr.Dropdown(["disabled", "random"] + BODY_TYPES, label="체형", value="disabled")
+                    default_tags = gr.Dropdown(["disabled", "random"] + DEFAULT_TAGS, label="기본 태그", value="disabled")
+                    roles = gr.Dropdown(["disabled", "random"] + ROLES, label="역할", value="disabled")
+                    hairstyles = gr.Dropdown(["disabled", "random"] + HAIRSTYLES, label="헤어스타일", value="disabled")
+                    clothing = gr.Dropdown(["disabled", "random"] + CLOTHING, label="의상", value="disabled")
             
                 with gr.Accordion("🏞️ 장면 설정", open=False):
-                    with gr.Box(elem_classes="settings-container"):
-                        place = gr.Dropdown(["disabled", "random"] + PLACE, label="장소", value="disabled")
-                        lighting = gr.Dropdown(["disabled", "random"] + LIGHTING, label="조명", value="disabled")
-                        composition = gr.Dropdown(["disabled", "random"] + COMPOSITION, label="구도", value="disabled")
-                        pose = gr.Dropdown(["disabled", "random"] + POSE, label="포즈", value="disabled")
-                        background = gr.Dropdown(["disabled", "random"] + BACKGROUND, label="배경", value="disabled")
+                    place = gr.Dropdown(["disabled", "random"] + PLACE, label="장소", value="disabled")
+                    lighting = gr.Dropdown(["disabled", "random"] + LIGHTING, label="조명", value="disabled")
+                    composition = gr.Dropdown(["disabled", "random"] + COMPOSITION, label="구도", value="disabled")
+                    pose = gr.Dropdown(["disabled", "random"] + POSE, label="포즈", value="disabled")
+                    background = gr.Dropdown(["disabled", "random"] + BACKGROUND, label="배경", value="disabled")
             
                 with gr.Accordion("🎭 스타일 & 아티스트", open=False):
-                    with gr.Box(elem_classes="settings-container"):
-                        additional_details = gr.Dropdown(["disabled", "random"] + ADDITIONAL_DETAILS, label="추가 디테일", value="disabled")
-                        photography_styles = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHY_STYLES, label="사진 스타일", value="disabled")
-                        device = gr.Dropdown(["disabled", "random"] + DEVICE, label="촬영 장비", value="disabled")
-                        photographer = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHER, label="사진작가", value="disabled")
-                        artist = gr.Dropdown(["disabled", "random"] + ARTIST, label="아티스트", value="disabled")
-                        digital_artform = gr.Dropdown(["disabled", "random"] + DIGITAL_ARTFORM, label="디지털 아트폼", value="disabled")
+                    additional_details = gr.Dropdown(["disabled", "random"] + ADDITIONAL_DETAILS, label="추가 디테일", value="disabled")
+                    photography_styles = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHY_STYLES, label="사진 스타일", value="disabled")
+                    device = gr.Dropdown(["disabled", "random"] + DEVICE, label="촬영 장비", value="disabled")
+                    photographer = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHER, label="사진작가", value="disabled")
+                    artist = gr.Dropdown(["disabled", "random"] + ARTIST, label="아티스트", value="disabled")
+                    digital_artform = gr.Dropdown(["disabled", "random"] + DIGITAL_ARTFORM, label="디지털 아트폼", value="disabled")
                 
                 generate_button = gr.Button("🚀 프롬프트 생성", variant="primary", elem_classes="generate-button")
 
             # 중간 패널 - 이미지와 출력
             with gr.Column(scale=2):
                 with gr.Accordion("🖼️ 이미지 캡션 생성", open=False):
-                    with gr.Box(elem_classes="settings-container"):
-                        input_image = gr.Image(label="이미지 업로드 (선택사항)", type="pil")
-                        caption_output = gr.Textbox(label="생성된 캡션", lines=3)
-                        with gr.Row():
-                            create_caption_button = gr.Button("📝 캡션 생성", variant="secondary")
-                            add_caption_button = gr.Button("➕ 프롬프트에 추가", variant="secondary")
-
-                with gr.Box(elem_classes="output-container"):
-                    gr.Markdown("### 📋 생성된 프롬프트")
-                    output = gr.Textbox(label="메인 프롬프트", lines=4)
-                    with gr.Accordion("고급 출력 옵션", open=False):
-                        t5xxl_output = gr.Textbox(label="T5XXL", lines=2)
-                        clip_l_output = gr.Textbox(label="CLIP L", lines=2)
-                        clip_g_output = gr.Textbox(label="CLIP G", lines=2)
+                    input_image = gr.Image(label="이미지 업로드 (선택사항)", type="pil")
+                    caption_output = gr.Textbox(label="생성된 캡션", lines=3)
+                    with gr.Row():
+                        create_caption_button = gr.Button("📝 캡션 생성", variant="secondary")
+                        add_caption_button = gr.Button("➕ 프롬프트에 추가", variant="secondary")
+
+                gr.HTML('<div class="section-header">📋 생성된 프롬프트</div>')
+                output = gr.Textbox(label="메인 프롬프트", lines=4)
+                with gr.Accordion("고급 출력 옵션", open=False):
+                    t5xxl_output = gr.Textbox(label="T5XXL", lines=2)
+                    clip_l_output = gr.Textbox(label="CLIP L", lines=2)
+                    clip_g_output = gr.Textbox(label="CLIP G", lines=2)
             
             # 오른쪽 패널 - OpenAI
             with gr.Column(scale=2):
-                with gr.Box(elem_classes="prompt-generator-container"):
-                    gr.Markdown("### 🤖 OpenAI 프롬프트 향상")
-                    gr.HTML("<p style='text-align: center; color: #95a5a6;'>사용 모델: gpt-4.1-mini</p>")
-                    
-                    with gr.Box(elem_classes="settings-container"):
-                        with gr.Row():
-                            happy_talk = gr.Checkbox(label="😊 Happy Talk", value=True, info="더 상세한 설명")
-                            compress = gr.Checkbox(label="🗜️ 압축", value=True, info="출력 길이 제한")
-                        
-                        compression_level = gr.Radio(
-                            ["soft", "medium", "hard"], 
-                            label="압축 강도", 
-                            value="hard",
-                            visible=True
-                        )
-                        
-                        poster = gr.Checkbox(label="🎬 영화 포스터 스타일", value=False)
-                        
-                        custom_base_prompt = gr.Textbox(
-                            label="🛠️ 커스텀 베이스 프롬프트", 
-                            lines=5,
-                            placeholder="OpenAI에게 특별한 지시사항을 입력하세요..."
-                        )
-                    
-                    generate_text_button = gr.Button("✨ AI로 프롬프트 향상", variant="primary", elem_classes="openai-button")
-                    
-                    text_output = gr.Textbox(
-                        label="🎯 AI 향상 결과", 
-                        lines=10,
-                        elem_classes="output-container"
-                    )
+                gr.HTML('<div class="section-header">🤖 OpenAI 프롬프트 향상</div>')
+                gr.HTML("<p style='text-align: center; color: #95a5a6;'>사용 모델: gpt-4.1-mini</p>")
+                
+                with gr.Row():
+                    happy_talk = gr.Checkbox(label="😊 Happy Talk", value=True, info="더 상세한 설명")
+                    compress = gr.Checkbox(label="🗜️ 압축", value=True, info="출력 길이 제한")
+                
+                compression_level = gr.Radio(
+                    ["soft", "medium", "hard"], 
+                    label="압축 강도", 
+                    value="hard",
+                    visible=True
+                )
+                
+                poster = gr.Checkbox(label="🎬 영화 포스터 스타일", value=False)
+                
+                custom_base_prompt = gr.Textbox(
+                    label="🛠️ 커스텀 베이스 프롬프트", 
+                    lines=5,
+                    placeholder="OpenAI에게 특별한 지시사항을 입력하세요..."
+                )
+                
+                generate_text_button = gr.Button("✨ AI로 프롬프트 향상", variant="primary", elem_classes="openai-button")
+                
+                text_output = gr.Textbox(
+                    label="🎯 AI 향상 결과", 
+                    lines=10,
+                    elem_classes="output-container"
+                )
 
         # 이벤트 핸들러들
         def create_caption(image):
@@ -631,7 +644,7 @@ def create_interface():
 
         # 전역 옵션 변경 함수
         def update_all_options(choice):
-            updates = {}
+            updates = []
             dropdown_list = [
                 artform, photo_type, body_types, default_tags, roles, hairstyles, clothing,
                 place, lighting, composition, pose, background, additional_details,
@@ -639,20 +652,22 @@ def create_interface():
             ]
             
             if choice == "Disabled":
-                for dropdown in dropdown_list:
-                    updates[dropdown] = gr.update(value="disabled")
+                updates = [gr.update(value="disabled") for _ in dropdown_list]
             elif choice == "Random":
-                for dropdown in dropdown_list:
-                    updates[dropdown] = gr.update(value="random")
+                updates = [gr.update(value="random") for _ in dropdown_list]
             else:  # No Figure Random
+                updates = []
                 # 캐릭터 관련 설정은 disabled
-                for dropdown in [photo_type, body_types, default_tags, roles, hairstyles, clothing, pose, additional_details]:
-                    updates[dropdown] = gr.update(value="disabled")
-                # 나머지는 random
-                for dropdown in [artform, place, lighting, composition, background, photography_styles, device, photographer, artist, digital_artform]:
-                    updates[dropdown] = gr.update(value="random")
+                character_dropdowns = [photo_type, body_types, default_tags, roles, hairstyles, clothing, pose, additional_details]
+                other_dropdowns = [artform, place, lighting, composition, background, photography_styles, device, photographer, artist, digital_artform]
+                
+                for dropdown in dropdown_list:
+                    if dropdown in character_dropdowns:
+                        updates.append(gr.update(value="disabled"))
+                    else:
+                        updates.append(gr.update(value="random"))
             
-            return list(updates.values())
+            return updates
         
         global_option.change(
             update_all_options,