Update app.py

app.py

@@ -1,580 +1,35 @@
-import gradio as gr
-import replicate
 import os
-from PIL import Image
-import requests
-from io import BytesIO
-import time
-import base64
-import json
+import sys
+import streamlit as st
+from tempfile import NamedTemporaryFile
 
-# Set up API keys from environment variables
-os.environ['REPLICATE_API_TOKEN'] = os.getenv('REPLICATE_API_TOKEN')
-FIREWORKS_API_KEY = os.getenv('FIREWORKS_API_KEY', '')
-BRAVE_API_KEY = os.getenv('BRAVE_API_KEY', '')  # Add Brave API key
-
-def search_design_trends(product_type):
-    """
-    Brave Search API를 사용하여 최신 디자인 트렌드와 기술 검색
-    """
-    if not BRAVE_API_KEY:
-        return "검색 결과 없음 (API 키 미설정)"
-    
-    try:
-        # 제품 디자인을 위한 검색 쿼리 구성
-        query = f"{product_type} design trends 2024 2025 materials technology innovation"
-        
-        url = "https://api.search.brave.com/res/v1/web/search"
-        headers = {
-            "Accept": "application/json",
-            "X-Subscription-Token": BRAVE_API_KEY
-        }
-        params = {
-            "q": query,
-            "count": 5
-        }
-        
-        response = requests.get(url, headers=headers, params=params, timeout=10)
-        
-        if response.status_code == 200:
-            data = response.json()
-            results = []
-            
-            if 'web' in data and 'results' in data['web']:
-                for item in data['web']['results'][:3]:
-                    results.append(f"- {item.get('title', '')}: {item.get('description', '')[:150]}")
-            
-            if results:
-                return "\n".join(results)
-            else:
-                return "제한된 검색 결과"
-        else:
-            return "현재 검색 불가"
-            
-    except Exception as e:
-        print(f"Brave 검색 오류: {str(e)}")
-        return "검색 결과 이용 불가"
-
-def enhance_prompt_with_team(user_input):
-    """
-    실제 웹 검색을 통한 팀 협업 방식의 향상된 프롬프트 생성
-    """
-    if not FIREWORKS_API_KEY:
-        # 기본 향상으로 폴백
-        return create_basic_design_prompt(user_input), "API 키가 설정되지 않음"
-    
-    try:
-        # 먼저 엔지니어를 위한 실제 검색 결과 획득
-        search_results = search_design_trends(user_input)
-        
-        url = "https://api.fireworks.ai/inference/v1/chat/completions"
-        
-        # 검색 통합된 향상된 시스템 프롬프트
-        system_prompt = """당신은 세 가지 전문 역할로 구성된 제품 디자인 팀입니다. 모든 응답은 한국어로 작성하세요:
-
-## 팀 역할:
-
-### 1. 감독 (크리에이티브 비전 리더)
-- 사용자 요구사항과 시장 포지셔닝 분석
-- 디자인 철학과 제약 조건 설정
-- 기능에 대한 최종 결정
-
-### 2. 엔지니어 (연구 데이터를 활용한 기술 전문가)
-- 제공된 검색 결과를 활용한 의사결정
-- 제조 가능성 평가
-- 재료와 기술 추천
-- 트렌드 기반 혁신 기능 제안
-
-### 3. 디자이너 (비주얼 전문가)
-- 상세한 시각적 사양 작성
-- 정확한 재료, 마감, 색상 지정
-- 렌더링 스타일과 분위기 정의
-- 텍스트 레이블과 주석 포함 필수
-
-## 출력 형식:
-
-**감독의 비전:**
-[간략한 비전 설명 - 2-3문장]
-
-**엔지니어의 기술 평가:**
-[검색 데이터 기반 3-4개 기술 권장사항]
-
-**디자이너의 최종 프롬프트:**
-[텍스트 레이블과 주석을 포함한 상세한 시각적 설명. 반드시 포함:
-- 제품명/모델을 타이틀 텍스트로
-- 화살표 주석과 함께 주요 기능 설명
-- 오버레이 텍스트로 기술 사양
-- 재료/구성요소 레이블
-예시: "상단에 'MODEL X-500' 텍스트 오버레이, '티타늄 프레임', 'OLED 디스플레이', '5000mAh 배터리'를 가리키는 화살표와 기능 설명"]
-
-**최종 디자인 보고서:**
-[한국어로 작성된 제품의 핵심 특징, 혁신 포인트, 타겟 시장, 예상 제조 비용 등을 포함한 종합 보고서 - 3-4문장]
-
-간결하지만 구체적으로 작성. 물리적 제품에만 집중하며 텍스트 주석 포함."""
-        
-        # 검색 결과와 함께 협업 프롬프트 생성
-        user_prompt = f"""제품 컨셉: {user_input}
-
-엔지니어의 연구 데이터:
-{search_results}
-
-위 입력과 연구를 바탕으로 팀 협업 실행:
-1. 감독이 비전 수립 (간략히)
-2. 엔지니어가 위 연구 데이터를 활용해 기능 추천
-3. 디자이너가 텍스트 레이블과 주석이 포함된 상세한 시각적 프롬프트 작성
-4. 최종 디자인 보고서 작성 (한국어)
-
-제조 가능하고 혁신적인 제품 디자인 창출에 집중하며, 명확한 텍스트 오버레이로 주요 기능, 사양, 구성요소 레이블 표시."""
-        
-        payload = {
-            "model": "accounts/fireworks/models/qwen3-235b-a22b-instruct-2507",  # Original model maintained
-            "max_tokens": 600,
-            "top_p": 0.9,
-            "temperature": 0.7,
-            "messages": [
-                {
-                    "role": "system",
-                    "content": system_prompt
-                },
-                {
-                    "role": "user",
-                    "content": user_prompt
-                }
-            ]
-        }
-        
-        headers = {
-            "Accept": "application/json",
-            "Content-Type": "application/json",
-            "Authorization": f"Bearer {FIREWORKS_API_KEY}"
-        }
-        
-        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=30)
-        
-        if response.status_code == 200:
-            result = response.json()
-            enhanced = result.get('choices', [{}])[0].get('message', {}).get('content', '')
-            if enhanced:
-                # 디자이너의 최종 프롬프트 추출
-                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:
-                        # 최종 프롬프트와 전체 토론 모두 반환
-                        final_prompt = parts[1].split("**")[0].strip()
-                        return final_prompt, enhanced
-                return enhanced, enhanced
-        
-        # API 실패 시 폴백
-        return create_basic_design_prompt(user_input), "팀 협업 불가 - 기본 프롬프트 사용"
-        
-    except requests.exceptions.Timeout:
-        print("Fireworks API 타임아웃 - 폴백 사용")
-        return create_basic_design_prompt(user_input), "API 타임아웃 - 폴백 프롬프트 생성 사용"
-    except Exception as e:
-        print(f"팀 협업 오류: {str(e)}")
-        return create_basic_design_prompt(user_input), f"오류: {str(e)[:100]}"
-
-def create_basic_design_prompt(user_input):
-    """
-    텍스트 주석이 포함된 기본 디자인 프롬프트 생성을 위한 폴백 함수
-    """
-    product_name = user_input.upper().replace(" ", "-")
-    return f"""상세한 텍스트 주석이 포함된 {user_input}의 전문 제품 디자인 컨셉:
-- 상단에 굵은 모던 폰트로 제품 타이틀 텍스트 오버레이: "{product_name} PRO 2025"
-- 주요 구성요소를 가리키는 화살표와 함께 기능 설명 레이블:
-  * "프리미엄 소재" → 본체
-  * "인체공학적 디자인" → 폼 팩터  
-  * "스마트 통합" → 제어 패널
-  * "지속가능한 제조" → 친환경 요소
-- 모서리에 기술 사양 텍스트 오버레이:
-  * 치수, 무게, 전력 사양
-- 주석이 포함된 깔끔한 산업 디자인
-- 스튜디오 조명과 함께 포토리얼리스틱 렌더링
-- 인포그래픽 요소가 포함된 전문 제품 사진 스타일
-- 깔끔한 산세리프 폰트(Helvetica/Arial 스타일)의 텍스트 레이블
-- 가독성을 위한 높은 대비 텍스트
-- 모든 레이블이 명확히 보이는 3/4 관점에서 표시"""
-
-def upload_image_to_hosting(image):
-    """
-    호스팅 서비스에 이미지 업로드
-    """
+def main():
     try:
