Update app.py

app.py

@@ -15,13 +15,13 @@ BRAVE_API_KEY = os.getenv('BRAVE_API_KEY', '')  # Add Brave API key
 
 def search_design_trends(product_type):
     """
-    Use Brave Search API to find current design trends and technologies
+    Brave Search API를 사용하여 최신 디자인 트렌드와 기술 검색
     """
     if not BRAVE_API_KEY:
-        return "No search results available (API key not set)"
+        return "검색 결과 없음 (API 키 미설정)"
     
     try:
-        # Construct search query for product design
+        # 제품 디자인을 위한 검색 쿼리 구성
         query = f"{product_type} design trends 2024 2025 materials technology innovation"
         
         url = "https://api.search.brave.com/res/v1/web/search"
@@ -47,84 +47,88 @@ def search_design_trends(product_type):
             if results:
                 return "\n".join(results)
             else:
-                return "Limited search results available"
+                return "제한된 검색 결과"
         else:
-            return "Search unavailable at the moment"
+            return "현재 검색 불가"
             
     except Exception as e:
-        print(f"Brave search error: {str(e)}")
-        return "Search results unavailable"
+        print(f"Brave 검색 오류: {str(e)}")
+        return "검색 결과 이용 불가"
 
 def enhance_prompt_with_team(user_input):
     """
-    Enhanced prompt generation using team collaboration with real web search
+    실제 웹 검색을 통한 팀 협업 방식의 향상된 프롬프트 생성
     """
     if not FIREWORKS_API_KEY:
-        # Fallback to basic enhancement
-        return create_basic_design_prompt(user_input)
+        # 기본 향상으로 폴백
+        return create_basic_design_prompt(user_input), "API 키가 설정되지 않음"
     
     try:
-        # First, get real search results for the engineer
+        # 먼저 엔지니어를 위한 실제 검색 결과 획득
         search_results = search_design_trends(user_input)
         
         url = "https://api.fireworks.ai/inference/v1/chat/completions"
         
-        # Enhanced system prompt with search integration
-        system_prompt = """You are a product design team with three specialized roles:
+        # 검색 통합된 향상된 시스템 프롬프트
+        system_prompt = """당신은 세 가지 전문 역할로 구성된 제품 디자인 팀입니다. 모든 응답은 한국어로 작성하세요:
 
-## TEAM ROLES:
+## 팀 역할:
 
-### 1. DIRECTOR (Creative Vision Leader)
-- Analyzes user requirements and market positioning
-- Sets design philosophy and constraints
-- Makes final decisions on features
+### 1. 감독 (크리에이티브 비전 리더)
+- 사용자 요구사항과 시장 포지셔닝 분석
+- 디자인 철학과 제약 조건 설정
+- 기능에 대한 최종 결정
 
-### 2. ENGINEER (Technical Expert with Research Data)
-- Uses provided search results to inform decisions
-- Evaluates manufacturing feasibility
-- Recommends materials and technologies
-- Suggests innovative features based on trends
+### 2. 엔지니어 (연구 데이터를 활용한 기술 전문가)
+- 제공된 검색 결과를 활용한 의사결정
+- 제조 가능성 평가
+- 재료와 기술 추천
+- 트렌드 기반 혁신 기능 제안
 
-### 3. DESIGNER (Visual Specialist)
-- Creates detailed visual specifications
-- Specifies exact materials, finishes, colors
-- Defines rendering style and atmosphere
-- MUST include text labels and annotations in the design
+### 3. 디자이너 (비주얼 전문가)
+- 상세한 시각적 사양 작성
+- 정확한 재료, 마감, 색상 지정
+- 렌더링 스타일과 분위기 정의
+- 텍스트 레이블과 주석 포함 필수
 
-## OUTPUT FORMAT:
+## 출력 형식:
 
-**DIRECTOR'S VISION:**
-[Brief vision statement - 2-3 sentences]
+**감독의 비전:**
+[간략한 비전 설명 - 2-3문장]
 
