Update app.py

app.py

@@ -3,6 +3,9 @@ import os, json, re, logging, requests, markdown, time, io
 from datetime import datetime
 
 import streamlit as st
+
+st.set_page_config(layout="wide") 
+
 from openai import OpenAI  # OpenAI 라이브러리
 
 from gradio_client import Client
@@ -312,48 +315,51 @@ def get_openai_client():
     )
 
 # ──────────────────────────────── New System Prompt for Idea Generation ─────────────────────
-def get_idea_system_prompt() -> str:
+# ───────────────── System Prompt (REPORT STYLE) ─────────────────
+def get_idea_system_prompt(selected_category: str | None = None) -> str:
     """
-    아이디어 생성용 시스템 프롬프트:
-    - 제공된 physical_transformation_categories를 활용
-    - 3가지 발상 접근: (1) 유사성 기반 증강, (2) 반유사성(대조) 기반 발상의 전환, (3) 무작위(우연적 착안)
-    - 각 아이디어를 논리적으로 설명 가능하도록 지시
-    - 아이디어마다 이미지 프롬프트도 함께 제시할 것
-    """
-    prompt = """
-반드시 한글(한국어)로 답변하라. You are an advanced Idea Generation AI. Your goal is to take the user's prompt (topic, concept, or question) and create **3 distinct ideas** around it, drawing from the vast categories of **physical or conceptual transformations**. Specifically, you must produce:
-
-1. **Similarity-based Augmentation**  
-   - Build upon the user's original concept by closely aligning to its core attributes.  
-   - Reference relevant categories or subcategories that naturally extend or amplify the original idea.  
-   - Provide a practical, logical explanation.  
-
-2. **Contrarian/Inverse Approach**  
-   - Intentionally invert or contradict the core assumptions of the original concept.  
-   - Use contrasting categories to spark unconventional thinking.  
-   - Clearly justify why this contrarian perspective might reveal novel possibilities.  
-
-3. **Random (Fortuitous Spark)**  
-   - Randomly select one or more categories or subcategories from any domain to introduce surprising or serendipitous ideas.  
-   - Show how this unexpected combination can yield fresh insights or opportunities.  
-
-For each of the 3 approaches:
-- Provide a concise name or title for the idea.  
-- Offer a clear, structured explanation (a few paragraphs) on how this idea works or could be realized.  
-- Incorporate references to the relevant category or categories from the provided classification, explaining how they link to the idea.  
-- Suggest a short, single-line **English prompt for generating an image** that best represents that idea.
-
-Finally:
-- Return your answer in Markdown format with headings (e.g., "## Idea 1: ...", "## Idea 2: ...", "## Idea 3: ...").
-- Do **not** mention that you are responding with "3 ideas" because of a prompt or instruction. Simply present them as the final structured output.
-- Never mention "blog," "SEO," or "prompt" instructions in your final text. 
-- Do **not** mention that you're an AI. Present the content as a straightforward creative text for each idea.
-
----
-**Important**: You have access to a list of "physical_transformation_categories" (in Korean) and possibly an English version. Use them to anchor or inspire each idea, but do not simply copy the entire list. Instead, selectively reference a few relevant subcategories.
+    Return an enhanced system prompt that forces:
+    ▸ Report-style output with tables
+    ▸ Clear focus on NEED ▸ VALUE ▸ PROBLEM SOLVED ▸ IMPLEMENTATION SNAPSHOT
+    ▸ Three idea modes: Similarity-based, Contrarian, Random
+    ▸ If `selected_category` is given, all ideas must chiefly reference
+      sub-items inside that category.
     """
+    cat_clause = (f"\n**Category Focus**: Concentrate on the Korean list "
+                  f"under “{selected_category}”. You *must* anchor every idea "
+                  "to 2-3 sub-items from this category and reference them by name.\n")
+    prompt = f"""
+반드시 한글(한국어)로 답변하라. You are an elite innovation consultant. Your deliverable is a concise **report** in Markdown.
+
+### OUTPUT SPECIFICATION
+Produce **exactly three ideas** (Similarity-Based, Contrarian, Random) using these sections **for *each* idea**:
+
+| 항목 | 내용 |
+|---|---|
+| **Idea Title** | 8-12 Korean characters + 영문 괄호별칭 |
+| **Need / Pain-Point** | 어떤 문제·기회 때문에 필요한가? (1-2 문장) |
+| **Core Concept** | 아이디어 개요 (2-3 문단) |
+| **Key Category Links** | 사용한 변환 카테고리/하위항목(3개 이하) |
+| **Value Proposition** | 기대 효과·가치 (불릿 3개) |
+| **Potential Obstacles** | 주요 리스크·제약 (불릿 2-3개) |
+| **Image Prompt** | 영문 1줄로 시각화를 위한 프롬프트 |
+
+- 포함된 표는 마크다운 그대로 출력.
+- 세 아이디어를 “## Idea 1”, “## Idea 2”, “## Idea 3” 헤딩으로 순서화.
+- 전문용어 최소화, 간결하지만 설득력 있게.
+
+### IDEA MODES
+1. **Similarity-Based Augmentation** – 기존 주제와 가장 가깝게 확장.
+2. **Contrarian / Inverse** – 주제의 전제를 뒤집어 새로운 시각 제시.
+3. **Random Spark** – 무작위 하위항목(카테고리 불문)을 결합한 의외의 통찰.
+
+{cat_clause if selected_category else ""}
+Always think step-by-step but **show only the final report**.
+"""
     return prompt.strip()
 
+
+
 # ──────────────────────────────── Brave Search API ────────────────────────
 @st.cache_data(ttl=3600)
 def brave_search(query: str, count: int = 20):
@@ -635,6 +641,16 @@ def idea_generator_app():
         "example3": "미래형 교육 서비스 구상"
     }
 
+    sb.subheader("Category Focus (Optional)")
+    category_keys = ["(None)"] + list(physical_transformation_categories.keys())
+    sb.selectbox(
+        "Generate ideas mainly using this category",
+        options=category_keys,
+        key="category_focus",
+        index=0  # 기본값 "(None)"
+    )
+
+    
     sb.subheader("Example Prompts")
     c1, c2, c3 = sb.columns(3)
     if c1.button("AI & 비즈니스", key="ex1"):