Update app.py

app.py

@@ -14,20 +14,27 @@ BRAVE_ENDPOINT = "https://api.search.brave.com/res/v1/web/search"
 IMAGE_API_URL  = "http://211.233.58.201:7896"
 MAX_TOKENS     = 7_999
 
-# 블로그 템플릿 및 스타일 정의
+# 블로그 템플릿 및 스타일 정의 (한글화)
 BLOG_TEMPLATES = {
-    "standard": "전문 블로그 작성 전문가로서 8단계 프레임워크를 따라 자연스럽고 매력적인 글 작성",
-    "tutorial": "단계별 튜토리얼 형식으로, 명확한 과정과 결과를 보여주는 가이드 작성",
-    "review": "제품/서비스 분석 중심의 리뷰 형식, 장단점 분석과 추천 포함",
-    "storytelling": "개인 경험이나 사례를 중심으로 한 스토리텔링 형식의 블로그 작성",
-    "seo_optimized": "검색엔진 최적화(SEO)를 고려한 키워드 중심 블로그 작성"
+    "standard": "표준 8단계 프레임워크 블로그",
+    "tutorial": "단계별 튜토리얼 형식",
+    "review": "제품/서비스 리뷰 형식",
+    "storytelling": "스토리텔링 형식",
+    "seo_optimized": "SEO 최적화 블로그"
 }
 
 BLOG_TONES = {
-    "professional": "전문적이고 공식적인 어조로 작성",
-    "casual": "친근하고 대화체 중심의 편안한 톤으로 작성",
-    "humorous": "유머와 재치를 가미한 가벼운 어조로 작성",
-    "storytelling": "이야기를 들려주듯 감성적이고 몰입감 있는 톤으로 작성"
+    "professional": "전문적이고 공식적인 톤",
+    "casual": "친근하고 대화체 중심 톤",
+    "humorous": "유머러스한 접근",
+    "storytelling": "이야기 중심의 접근"
+}
+
+# 예제 블로그 주제
+EXAMPLE_TOPICS = {
+    "example1": "2025년 바뀐 부동산 세금 제도: 일반 가정에 미치는 영향과 절세 전략",
+    "example2": "2025년 여름 전국 지역별 대표 축제 총정리와 숨은 명소 추천",
+    "example3": "2025년 주목해야 할 신성장 산업 투자 가이드: 인공지능 관련 발굴 전략"
 }
 
 # ──────────────────────────────── 로깅 ──────────────────────────────────────
@@ -155,7 +162,7 @@ def get_system_prompt(template="standard", tone="professional", word_count=1750)
     return final_prompt
 
 # ──────────────────────────────── Brave Search API ─────────────────────────