-**ENGINEER'S TECHNICAL ASSESSMENT:**
-[Based on search data, provide 3-4 technical recommendations]
+**엔지니어의 기술 평가:**
+[검색 데이터 기반 3-4개 기술 권장사항]
 
-**DESIGNER'S FINAL PROMPT:**
-[Detailed visual description INCLUDING text labels and annotations. MUST specify:
-- Product name/model as title text
-- Key feature callouts with arrow annotations
-- Technical specifications as overlay text
-- Material/component labels
-Example: "with text overlay showing 'MODEL X-500' at top, feature callouts pointing to 'Titanium Frame', 'OLED Display', '5000mAh Battery' with arrows"]
+**디자이너의 최종 프롬프트:**
+[텍스트 레이블과 주석을 포함한 상세한 시각적 설명. 반드시 포함:
+- 제품명/모델을 타이틀 텍스트로
+- 화살표 주석과 함께 주요 기능 설명
+- 오버레이 텍스트로 기술 사양
+- 재료/구성요소 레이블
+예시: "상단에 'MODEL X-500' 텍스트 오버레이, '티타늄 프레임', 'OLED 디스플레이', '5000mAh 배터리'를 가리키는 화살표와 기능 설명"]
 
-Keep responses concise but specific. Focus on PHYSICAL PRODUCTS only with TEXT ANNOTATIONS."""
+**최종 디자인 보고서:**
+[한국어로 작성된 제품의 핵심 특징, 혁신 포인트, 타겟 시장, 예상 제조 비용 등을 포함한 종합 보고서 - 3-4문장]
+
+간결하지만 구체적으로 작성. 물리적 제품에만 집중하며 텍스트 주석 포함."""
         
-        # Create collaborative prompt with search results
-        user_prompt = f"""Product concept: {user_input}
+        # 검색 결과와 함께 협업 프롬프트 생성
+        user_prompt = f"""제품 컨셉: {user_input}
 
-ENGINEER'S RESEARCH DATA:
+엔지니어의 연구 데이터:
 {search_results}
 
-Based on this input and research, execute the team collaboration:
-1. Director establishes vision (brief)
-2. Engineer uses the research data above to recommend features
-3. Designer creates a detailed visual prompt WITH TEXT LABELS AND ANNOTATIONS
+위 입력과 연구를 바탕으로 팀 협업 실행:
+1. 감독이 비전 수립 (간략히)
+2. 엔지니어가 위 연구 데이터를 활용해 기능 추천
+3. 디자이너가 텍스트 레이블과 주석이 포함된 상세한 시각적 프롬프트 작성
+4. 최종 디자인 보고서 작성 (한국어)
 
