Update app.py

app.py

@@ -369,44 +369,26 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
                 char_limit = compression_chars[compression_level]
                 base_prompt += f" Compress the output to be concise while retaining key visual details. MAX OUTPUT SIZE no more than {char_limit} characters."
 
-            # OpenAI API 요청 포맷 사용
-            response = self.client.responses.create(
+            # 올바른 OpenAI API 호출 형식
+            response = self.client.chat.completions.create(
                 model=openai_model,
-                input=[
+                messages=[
                     {
                         "role": "system",
-                        "content": [
-                            {
-                                "type": "input_text",
-                                "text": "You are a helpful assistant. Try your best to give best response possible to user."
-                            }
-                        ]
+                        "content": "You are a helpful assistant. Try your best to give best response possible to user."
                     },
                     {
                         "role": "user",
-                        "content": [
-                            {
-                                "type": "input_text",
-                                "text": f"{base_prompt}\nDescription: {input_text}"
-                            }
-                        ]
+                        "content": f"{base_prompt}\nDescription: {input_text}"
                     }
                 ],
-                text={
-                    "format": {
-                        "type": "text"
-                    }
-                },
-                reasoning={},
-                tools=[],
                 temperature=1,
-                max_output_tokens=2048,
-                top_p=1,
-                store=True
+                max_tokens=2048,
+                top_p=1
             )
             
             # 응답 추출
-            output = response.content[0].text
+            output = response.choices[0].message.content
             
             # 출력 정리
             if ": " in output:
@@ -416,108 +398,198 @@ You are allowed to make up film and branding names, and do them like 80's, 90's
                 if len(sentences) > 1:
                     output = ". ".join(sentences[1:]).strip()
             
+            # 프롬프트 저장
+            self.save_prompt(output)
+            
             return output
 
         except Exception as e:
             print(f"An error occurred: {e}")
             return f"Error occurred while processing the request: {str(e)}"
 
+# 개선된 CSS
 css = """
 footer {
     visibility: hidden;
 }
+.main-title {
+    text-align: center;
+    margin: 20px 0;
+}
+.main-title h1 {
+    color: #2c3e50;
+    font-size: 2.5em;
+}
+.badge-container {
+    display: flex;
+    justify-content: center;
+    gap: 15px;
+    margin: 20px 0;
+}
+.prompt-generator-container {
+    background: #f8f9fa;
+    border-radius: 10px;
+    padding: 20px;
+    margin: 10px 0;
+}
+.settings-container {
+    background: #ffffff;
+    border-radius: 8px;
+    padding: 15px;
+    margin: 10px 0;
+    box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+}
+.output-container {
+    background: #e8f4f8;
+    border-radius: 8px;
+    padding: 15px;
+    margin: 10px 0;
+}
+.generate-button {
+    background: #3498db !important;
+    color: white !important;
+    font-size: 1.1em !important;
+    padding: 12px 24px !important;
+    margin: 10px 0 !important;
+}
+.openai-button {
+    background: #27ae60 !important;
+    color: white !important;
+    font-size: 1.1em !important;
+    padding: 12px 24px !important;
+}
 """
 
 def create_interface():
     prompt_generator = PromptGenerator()
     openai_node = OpenAIGenerationNode()
+    
     with gr.Blocks(theme="soft", css=css) as demo:
-
+        # 헤더
+        with gr.Row(elem_classes="main-title"):
+            gr.HTML("""
+                <h1>🎨 Flux Prompt Generator</h1>
+                <p style="text-align: center; color: #7f8c8d; font-size: 1.1em;">
+                    한글 입력 지원 | AI 기반 프롬프트 생성기
+                </p>
+            """)
         
-        gr.HTML("<h1>Flux Prompt Generator(프롬프트에 한글 입력 가능)</h1>")
-
-    gr.HTML(
-        """
-        <div class='container' style='display:flex; justify-content:center; gap:12px;'>
-            <a href="https://huggingface.co/spaces/openfree/Best-AI" target="_blank">
-                <img src="https://img.shields.io/static/v1?label=OpenFree&message=BEST%20AI%20Services&color=%230000ff&labelColor=%23000080&logo=huggingface&logoColor=%23ffa500&style=for-the-badge" alt="OpenFree badge">
-            </a>
-    
-            <a href="https://discord.gg/openfreeai" target="_blank">
-                <img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%230000ff&labelColor=%23800080&logo=discord&logoColor=white&style=for-the-badge" alt="Discord badge">
-            </a>
-        </div>
-            """
-    )
-
+        # 배지
+        with gr.Row(elem_classes="badge-container"):
+            gr.HTML("""
+                <a href="https://huggingface.co/spaces/openfree/Best-AI" target="_blank">
+                    <img src="https://img.shields.io/static/v1?label=OpenFree&message=BEST%20AI%20Services&color=%230000ff&labelColor=%23000080&logo=huggingface&logoColor=%23ffa500&style=for-the-badge" alt="OpenFree badge">
+                </a>
+                <a href="https://discord.gg/openfreeai" target="_blank">
+                    <img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&color=%230000ff&labelColor=%23800080&logo=discord&logoColor=white&style=for-the-badge" alt="Discord badge">
+                </a>
+            """)
         
+        # 메인 컨테이너
         with gr.Row():
+            # 왼쪽 패널 - 설정
             with gr.Column(scale=2):
-                with gr.Accordion("Basic Settings"):
-                    seed = gr.Slider(0, 30000, label='Seed', step=1, value=random.randint(0,30000))
-                    custom = gr.Textbox(label="Custom Input Prompt (optional)")
-                    subject = gr.Textbox(label="Subject (optional)")
+                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="예: 아름다운 여성, 귀여운 고양이 등...")
                     
-                    # Add the radio button for global option selection
+                    # 전역 옵션 설정
+                    gr.Markdown("### ⚡ 빠른 설정")
                     global_option = gr.Radio(
                         ["Disabled", "Random", "No Figure Rand"],
-                        label="Set all options to:",
-                        value="Disabled"
+                        label="모든 옵션 일괄 설정:",
+                        value="Disabled",
+                        info="한 번에 모든 설정을 변경합니다"
                     )
                 
-                with gr.Accordion("Artform and Photo Type", open=False):
-                    artform = gr.Dropdown(["disabled", "random"] + ARTFORM, label="Artform", value="disabled")
-                    photo_type = gr.Dropdown(["disabled", "random"] + PHOTO_TYPE, label="Photo Type", value="disabled")
+                # 상세 설정들
+                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")
             
-                with gr.Accordion("Character Details", open=False):
-                    body_types = gr.Dropdown(["disabled", "random"] + BODY_TYPES, label="Body Types", value="disabled")
-                    default_tags = gr.Dropdown(["disabled", "random"] + DEFAULT_TAGS, label="Default Tags", value="disabled")
-                    roles = gr.Dropdown(["disabled", "random"] + ROLES, label="Roles", value="disabled")
-                    hairstyles = gr.Dropdown(["disabled", "random"] + HAIRSTYLES, label="Hairstyles", value="disabled")
-                    clothing = gr.Dropdown(["disabled", "random"] + CLOTHING, label="Clothing", 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")
             
-                with gr.Accordion("Scene Details", open=False):
-                    place = gr.Dropdown(["disabled", "random"] + PLACE, label="Place", value="disabled")
-                    lighting = gr.Dropdown(["disabled", "random"] + LIGHTING, label="Lighting", value="disabled")
-                    composition = gr.Dropdown(["disabled", "random"] + COMPOSITION, label="Composition", value="disabled")
-                    pose = gr.Dropdown(["disabled", "random"] + POSE, label="Pose", value="disabled")
-                    background = gr.Dropdown(["disabled", "random"] + BACKGROUND, label="Background", 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")
             
-                with gr.Accordion("Style and Artist", open=False):
-                    additional_details = gr.Dropdown(["disabled", "random"] + ADDITIONAL_DETAILS, label="Additional Details", value="disabled")
-                    photography_styles = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHY_STYLES, label="Photography Styles", value="disabled")
-                    device = gr.Dropdown(["disabled", "random"] + DEVICE, label="Device", value="disabled")
-                    photographer = gr.Dropdown(["disabled", "random"] + PHOTOGRAPHER, label="Photographer", value="disabled")
-                    artist = gr.Dropdown(["disabled", "random"] + ARTIST, label="Artist", value="disabled")
-                    digital_artform = gr.Dropdown(["disabled", "random"] + DIGITAL_ARTFORM, label="Digital Artform", 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")
                 
-                generate_button = gr.Button("Generate Prompt")
+                generate_button = gr.Button("🚀 프롬프트 생성", variant="primary", elem_classes="generate-button")
 
+            # 중간 패널 - 이미지와 출력
             with gr.Column(scale=2):
-                with gr.Accordion("Image and Caption", open=False):
-                    input_image = gr.Image(label="Input Image (optional)")
-                    caption_output = gr.Textbox(label="Generated Caption", lines=3)
-                    create_caption_button = gr.Button("Create Caption")
-                    add_caption_button = gr.Button("Add Caption to Prompt")
-
-                with gr.Accordion("Prompt Generation", open=True):
-                    output = gr.Textbox(label="Generated Prompt / Input Text", lines=4)
-                    t5xxl_output = gr.Textbox(label="T5XXL Output", visible=True)
-                    clip_l_output = gr.Textbox(label="CLIP L Output", visible=True)
-                    clip_g_output = gr.Textbox(label="CLIP G Output", visible=True)
+                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)
             
+            # 오른쪽 패널 - OpenAI
             with gr.Column(scale=2):
-                with gr.Accordion("Prompt Generation with OpenAI", open=True):
-                    # 모델 선택 제거 및 안내 텍스트 변경
-                    gr.HTML("<small>기본 모델: gpt-4.1-mini</small>")
-                    happy_talk = gr.Checkbox(label="Happy Talk", value=True)
-                    compress = gr.Checkbox(label="Compress", value=True)
-                    compression_level = gr.Radio(["soft", "medium", "hard"], label="Compression Level", value="hard")
-                    poster = gr.Checkbox(label="Poster", value=False)
-                    custom_base_prompt = gr.Textbox(label="Custom Base Prompt", lines=5)
-                generate_text_button = gr.Button("Generate Prompt with OpenAI")
-                text_output = gr.Textbox(label="Generated Text", lines=10)
+                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"
+                    )
 
+        # 이벤트 핸들러들
         def create_caption(image):
             if image is not None:
                 return florence_caption(image)
@@ -543,7 +615,6 @@ def create_interface():
             outputs=[output]
         )
 