-        # imgbb 먼저 시도
-        buffered = BytesIO()
-        image.save(buffered, format="PNG")
-        buffered.seek(0)
-        img_base64 = base64.b64encode(buffered.getvalue()).decode()
+        # Get the code from secrets
+        code = os.environ.get("MAIN_CODE")
         
-        response = requests.post(
-            "https://api.imgbb.com/1/upload",
-            data={
-                'key': '6d207e02198a847aa98d0a2a901485a5',
-                'image': img_base64,
-            },
-            timeout=10
-        )
+        if not code:
+            st.error("⚠️ The application code wasn't found in secrets. Please add the MAIN_CODE secret.")
+            return
         
-        if response.status_code == 200:
-            data = response.json()
-            if data.get('success'):
-                return data['data']['url']
-    except:
-        pass
-    
-    # base64로 폴백
-    buffered = BytesIO()
-    image.save(buffered, format="PNG")
-    buffered.seek(0)
-    img_base64 = base64.b64encode(buffered.getvalue()).decode()
-    return f"data:image/png;base64,{img_base64}"
-
-def process_product_design(prompt, enhance_prompt_flag, image1, image2=None):
-    """
-    팀 협업을 통한 제품 디자인 처리
-    """
-    if not os.getenv('REPLICATE_API_TOKEN'):
-        return None, prompt, "⚠️ REPLICATE_API_TOKEN을 설정하세요", ""
-    
-    try:
-        team_discussion = ""
-        final_prompt = prompt
+        # Create a temporary Python file
+        with NamedTemporaryFile(suffix='.py', delete=False, mode='w') as tmp:
+            tmp.write(code)
+            tmp_path = tmp.name
         