-Focus on creating a manufacturable, innovative product design with clear text overlays showing key features, specifications, and component labels."""
+제조 가능하고 혁신적인 제품 디자인 창출에 집중하며, 명확한 텍스트 오버레이로 주요 기능, 사양, 구성요소 레이블 표시."""
         
         payload = {
             "model": "accounts/fireworks/models/qwen3-235b-a22b-instruct-2507",  # Original model maintained
-            "max_tokens": 500,
+            "max_tokens": 600,
             "top_p": 0.9,
             "temperature": 0.7,
             "messages": [
@@ -151,51 +155,52 @@ Focus on creating a manufacturable, innovative product design with clear text ov
             result = response.json()
             enhanced = result.get('choices', [{}])[0].get('message', {}).get('content', '')
             if enhanced:
-                # Extract designer's final prompt
-                if "DESIGNER'S FINAL PROMPT:" in enhanced:
-                    parts = enhanced.split("DESIGNER'S FINAL PROMPT:")
+                # 디자이너의 최종 프롬프트 추출
+                if "디자이너의 최종 프롬프트:" in enhanced or "DESIGNER'S FINAL PROMPT:" in enhanced:
+                    parts = enhanced.split("디자이너의 최종 프롬프트:" if "디자이너의 최종 프롬프트:" in enhanced else "DESIGNER'S FINAL PROMPT:")
                     if len(parts) > 1:
-                        # Return both the final prompt and the full discussion
-                        return parts[1].strip(), enhanced
+                        # 최종 프롬프트와 전체 토론 모두 반환
+                        final_prompt = parts[1].split("**")[0].strip()
+                        return final_prompt, enhanced
                 return enhanced, enhanced
         
-        # Fallback if API fails
-        return create_basic_design_prompt(user_input), "Team collaboration unavailable - using basic prompt"
+        # API 실패 시 폴백
+        return create_basic_design_prompt(user_input), "팀 협업 불가 - 기본 프롬프트 사용"
         
     except requests.exceptions.Timeout:
-        print("Fireworks API timeout - using fallback")
-        return create_basic_design_prompt(user_input), "API timeout - using fallback prompt generation"
+        print("Fireworks API 타임아웃 - 폴백 사용")
+        return create_basic_design_prompt(user_input), "API 타임아웃 - 폴백 프롬프트 생성 사용"
     except Exception as e:
-        print(f"Team collaboration error: {str(e)}")
-        return create_basic_design_prompt(user_input), f"Error: {str(e)[:100]}"
+        print(f"팀 협업 오류: {str(e)}")
+        return create_basic_design_prompt(user_input), f"오류: {str(e)[:100]}"
 
 def create_basic_design_prompt(user_input):
     """
-    Fallback function for basic design prompt generation with text annotations
+    텍스트 주석이 포함된 기본 디자인 프롬프트 생성을 위한 폴백 함수
     """
     product_name = user_input.upper().replace(" ", "-")
-    return f"""A professional product design concept for {user_input} with detailed text annotations:
-- Product title text overlay: "{product_name} PRO 2025" in bold modern font at top
-- Feature callout labels with arrows pointing to key components:
-  * "Premium Materials" → body
-  * "Ergonomic Design" → form factor  
-  * "Smart Integration" → control panel
-  * "Sustainable Manufacturing" → eco-friendly elements
-- Technical specifications text overlay in corner:
-  * Dimensions, weight, power specs
-- Clean industrial design with annotations
-- Photorealistic rendering with studio lighting
-- Professional product photography style with infographic elements
-- Text labels in clean sans-serif font (Helvetica/Arial style)
-- High contrast text for readability
-- Shown from 3/4 perspective with all labels clearly visible"""
+    return f"""상세한 텍스트 주석이 포함된 {user_input}의 전문 제품 디자인 컨셉:
+- 상단에 굵은 모던 폰트로 제품 타이틀 텍스트 오버레이: "{product_name} PRO 2025"
+- 주요 구성요소를 가리키는 화살표와 함께 기능 설명 레이블:
+  * "프리미엄 소재" → 본체
+  * "인체공학적 디자인" → 폼 팩터  
+  * "스마트 통합" → 제어 패널
+  * "지속가능한 제조" → 친환경 요소
+- 모서리에 기술 사양 텍스트 오버레이:
+  * 치수, 무게, 전력 사양
+- 주석이 포함된 깔끔한 산업 디자인
+- 스튜디오 조명과 함께 포토리얼리스틱 렌더링
+- 인포그래픽 요소가 포함된 전문 제품 사진 스타일
+- 깔끔한 산세리프 폰트(Helvetica/Arial 스타일)의 텍스트 레이블
+- 가독성을 위한 높은 대비 텍스트
+- 모든 레이블이 명확히 보이는 3/4 관점에서 표시"""
 
 def upload_image_to_hosting(image):
     """
-    Upload image to hosting service
+    호스팅 서비스에 이미지 업로드
     """
     try:
-        # Try imgbb first
+        # imgbb 먼저 시도
         buffered = BytesIO()
         image.save(buffered, format="PNG")
         buffered.seek(0)
@@ -217,7 +222,7 @@ def upload_image_to_hosting(image):
     except:
         pass
     
-    # Fallback to base64
+    # base64로 폴백
     buffered = BytesIO()
     image.save(buffered, format="PNG")
     buffered.seek(0)
@@ -226,34 +231,34 @@ def upload_image_to_hosting(image):
 
 def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
     """
-    Process product design with team collaboration
+    팀 협업을 통한 제품 디자인 처리
     """
     if not os.getenv('REPLICATE_API_TOKEN'):
-        return None, prompt, "⚠️ Please set REPLICATE_API_TOKEN", ""
+        return None, prompt, "⚠️ REPLICATE_API_TOKEN을 설정하세요", ""
     
     try:
         team_discussion = ""
         final_prompt = prompt
         
-        # Apply team collaboration if enabled
+        # 활성화된 경우 팀 협업 적용
         if enhance_prompt_flag:
             result = enhance_prompt_with_team(prompt)
             if isinstance(result, tuple):
                 final_prompt, team_discussion = result
             else:
                 final_prompt = result
-                team_discussion = "Team collaboration completed"
+                team_discussion = "팀 협업 완료"
             
-            # Format team discussion for display
-            if "DIRECTOR'S VISION:" in team_discussion:
-                team_discussion = f"### 🎬 Team Collaboration Process\n\n{team_discussion}"
+            # 표시를 위한 팀 토론 형식화
+            if "감독의 비전:" in team_discussion or "DIRECTOR'S VISION:" in team_discussion:
+                team_discussion = f"### 🎬 팀 협업 프로세스\n\n{team_discussion}"
         
-        # Prepare model input
+        # 모델 입력 준비
         model_input = {
             "prompt": final_prompt
         }
         
-        # Add reference images if provided
+        # 제공된 경우 참조 이미지 추가
         if image1 or image2:
             image_urls = []
             
@@ -266,13 +271,13 @@ def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
                 image_urls.append(url2)
             
             model_input["image_input"] = image_urls
-            status_msg = "✅ Product design generated with style references!"
+            status_msg = "✅ 스타일 참조와 함께 제품 디자인 생성 완료!"
         else:
-            status_msg = "✅ Product design generated successfully!"
+            status_msg = "✅ 제품 디자인이 성공적으로 생성되었습니다!"
         
-        # Generate image with Replicate
+        # Replicate로 이미지 생성
         try:
-            # Add text annotation emphasis to the prompt
+            # 프롬프트에 텍스트 주석 강조 추가
             annotated_prompt = f"{final_prompt}. Include clear text labels, feature callouts with arrows, product name/model as title, technical specifications overlay, component annotations in clean modern typography"
             
             output = replicate.run(
@@ -286,7 +291,7 @@ def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
                 }
             )
         except:
-            # Fallback to original model with text annotations
+            # 텍스트 주석과 함께 원본 모델로 폴백
             annotated_prompt = f"{final_prompt}. With text overlays showing product name, key features with arrow callouts, specifications, and component labels in professional typography"
             
             output = replicate.run(
@@ -301,9 +306,9 @@ def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
             )
         
         if output is None:
-            return None, final_prompt, "❌ No output received from model", team_discussion
+            return None, final_prompt, "❌ 모델로부터 출력을 받지 못했습니다", team_discussion
         
-        # Process output
+        # 출력 처리
         output_url = None
         if isinstance(output, str):
             output_url = output
@@ -318,17 +323,17 @@ def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
                 img = Image.open(BytesIO(response.content))
                 return img, final_prompt, status_msg, team_discussion
         
-        return None, final_prompt, "❌ Could not process output", team_discussion
+        return None, final_prompt, "❌ 출력을 처리할 수 없습니다", team_discussion
         
     except Exception as e:
         error_msg = str(e)
-        return None, prompt, f"❌ Error: {error_msg[:200]}", ""
+        return None, prompt, f"❌ 오류: {error_msg[:200]}", ""
 
-# Professional Product Design CSS
+# 전문 제품 디자인 CSS
 css = """
 .gradio-container {
     background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
-    font-family: 'Inter', -apple-system, BlinkMacSystemFont, 'Segoe UI', sans-serif;
+    font-family: 'Pretendard', 'Noto Sans KR', -apple-system, BlinkMacSystemFont, sans-serif;
     min-height: 100vh;
 }
 .header-container {
@@ -365,7 +370,7 @@ css = """
     text-align: center;
     font-weight: 500;
     color: #ffd700;
-    max-width: 500px;
+    max-width: 600px;
     font-size: 0.9rem;
 }
 .main-content {
@@ -409,11 +414,12 @@ css = """
     padding: 1.2rem;
     margin-top: 1rem;
     border: 1px solid #dee2e6;
-    font-family: 'Courier New', monospace;
+    font-family: 'Pretendard', 'Noto Sans KR', monospace;
     font-size: 0.85rem;
-    max-height: 300px;
+    max-height: 400px;
     overflow-y: auto;
     white-space: pre-wrap;
+    line-height: 1.6;
 }
 .enhanced-prompt-box {
     background: #f0f0f0;
@@ -444,83 +450,83 @@ footer {
 }
 """
 
-# Create Gradio interface
+# Gradio 인터페이스 생성
 with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
     with gr.Column(elem_classes="header-container"):
         gr.HTML("""
-            <h1 class="logo-text">🏭 PRODUCT DESIGN STUDIO</h1>
-            <p class="subtitle">AI-Powered Industrial Design with Annotated Feature Guides</p>
+            <h1 class="logo-text">🏭 제품 디자인 스튜디오</h1>
+            <p class="subtitle">AI 기반 산업 디자인 시스템 - 웹 리서치와 주석 기능 포함</p>
             <div class="mode-indicator">
-                👥 Director → 🔍 Engineer (Research) → 🎨 Designer (with Text Annotations)
+                👥 감독 → 🔍 엔지니어 (Brave 검색) → 🎨 디자이너 (텍스트 주석 포함)
             </div>
         """)
 
     with gr.Column(elem_classes="main-content"):
         gr.HTML("""
             <div class="info-box">
-                <strong>🎯 Professional Product Design Process with Annotations:</strong><br>
-                • <b>Director:</b> Establishes vision and requirements<br>
-                • <b>Engineer:</b> Searches current trends & technologies via Brave API<br>
-                • <b>Designer:</b> Creates visual specs with <b>text labels & feature callouts</b><br>
-                • <b>Output:</b> Annotated product designs with key features highlighted<br>
-                • <b>Focus:</b> Physical products with infographic-style annotations
+                <strong>🎯 전문 제품 디자인 프로세스 (주석 포함):</strong><br>
+                • <b>감독:</b> 비전과 요구사항 수립<br>
+                • <b>엔지니어:</b> Brave API로 최신 트렌드와 기술 검색<br>
+                • <b>디자이너:</b> <b>텍스트 레이블과 기능 설명</b>이 포함된 시각 사양 작성<br>
+                • <b>결과물:</b> 주요 기능이 강조된 주석 포함 제품 디자인<br>
+                • <b>집중 분야:</b> 인포그래픽 스타일 주석이 포함된 물리적 제품
             </div>
         """)
         
         with gr.Row(equal_height=True):
             with gr.Column(scale=1):
                 prompt = gr.Textbox(
-                    label="📝 Product Concept",
-                    placeholder="Enter product type: 'ergonomic keyboard', 'smart watch', 'office chair', 'coffee maker'...",
+                    label="📝 제품 컨셉",
+                    placeholder="제품 유형 입력: '스마트 운동화', '미래형 주방용품', '친환경 가방', '모듈형 가구'...",
                     lines=2,
-                    value="wireless earbuds",
+                    value="차세대 무선 이어폰",
                     elem_classes="prompt-input"
                 )
                 
                 enhance_prompt_checkbox = gr.Checkbox(
-                    label="🚀 Enable Team Collaboration with Web Research",
+                    label="🚀 웹 리서치를 통한 팀 협업 활성화",
                     value=True,
-                    info="Activates Director + Engineer (Brave Search) + Designer process"
+                    info="감독 + 엔지니어(Brave 검색) + 디자이너 프로세스 활성화"
                 )
                 
                 with gr.Row():
                     image1 = gr.Image(
-                        label="Style Reference 1 (Optional)",
+                        label="스타일 참조 1 (선택사항)",
                         type="pil",
                         height=180
                     )
                     image2 = gr.Image(
-                        label="Style Reference 2 (Optional)",
+                        label="스타일 참조 2 (선택사항)",
                         type="pil",
                         height=180
                     )
                 
                 generate_btn = gr.Button(
-                    "🎨 Generate Product Design",
+                    "🎨 제품 디자인 생성",
                     variant="primary",
                     size="lg"
                 )
             
             with gr.Column(scale=1):
                 output_image = gr.Image(
-                    label="Generated Product Design",
+                    label="생성된 제품 디자인",
                     type="pil",
                     height=400,
                     elem_classes="image-container"
                 )
                 
                 enhanced_prompt_display = gr.Textbox(
-                    label="📋 Final Design Specifications",
+                    label="📋 최종 디자인 사양",
                     interactive=False,
                     lines=3,
                     elem_classes="enhanced-prompt-box"
                 )
                 
                 status = gr.Textbox(
-                    label="Status",
+                    label="상태",
                     interactive=False,
                     elem_classes="status-text",
-                    value="Ready to design..."
+                    value="디자인 준비 중..."
                 )
         
         with gr.Row():
@@ -529,42 +535,46 @@ with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
                 elem_classes="team-discussion"
             )
     