-        # 함수 호출 시 파라미터 수정
         generate_text_button.click(
             lambda input_text, happy_talk, compress, compression_level, poster, custom_base_prompt: 
                 openai_node.generate(input_text, happy_talk, compress, compression_level, poster, custom_base_prompt),
@@ -551,28 +622,37 @@ def create_interface():
             outputs=text_output
         )
 
+        # 압축 체크박스에 따른 압축 강도 표시/숨기기
+        compress.change(
+            lambda x: gr.update(visible=x),
+            inputs=[compress],
+            outputs=[compression_level]
+        )
+
+        # 전역 옵션 변경 함수
         def update_all_options(choice):
             updates = {}
+            dropdown_list = [
+                artform, photo_type, body_types, default_tags, roles, hairstyles, clothing,
+                place, lighting, composition, pose, background, additional_details,
+                photography_styles, device, photographer, artist, digital_artform
+            ]
+            
             if choice == "Disabled":
-                for dropdown in [
-                    artform, photo_type, body_types, default_tags, roles, hairstyles, clothing,
-                    place, lighting, composition, pose, background, additional_details,
-                    photography_styles, device, photographer, artist, digital_artform
-                ]:
+                for dropdown in dropdown_list:
                     updates[dropdown] = gr.update(value="disabled")
             elif choice == "Random":
-                for dropdown in [
-                    artform, photo_type, body_types, default_tags, roles, hairstyles, clothing,
-                    place, lighting, composition, pose, background, additional_details,
-                    photography_styles, device, photographer, artist, digital_artform
-                ]:
+                for dropdown in dropdown_list:
                     updates[dropdown] = gr.update(value="random")
             else:  # No Figure Random
+                # 캐릭터 관련 설정은 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")
-            return updates
+            
+            return list(updates.values())
         
         global_option.change(
             update_all_options,