-def brave_search(query: str, count: int = 5):
+def brave_search(query: str, count: int = 20):  # 기본값을 20으로 변경
     """
     Brave Web Search API 호출 → list[dict]
     반환 필드: index, title, link, snippet, displayed_link
@@ -168,7 +175,7 @@ def brave_search(query: str, count: int = 5):
         "Accept-Encoding": "gzip",
         "X-Subscription-Token": BRAVE_KEY
     }
-    params = {"q": query, "count": str(count)}
+    params = {"q": query, "count": str(count)}  # 카운트 파라미터 전달
     
     for attempt in range(3):  # 최대 3번 재시도
         try:
@@ -185,7 +192,7 @@ def brave_search(query: str, count: int = 5):
                 raise ValueError("검색 결과가 없습니다")
                 
             arts = []
-            for i, res in enumerate(raw[:count], 1):
+            for i, res in enumerate(raw[:count], 1):  # count만큼 반복
                 url   = res.get("url", res.get("link", ""))
                 host  = re.sub(r"https?://(www\.)?", "", url).split("/")[0]
                 arts.append({
@@ -221,7 +228,7 @@ def mock_results(query: str) -> str:
 def do_web_search(query: str) -> str:
     """웹 검색 수행 및 결과 포맷팅"""
     try:
-        arts = brave_search(query, 5)
+        arts = brave_search(query, 20)  # 여기도 20으로 변경
         if not arts:
             logging.warning("검색 결과 없음, 대체 콘텐츠 사용")
             return mock_results(query)
@@ -271,7 +278,7 @@ def keywords(text: str, top=5):
 
 # ──────────────────────────────── Streamlit UI ────────────────────────────
 def ginigen_app():
-    st.title("Ginigen Blog")
+    st.title("지니젠 블로그")
 
     # 세션 기본값
     defaults = dict(
@@ -294,17 +301,31 @@ def ginigen_app():
     # 블로그 템플릿 및 스타일 선택
     sb.subheader("블로그 스타일 설정")
     sb.selectbox("블로그 템플릿", options=list(BLOG_TEMPLATES.keys()), 
-                format_func=lambda x: x.replace("_", " ").title(),
+                format_func=lambda x: BLOG_TEMPLATES[x],
                 key="blog_template")
     
     sb.selectbox("블로그 톤", options=list(BLOG_TONES.keys()),
-                format_func=lambda x: x.replace("_", " ").title(),
+                format_func=lambda x: BLOG_TONES[x],
                 key="blog_tone")
     
     sb.slider("블로그 길이 (단어 수)", 800, 3000, 1750, key="word_count")
     
+    # 예제 주제 선택
+    sb.subheader("예제 주제")
+    if sb.button("자바스크립트 프레임워크 비교", key="example1"):
+        st.session_state.example_topic = EXAMPLE_TOPICS["example1"]
+        st.experimental_rerun()
+        
+    if sb.button("초보자를 위한 주식 투자 가이드", key="example2"):
+        st.session_state.example_topic = EXAMPLE_TOPICS["example2"]
+        st.experimental_rerun()
+        
+    if sb.button("효과적인 시간 관리 전략", key="example3"):
+        st.session_state.example_topic = EXAMPLE_TOPICS["example3"]
+        st.experimental_rerun()
+    
     sb.subheader("기타 설정")
-    sb.toggle("자동 저장",   key="auto_save")
+    sb.toggle("자동 저장", key="auto_save")
     sb.toggle("이미지 자동 생성", key="generate_image")
     
     # 웹 검색 토글 (모니터링을 위해 유지하되 기본값은 False)
@@ -322,7 +343,7 @@ def ginigen_app():
         title = title.group(1).strip() if title else "blog"
         sb.subheader("최근 블로그 다운로드")
         c1, c2 = sb.columns(2)
-        c1.download_button("Markdown", latest_blog,
+        c1.download_button("마크다운", latest_blog,
                            file_name=f"{title}.md", mime="text/markdown")
         c2.download_button("HTML", md_to_html(latest_blog, title),
                            file_name=f"{title}.html", mime="text/html")
@@ -350,8 +371,16 @@ def ginigen_app():
             if "image" in m:
                 st.image(m["image"], caption=m.get("image_caption", ""))
 
-    # ── 사용자 입력
-    if prompt := st.chat_input("무엇을 도와드릴까요?"):
+    # ── 사용자 입력 또는 예제 주제 처리
+    prompt = st.chat_input("무엇을 도와드릴까요?")
+    
+    # 예제 주제가 선택되었으면 처리
+    if hasattr(st.session_state, 'example_topic') and st.session_state.example_topic:
+        prompt = st.session_state.example_topic
+        # 처리 후 예제 주제 초기화
+        del st.session_state.example_topic
+    
+    if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt})
         with st.chat_message("user"): st.markdown(prompt)
 
@@ -400,7 +429,7 @@ def ginigen_app():
             # 본문 다운로드 버튼 (MD / HTML)
             st.subheader("이 블로그 다운로드")
             b1, b2 = st.columns(2)
-            b1.download_button("Markdown", answer,
+            b1.download_button("마크다운", answer,
                                file_name=f"{prompt[:30]}.md", mime="text/markdown")
             b2.download_button("HTML", md_to_html(answer, prompt[:30]),
                                file_name=f"{prompt[:30]}.html", mime="text/html")