-    # Connect event handler
+    # 이벤트 핸들러 연결
     generate_btn.click(
         fn=process_product_design,
         inputs=[prompt, enhance_prompt_checkbox, image1, image2],
         outputs=[output_image, enhanced_prompt_display, status, team_discussion_display]
     )
     
-    # Product examples
+    # 창의적인 한글 제품 예시
     gr.Examples(
         examples=[
-            ["wireless earbuds", True, None, None],
-            ["ergonomic office chair", True, None, None],
-            ["smart home speaker", True, None, None],
-            ["electric toothbrush", True, None, None],
-            ["portable coffee maker", True, None, None],
-            ["gaming keyboard", True, None, None],
-            ["fitness tracker", True, None, None],
-            ["desk lamp", True, None, None],
+            ["공중 부양 스피커", True, None, None],
+            ["변형 가능한 다목적 가구", True, None, None],
+            ["자가 충전 스마트 신발", True, None, None],
+            ["홀로그램 디스플레이 시계", True, None, None],
+            ["접이식 전기 스쿠터 헬멧", True, None, None],
+            ["AI 요리 도우미 로봇", True, None, None],
+            ["투명 OLED 노트북", True, None, None],
+            ["생체인식 스마트 도어락", True, None, None],
+            ["모듈형 캠핑 장비 시스템", True, None, None],
+            ["웨어러블 공기청정기", True, None, None],
+            ["자동 온도조절 텀블러", True, None, None],
+            ["미래형 반려동물 케어 스테이션", True, None, None],
         ],
         inputs=[prompt, enhance_prompt_checkbox, image1, image2],
-        label="💡 Product Design Examples"
+        label="💡 창의적인 제품 디자인 예시"
     )
 