-        # 활성화된 경우 팀 협업 적용
-        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 = "팀 협업 완료"
-            
-            # 표시를 위한 팀 토론 형식화
-            if "감독의 비전:" in team_discussion or "DIRECTOR'S VISION:" in team_discussion:
-                team_discussion = f"### 🎬 팀 협업 프로세스\n\n{team_discussion}"
-        
-        # 모델 입력 준비
-        model_input = {
-            "prompt": final_prompt
-        }
-        
-        # 제공된 경우 참조 이미지 추가
-        if image1 or image2:
-            image_urls = []
-            
-            if image1:
-                url1 = upload_image_to_hosting(image1)
-                image_urls.append(url1)
-            
-            if image2:
-                url2 = upload_image_to_hosting(image2)
-                image_urls.append(url2)
-            
-            model_input["image_input"] = image_urls
-            status_msg = "✅ 스타일 참조와 함께 제품 디자인 생성 완료!"
-        else:
-            status_msg = "✅ 제품 디자인이 성공적으로 생성되었습니다!"
+        # Execute the code
+        exec(compile(code, tmp_path, 'exec'), globals())
         
-        # Replicate로 이미지 생성
+        # Clean up the temporary file
         try:
-            # 프롬프트에 텍스트 주석 강조 추가
-            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(
-                "black-forest-labs/flux-schnell",  # Alternative model that works better
-                input={
-                    "prompt": annotated_prompt,
-                    "num_outputs": 1,
-                    "aspect_ratio": "1:1",
-                    "output_format": "webp",
-                    "output_quality": 90
-                }
-            )
+            os.unlink(tmp_path)
         except:
-            # 텍스트 주석과 함께 원본 모델로 폴백
-            annotated_prompt = f"{final_prompt}. With text overlays showing product name, key features with arrow callouts, specifications, and component labels in professional typography"
+            pass
             
-            output = replicate.run(
-                "stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b",
-                input={
-                    "prompt": annotated_prompt,
-                    "negative_prompt": "low quality, blurry, distorted, unreadable text, messy labels",
-                    "width": 1024,
-                    "height": 1024,
-                    "num_outputs": 1
-                }
-            )
-        
-        if output is None:
-            return None, final_prompt, "❌ 모델로부터 출력을 받지 못했습니다", team_discussion
-        
-        # 출력 처리
-        output_url = None
-        if isinstance(output, str):
-            output_url = output
-        elif isinstance(output, list) and len(output) > 0:
-            output_url = output[0]
-        elif hasattr(output, 'url'):
-            output_url = output.url
-        
-        if output_url:
-            response = requests.get(output_url, timeout=30)
-            if response.status_code == 200:
-                img = Image.open(BytesIO(response.content))
-                return img, final_prompt, status_msg, team_discussion
-        
-        return None, final_prompt, "❌ 출력을 처리할 수 없습니다", team_discussion
-        
     except Exception as e:
