Update app.py

app.py

@@ -42,7 +42,7 @@ SQUARE_THEORY_PROMPT = """
 당신은 Square Theory 전문가입니다. Square Theory는 4개의 단어가 의미적 관계로 연결되어 사각형을 이루는 구조입니다.
 브랜드명이 Square를 완성하며 "아하!" 모먼트를 만들어야 합니다.
 
-사용자 입력(업종/키워드)을 받아 다음 형식의 JSON 배열을 생성하세요:
+사용자 입력(업종/키워드)을 받아 다음 JSON 형식의 배열을 생성하세요:
 {
   "brand_name": "브랜드명",
   "tl": "왼쪽상단", "tr": "오른쪽상단", "bl": "왼쪽하단", "br": "오른쪽하단",
@@ -51,13 +51,15 @@ SQUARE_THEORY_PROMPT = """
   "slogan": "슬로건",
   "explanation": "왜 효과적인지"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 2. CONCEPTUAL BLENDING =====
 CONCEPTUAL_BLENDING_PROMPT = """
 당신은 Conceptual Blending Theory 전문가입니다. 두 개 이상의 개념을 혼합하여 새로운 의미를 창출합니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "input_space1": "첫 번째 개념",
@@ -70,6 +72,7 @@ CONCEPTUAL_BLENDING_PROMPT = """
 }
 
 예시: Netflix = Net(인터넷) + Flix(영화) → 온라인 스트리밍의 새로운 개념
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 3. SOUND SYMBOLISM =====
@@ -82,7 +85,7 @@ SOUND_SYMBOLISM_PROMPT = """
 - 유음(l,r,m,n): 부드럽고 유연함
 - 파열음(p,t,k,b,d,g): 강하고 역동적
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "phonetic_analysis": "음성 분석",
@@ -92,13 +95,15 @@ SOUND_SYMBOLISM_PROMPT = """
   "pronunciation_guide": "발음 가이드",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 4. LINGUISTIC RELATIVITY =====
 LINGUISTIC_RELATIVITY_PROMPT = """
 당신은 Linguistic Relativity 전문가입니다. 언어별 사고방식 차이를 고려합니다.
 
-다언어 브랜드 생성 형식:
+다음 JSON 형식으로 다언어 브랜드를 생성하세요:
 {
   "brand_name": "글로벌 브랜드명",
   "korean_adaptation": "한국어 적응",
@@ -108,6 +113,8 @@ LINGUISTIC_RELATIVITY_PROMPT = """
   "localization_strategy": "현지화 전략",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 5. ARCHETYPE THEORY =====
@@ -116,7 +123,7 @@ ARCHETYPE_THEORY_PROMPT = """
 
 12 원형: Innocent, Hero, Outlaw, Explorer, Creator, Ruler, Magician, Lover, Caregiver, Jester, Sage, Regular Guy
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "archetype": "선택된 원형",
@@ -127,13 +134,15 @@ ARCHETYPE_THEORY_PROMPT = """
   "slogan": "슬로건",
   "mythology_reference": "신화적 참조"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 6. JOBS-TO-BE-DONE =====
 JOBS_TO_BE_DONE_PROMPT = """
 당신은 Jobs-to-be-Done Theory 전문가입니다. 고객이 해결하려는 '일'에 초점을 맞춥니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "functional_job": "기능적 일",
@@ -144,6 +153,8 @@ JOBS_TO_BE_DONE_PROMPT = """
   "slogan": "슬로건",
   "value_proposition": "가치 제안"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 7. SCAMPER METHOD =====
@@ -158,7 +169,7 @@ P - Put to another use (다른 용도)
 E - Eliminate (제거)
 R - Reverse (역전)
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "scamper_technique": "사용된 기법",
@@ -167,13 +178,15 @@ R - Reverse (역전)
   "innovative_aspect": "혁신적 측면",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 8. DESIGN THINKING =====
 DESIGN_THINKING_PROMPT = """
 당신은 IDEO의 Design Thinking 전문가입니다. 인간 중심 혁신을 추구합니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "user_insight": "사용자 통찰",
@@ -184,13 +197,15 @@ DESIGN_THINKING_PROMPT = """
   "prototype_concept": "프로토타입 컨셉",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 9. BIOMIMICRY =====
 BIOMIMICRY_PROMPT = """
 당신은 Biomimicry 전문가입니다. 자연에서 영감을 받은 브랜드를 만듭니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "natural_inspiration": "자연적 영감원",
@@ -200,13 +215,15 @@ BIOMIMICRY_PROMPT = """
   "adaptation_strategy": "적응 전략",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 10. COGNITIVE LOAD =====
 COGNITIVE_LOAD_PROMPT = """
 당신은 Cognitive Load Theory 전문가입니다. 인지 처리를 최소화하는 브랜드를 만듭니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "syllable_count": "음절 수",
@@ -216,13 +233,15 @@ COGNITIVE_LOAD_PROMPT = """
   "cognitive_fluency": "인지적 유창성",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 11. VON RESTORFF EFFECT =====
 VON_RESTORFF_PROMPT = """
 당신은 Von Restorff Effect 전문가입니다. 독특하고 기억에 남는 브랜드를 만듭니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "category_norm": "카테고리 표준",
@@ -232,13 +251,15 @@ VON_RESTORFF_PROMPT = """
   "attention_triggers": "주의 트리거",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 12. NETWORK EFFECTS =====
 NETWORK_EFFECTS_PROMPT = """
 당신은 Network Effects 전문가입니다. 네트워크 가치를 극대화하는 브랜드를 만듭니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "network_type": "네트워크 유형",
@@ -248,13 +269,15 @@ NETWORK_EFFECTS_PROMPT = """
   "network_value": "네트워크 가치",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 13. MEMETICS =====
 MEMETICS_PROMPT = """
 당신은 Memetics 전문가입니다. 문화적으로 복제되고 진화하는 브랜드를 만듭니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "meme_structure": "밈 구조",
@@ -264,13 +287,15 @@ MEMETICS_PROMPT = """
   "transmission_channels": "전달 채널",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 14. COLOR PSYCHOLOGY =====
 COLOR_PSYCHOLOGY_PROMPT = """
 당신은 Color Psychology 전문가입니다. 색상 연상과 감정을 활용한 브랜드를 만듭니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "primary_color": "주 색상",
@@ -280,13 +305,15 @@ COLOR_PSYCHOLOGY_PROMPT = """
   "industry_alignment": "업종 정렬",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # ===== 15. GESTALT PRINCIPLES =====
 GESTALT_PROMPT = """
 당신은 Gestalt Theory 전문가입니다. 지각 원리를 활용한 브랜드를 만듭니다.
 
-브랜드 생성 형식:
+다음 JSON 형식으로 브랜드를 생성하세요:
 {
   "brand_name": "브랜드명",
   "gestalt_principle": "활용 원칙",
@@ -296,6 +323,8 @@ GESTALT_PROMPT = """
   "closure_effect": "폐쇄 효과",
   "slogan": "슬로건"
 }