-# Launch settings
+# 실행 설정
 if __name__ == "__main__":
-    # Check for required API keys
+    # 필요한 API 키 확인
     if not os.getenv('REPLICATE_API_TOKEN'):
-        print("⚠️ Warning: REPLICATE_API_TOKEN not set")
+        print("⚠️ 경고: REPLICATE_API_TOKEN이 설정되지 않음")
     if not os.getenv('FIREWORKS_API_KEY'):
-        print("⚠️ Warning: FIREWORKS_API_KEY not set - will use fallback prompts")
+        print("⚠️ 경고: FIREWORKS_API_KEY가 설정되지 않음 - 폴백 프롬프트 사용")
     if not os.getenv('BRAVE_API_KEY'):
-        print("⚠️ Warning: BRAVE_API_KEY not set - web search unavailable")
+        print("⚠️ 경고: BRAVE_API_KEY가 설정되지 않음 - 웹 검색 불가")
     
     demo.launch(
-        share=False,  # Set to False for Hugging Face Spaces
+        share=False,  # Hugging Face Spaces용으로 False 설정
         server_name="0.0.0.0",
         server_port=7860,
-        ssr_mode=False  # Disable SSR to avoid warning
+        ssr_mode=False  # 경고 방지를 위해 SSR 비활성화
     )