-        error_msg = str(e)
-        return None, prompt, f"❌ 오류: {error_msg[:200]}", ""
-
-# 전문 제품 디자인 CSS
-css = """
-.gradio-container {
-    background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
-    font-family: 'Pretendard', 'Noto Sans KR', -apple-system, BlinkMacSystemFont, sans-serif;
-    min-height: 100vh;
-}
-.header-container {
-    background: linear-gradient(135deg, #2d2d2d 0%, #0f0f0f 100%);
-    padding: 2rem;
-    border-radius: 20px;
-    margin-bottom: 2rem;
-    box-shadow: 0 10px 40px rgba(0, 0, 0, 0.5);
-    border: 1px solid rgba(255, 215, 0, 0.2);
-}
-.logo-text {
-    font-size: 2.8rem;
-    font-weight: 800;
-    color: #ffd700;
-    text-align: center;
-    margin: 0;
-    letter-spacing: -1px;
-    text-shadow: 2px 2px 4px rgba(0,0,0,0.8);
-}
-.subtitle {
-    color: #b0b0b0;
-    text-align: center;
-    font-size: 1rem;
-    margin-top: 0.5rem;
-    font-weight: 400;
-    letter-spacing: 0.5px;
-}
-.mode-indicator {
-    background: rgba(255, 215, 0, 0.1);
-    border: 1px solid rgba(255, 215, 0, 0.3);
-    border-radius: 10px;
-    padding: 0.6rem 1.2rem;
-    margin: 1rem auto;
-    text-align: center;
-    font-weight: 500;
-    color: #ffd700;
-    max-width: 600px;
-    font-size: 0.9rem;
-}
-.main-content {
-    background: rgba(255, 255, 255, 0.98);
-    border-radius: 20px;
-    padding: 2rem;
-    box-shadow: 0 5px 20px rgba(0, 0, 0, 0.3);
-}
-.gr-button-primary {
-    background: linear-gradient(135deg, #2d2d2d 0%, #0f0f0f 100%) !important;
-    border: 1px solid #ffd700 !important;
-    color: #ffd700 !important;
-    font-weight: 600 !important;
-    font-size: 1rem !important;
-    padding: 1rem 2rem !important;
-    border-radius: 10px !important;
-    transition: all 0.3s ease !important;
-    text-transform: uppercase;
-    letter-spacing: 0.5px;
-    width: 100%;
-    margin-top: 1rem !important;
-}
-.gr-button-primary:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 8px 25px rgba(255, 215, 0, 0.25) !important;
-    background: linear-gradient(135deg, #ffd700 0%, #ffed4e 100%) !important;
-    color: #000 !important;
-}
-.info-box {
-    background: linear-gradient(135deg, #fff3cd 0%, #ffeaa7 100%);
-    border-radius: 10px;
-    padding: 1rem;
-    margin-bottom: 1.5rem;
-    border-left: 4px solid #ffd700;
-    color: #333;
-    font-size: 0.9rem;
-}
-.team-discussion {
-    background: #f8f9fa;
-    border-radius: 10px;
-    padding: 1.2rem;
-    margin-top: 1rem;
-    border: 1px solid #dee2e6;
-    font-family: 'Pretendard', 'Noto Sans KR', monospace;
-    font-size: 0.85rem;
-    max-height: 400px;
-    overflow-y: auto;
-    white-space: pre-wrap;
-    line-height: 1.6;
-}
-.enhanced-prompt-box {
-    background: #f0f0f0;
-    border-radius: 10px;
-    padding: 1rem;
-    margin-top: 0.5rem;
-    border-left: 3px solid #ffd700;
-    font-size: 0.9rem;
-}
-.gr-input, .gr-textarea {
-    background: #ffffff !important;
-    border: 1px solid #d0d0d0 !important;
-    border-radius: 8px !important;
-    font-size: 0.95rem !important;
-    padding: 0.7rem !important;
-}
-.gr-input:focus, .gr-textarea:focus {
-    border-color: #ffd700 !important;
-    box-shadow: 0 0 0 3px rgba(255, 215, 0, 0.1) !important;
-}
-.image-container {
-    border-radius: 10px !important;
-    overflow: hidden;
-    border: 1px solid #e0e0e0 !important;
-}
-footer {
-    display: none !important;
-}
-"""
-
-# 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">🏭 제품 디자인 스튜디오</h1>
-            <p class="subtitle">AI 기반 산업 디자인 시스템 - 웹 리서치와 주석 기능 포함</p>
-            <div class="mode-indicator">
-                👥 감독 → 🔍 엔지니어 (Brave 검색) → 🎨 디자이너 (텍스트 주석 포함)
-            </div>
-        """)
-
-    with gr.Column(elem_classes="main-content"):
-        gr.HTML("""
-            <div class="info-box">
-                <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="📝 제품 컨셉",
-                    placeholder="제품 유형 입력: '스마트 운동화', '미래형 주방용품', '친환경 가방', '모듈형 가구'...",
-                    lines=2,
-                    value="차세대 무선 이어폰",
-                    elem_classes="prompt-input"
-                )
-                
-                enhance_prompt_checkbox = gr.Checkbox(
-                    label="🚀 웹 리서치를 통한 팀 협업 활성화",
-                    value=True,
-                    info="감독 + 엔지니어(Brave 검색) + 디자이너 프로세스 활성화"
-                )
-                
-                with gr.Row():
-                    image1 = gr.Image(
-                        label="스타일 참조 1 (선택사항)",
-                        type="pil",
-                        height=180
-                    )
-                    image2 = gr.Image(
-                        label="스타일 참조 2 (선택사항)",
-                        type="pil",
-                        height=180
-                    )
-                
-                generate_btn = gr.Button(
-                    "🎨 제품 디자인 생성",
-                    variant="primary",
-                    size="lg"
-                )
-            
-            with gr.Column(scale=1):
-                output_image = gr.Image(
-                    label="생성된 제품 디자인",
-                    type="pil",
-                    height=400,
-                    elem_classes="image-container"
-                )
-                
-                enhanced_prompt_display = gr.Textbox(
-                    label="📋 최종 디자인 사양",
-                    interactive=False,
-                    lines=3,
-                    elem_classes="enhanced-prompt-box"
-                )
-                
-                status = gr.Textbox(
-                    label="상태",
-                    interactive=False,
-                    elem_classes="status-text",
-                    value="디자인 준비 중..."
-                )
-        
-        with gr.Row():
-            team_discussion_display = gr.Markdown(
-                value="",
-                elem_classes="team-discussion"
-            )
-    
-    # 이벤트 핸들러 연결
-    generate_btn.click(
-        fn=process_product_design,
-        inputs=[prompt, enhance_prompt_checkbox, image1, image2],
-        outputs=[output_image, enhanced_prompt_display, status, team_discussion_display]
-    )
-    
-    # 창의적인 한글 제품 예시
-    gr.Examples(
-        examples=[
-            ["공중 부양 스피커", 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="💡 창의적인 제품 디자인 예시"
-    )
+        st.error(f"⚠️ Error loading or executing the application: {str(e)}")
+        import traceback
+        st.code(traceback.format_exc())
 
-# 실행 설정
 if __name__ == "__main__":
-    # 필요한 API 키 확인
-    if not os.getenv('REPLICATE_API_TOKEN'):
-        print("⚠️ 경고: REPLICATE_API_TOKEN이 설정되지 않음")
-    if not os.getenv('FIREWORKS_API_KEY'):
-        print("⚠️ 경고: FIREWORKS_API_KEY가 설정되지 않음 - 폴백 프롬프트 사용")
-    if not os.getenv('BRAVE_API_KEY'):
-        print("⚠️ 경고: BRAVE_API_KEY가 설정되지 않음 - 웹 검색 불가")
-    
-    demo.launch(
-        share=False,  # Hugging Face Spaces용으로 False 설정
-        server_name="0.0.0.0",
-        server_port=7860,
-        ssr_mode=False  # 경고 방지를 위해 SSR 비활성화
-    )
+    main()