+
+반드시 유효한 JSON 형식으로 응답하세요.
 """
 
 # 이론별 프롬프트 매핑
@@ -324,7 +353,12 @@ def generate_by_theory(industry: str, keywords: str, theory: str, count: int = 3
         return "⚠️ 업종과 키워드를 입력해주세요.", ""
     
     prompt = THEORY_PROMPTS.get(theory, SQUARE_THEORY_PROMPT)
-    user_input = f"업종: {industry}\n키워드: {keywords}\n\n위 정보로 {count}개의 브랜드를 생성하세요."
+    user_input = f"""업종: {industry}
+키워드: {keywords}
+
+위 정보로 {count}개의 브랜드를 생성하세요. 
+반드시 JSON 형식으로 응답하세요. 
+결과는 브랜드 객체들의 JSON 배열이어야 합니다."""
     
     try:
         response = client.chat.completions.create(
@@ -343,7 +377,23 @@ def generate_by_theory(industry: str, keywords: str, theory: str, count: int = 3
         
         # 응답 정규화
         if isinstance(data, dict):
-            results = data.get("results", data.get("brands", [data]))
+            if "results" in data:
+                results = data["results"]
+            elif "brands" in data:
+                results = data["brands"]
+            elif "brand_name" in data:
+                # 단일 브랜드를 배열로 변환
+                results = [data]
+            else:
+                # 래핑된 응답 처리
+                results = []
+                for key, value in data.items():
+                    if isinstance(value, list):
+                        results = value
+                        break
+                    elif isinstance(value, dict) and "brand_name" in value:
+                        results = [value]
+                        break
         else:
             results = data
             
@@ -497,15 +547,21 @@ def visualize_square_brand(brand: Dict) -> str:
 
 def visualize_conceptual_blend(brand: Dict) -> str:
     """Conceptual Blending 시각화"""
+    brand_name = brand.get('brand_name', 'Brand')
+    input1 = brand.get('input_space1', 'Concept 1')
+    input2 = brand.get('input_space2', 'Concept 2')
+    blended = brand.get('blended_space', 'Blended Concept')
+    slogan = brand.get('slogan', '')
+    
     return f"""
 <div style="max-width: 700px; margin: 20px auto; font-family: -apple-system, sans-serif;">
-    <h2 style="text-align: center; color: #2c3e50;">{brand.get('brand_name', 'Brand')}</h2>
+    <h2 style="text-align: center; color: #2c3e50;">{brand_name}</h2>
     
     <div style="display: flex; justify-content: space-around; align-items: center; margin: 30px 0;">
         <div style="text-align: center; padding: 20px; background: #3498db; color: white; border-radius: 50%; width: 120px; height: 120px; display: flex; align-items: center; justify-content: center;">
             <div>
                 <strong>Input 1</strong><br>
-                {brand.get('input_space1', '')}
+                {input1}
             </div>
         </div>
         
@@ -514,7 +570,7 @@ def visualize_conceptual_blend(brand: Dict) -> str:
         <div style="text-align: center; padding: 20px; background: #e74c3c; color: white; border-radius: 50%; width: 120px; height: 120px; display: flex; align-items: center; justify-content: center;">
             <div>
                 <strong>Input 2</strong><br>
-                {brand.get('input_space2', '')}
+                {input2}
             </div>
         </div>
         
@@ -523,12 +579,12 @@ def visualize_conceptual_blend(brand: Dict) -> str:
         <div style="text-align: center; padding: 20px; background: #27ae60; color: white; border-radius: 50%; width: 150px; height: 150px; display: flex; align-items: center; justify-content: center;">
             <div>
                 <strong>Blend</strong><br>
-                {brand.get('blended_space', '')}
+                {blended}
             </div>
         </div>
     </div>
     
-    <p style="text-align: center; font-style: italic;">"{brand.get('slogan', '')}"</p>
+    <p style="text-align: center; font-style: italic;">"{slogan}"</p>
 </div>
 """