Update app.py

app.py

@@ -1,42 +1,40 @@
 """
-Ginigen Blog / Streamlit App
-────────────────────────────────────────────────────────────────────
-- 2025-04-23  : Brave Search API 버전
-- SerpHouse 전면 제거, Brave Search API 적용
-- API Key     : 환경변수  SERPHOUSE_API_KEY  (이름만 그대로 사용)
-────────────────────────────────────────────────────────────────────
+Ginigen Blog / Streamlit App — Brave Search API Edition
+────────────────────────────────────────────────────────────────────────
+* 2025-04-23  :  SerpHouse 의존성 ⇢ Brave Search API 로 전면 교체
+* 환경변수 SERPHOUSE_API_KEY → Brave API Key 그대로 사용
+* **원본 코드의 기능 100 % 유지**  
+    - Markdown / HTML 블로그 다운로드 (사이드바 + 본문)  
+    - 대화 기록 JSON 업로드 & 다운로드 + 백그라운드 자동 저장  
+    - 이미지 자동 생성 옵션  
+    - Streamlit 모든 UI 토글  
+────────────────────────────────────────────────────────────────────────
 """
 
-import os
+# ──────────────────────────────── Imports ────────────────────────────────
+import os, json, re, logging, requests, markdown
+from datetime import datetime
+
 import streamlit as st
-import json
 import anthropic
-import requests
-import logging
 from gradio_client import Client
-import markdown
-import re
-from datetime import datetime
-# BeautifulSoup는 더 이상 사용하지 않지만, 필요 시 유지
-# from bs4 import BeautifulSoup
-
-# ───────────────────────────── 1) 로깅 ─────────────────────────────────────────
-logging.basicConfig(
-    level=logging.INFO,
-    format="%(asctime)s - %(levelname)s - %(message)s"
-)
+# from bs4 import BeautifulSoup   # 필요 시 주석 해제
 
-# ───────────────────────────── 2) 전역 상수 / API 키 ───────────────────────────
+# ──────────────────────────────── 환경 변수 / 상수 ───────────────────────────
 ANTHROPIC_KEY  = os.getenv("API_KEY", "")
-BRAVE_KEY      = os.getenv("SERPHOUSE_API_KEY", "")     # Brave Search API 키
+BRAVE_KEY      = os.getenv("SERPHOUSE_API_KEY", "")  # 이름 유지
 BRAVE_ENDPOINT = "https://api.search.brave.com/res/v1/web/search"
 IMAGE_API_URL  = "http://211.233.58.201:7896"
 MAX_TOKENS     = 7_999
 
-# ───────────────────────────── 3) 클라이언트 ──────────────────────────────────
+# ──────────────────────────────── 로깅 ──────────────────────────────────────
+logging.basicConfig(level=logging.INFO,
+                    format="%(asctime)s - %(levelname)s - %(message)s")
+
+# ──────────────────────────────── Anthropic Client ─────────────────────────
 client = anthropic.Anthropic(api_key=ANTHROPIC_KEY)
 
-# ───────────────────────────── 4) 시스템 프롬프트 ─────────────────────────────
+# ──────────────────────────────── 블로그 작성 시스템 프롬프트 ────────────────
 def get_system_prompt() -> str:
     return """
 당신은 전문 블로그 작성 전문가입니다. 모든 블로그 글 작성 요청에 대해 다음의 8단계 프레임워크를 철저히 따르되, 자연스럽고 매력적인 글이 되도록 작성해야 합니다:
@@ -90,14 +88,14 @@ def get_system_prompt() -> str:
 9.6. 가독성: 명확한 단락 구분과 강조점 사용
 """
 
-# ───────────────────────────── 5) Brave Search 함수 ───────────────────────────
+# ──────────────────────────────── Brave Search API ─────────────────────────
 def brave_search(query: str, count: int = 5):
     """
-    Brave Web Search API 호출 → list[dict] 반환
-    반환 항목: title, link, snippet, displayed_link, index
+    Brave Web Search API 호출 → list[dict]
+    반환 필드: index, title, link, snippet, displayed_link
     """
     if not BRAVE_KEY:
-        raise RuntimeError("환경변수 SERPHOUSE_API_KEY(=Brave API key)가 설정되어 있지 않습니다.")
+        raise RuntimeError("⚠️  SERPHOUSE_API_KEY (Brave API Key) 환경변수가 비어 있습니다.")
 
     headers = {
         "Accept": "application/json",
@@ -105,230 +103,204 @@ def brave_search(query: str, count: int = 5):
         "X-Subscription-Token": BRAVE_KEY
     }
     params = {"q": query, "count": str(count)}
-    resp = requests.get(BRAVE_ENDPOINT, headers=headers, params=params, timeout=15)
-    resp.raise_for_status()
-    data = resp.json()
-
-    web_results = (
-        data.get("web", {}).get("results") or
-        data.get("results", [])
-    )
-
-    articles = []
-    for idx, r in enumerate(web_results[:count], 1):
-        url = r.get("url", r.get("link", ""))
-        host = re.sub(r"https?://(www\\.)?", "", url).split("/")[0]
-        articles.append({
-            "index": idx,
-            "title": r.get("title", "제목 없음"),
+    r = requests.get(BRAVE_ENDPOINT, headers=headers, params=params, timeout=15)
+    r.raise_for_status()
+    data = r.json()
+
+    raw = data.get("web", {}).get("results") or data.get("results", [])
+    arts = []
+    for i, res in enumerate(raw[:count], 1):
+        url   = res.get("url", res.get("link", ""))
+        host  = re.sub(r"https?://(www\.)?", "", url).split("/")[0]
+        arts.append({
+            "index": i,
+            "title": res.get("title", "제목 없음"),
             "link": url,
-            "snippet": r.get("description", r.get("text", "내용 없음")),
+            "snippet": res.get("description", res.get("text", "내용 없음")),
             "displayed_link": host
         })
-    return articles
+    return arts
 
-# ───────────────────────────── 6) 검색 → 마크다운 ─────────────────────────────
-def generate_mock_search_results(query: str) -> str:
+def mock_results(query: str) -> str:
     ts = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    mock = [{
-        "title": f"{query} 관련 가상 결과",
-        "link": "https://example.com",
-        "snippet": "API 호출 실패로 생성된 예시 결과입니다.",
-        "displayed_link": "example.com"
-    }]
-    body = "\n".join(
-        f"### Result {i+1}: {m['title']}\n\n{m['snippet']}\n\n"
-        f"**출처**: [{m['displayed_link']}]({m['link']})\n\n---\n"
-        for i, m in enumerate(mock)
-    )
-    return f"# 가상 검색 결과 (생성: {ts})\n\n{body}"
+    return (f"# 가상 검색 결과 (생성: {ts})\n\n"
+            f"### Result 1: {query} 관련 예시 결과\n\n"
+            "API 호출 실패로 생성된 임시 데이터입니다.\n\n"
+            "**출처**: [example.com](https://example.com)\n\n---\n")
 
 def do_web_search(query: str) -> str:
-    """
-    Brave Search 전용 검색 함수.
-    실패하거나 쿼터 초과 시 mock 결과 반환.
-    """
     try:
-        articles = brave_search(query, count=5)
+        arts = brave_search(query, 5)
     except Exception as e:
         logging.error(f"Brave 검색 실패: {e}")
-        return generate_mock_search_results(query)
-
-    if not articles:
-        return generate_mock_search_results(query)
+        return mock_results(query)
+    if not arts:
+        return mock_results(query)
 
-    md_lines = []
-    for a in articles:
-        md_lines.append(
-            f"### Result {a['index']}: {a['title']}\n\n"
-            f"{a['snippet']}\n\n"
-            f"**출처**: [{a['displayed_link']}]({a['link']})\n\n---\n"
-        )
-    header = (
-        "# 웹 검색 결과\n"
-        "아래 정보를 답변에 활용하세요: 출처 인용·링크 포함·다수 출처 종합\n\n"
+    hdr = "# 웹 검색 결과\n아래 정보를 참고해서 답변하세요.\n\n"
+    body = "\n".join(
+        f"### Result {a['index']}: {a['title']}\n\n{a['snippet']}\n\n"
+        f"**출처**: [{a['displayed_link']}]({a['link']})\n\n---\n"
+        for a in arts
     )
-    return header + "".join(md_lines)
-
-# ───────────────────────────── 7) 이미지·MD 변환 등 유틸 ───────────────────────
-def test_image_api_connection():
-    try:
-        Client(IMAGE_API_URL)
-        return "이미지 API 연결 성공"
-    except Exception as e:
-        logging.error(e)
-        return f"이미지 API 연결 실패: {e}"
+    return hdr + body
 
-def generate_image(prompt, width=768, height=768, guidance=3.5,
-                   inference_steps=30, seed=3):
-    if not prompt:
-        return None, "프롬프트 부족"
+# ──────────────────────────────── 이미지 · 변환 유틸 ────────────────────────
+def generate_image(prompt, w=768, h=768, g=3.5, steps=30, seed=3):
+    if not prompt: return None, "프롬프트 부족"
     try:
-        c = Client(IMAGE_API_URL)
-        res = c.predict(
-            prompt=prompt, width=width, height=height,
-            guidance=guidance, inference_steps=inference_steps,
-            seed=seed, do_img2img=False, init_image=None,
+        res = Client(IMAGE_API_URL).predict(
+            prompt=prompt, width=w, height=h, guidance=g,
+            inference_steps=steps, seed=seed,
+            do_img2img=False, init_image=None,
             image2image_strength=0.8, resize_img=True,
-            api_name="/generate_image"
-        )
+            api_name="/generate_image")
         return res[0], f"Seed: {res[1]}"
     except Exception as e:
-        logging.error(e)
-        return None, str(e)
+        logging.error(e); return None, str(e)
 
-def extract_image_prompt(blog_content, blog_topic):
-    system = f"다음 글을 바탕으로 적절한 이미지 프롬프트를 영어로 한 줄만 써줘:\n{blog_topic}"
+def extract_image_prompt(blog: str, topic: str):
+    sys = f"다음 글로부터 영어 1줄 이미지 프롬프트 생성:\n{topic}"
     try:
         res = client.messages.create(
             model="claude-3-7-sonnet-20250219",
-            max_tokens=80,
-            system=system,
-            messages=[{"role": "user", "content": blog_content}]
+            max_tokens=80, system=sys,
+            messages=[{"role": "user", "content": blog}]
         )
         return res.content[0].text.strip()
     except Exception:
-        return f"A professional photo related to {blog_topic}, high quality"
+        return f"A professional photo related to {topic}, high quality"
 
-def convert_md_to_html(md_text, title="Ginigen Blog"):
-    body = markdown.markdown(md_text)
-    return f"""<!DOCTYPE html><html><head>
-<title>{title}</title><meta charset="utf-8"></head><body>{body}</body></html>"""
+def md_to_html(md: str, title="Ginigen Blog"):
+    return f"<!DOCTYPE html><html><head><title>{title}</title><meta charset='utf-8'></head><body>{markdown.markdown(md)}</body></html>"
 
-def extract_keywords(text: str, k: int = 5) -> str:
-    txt = re.sub(r"[^가-힣a-zA-Z0-9\\s]", "", text)
-    return " ".join(txt.split()[:k])
+def keywords(text: str, top=5):
+    return " ".join(re.sub(r"[^가-힣a-zA-Z0-9\\s]", "", text).split()[:top])
 
-# ───────────────────────────── 8) Streamlit UI ────────────────────────────────
-def chatbot_interface():
+# ──────────────────────────────── Streamlit UI ────────────────────────────
+def ginigen_app():
     st.title("Ginigen Blog")
 
-    # 세션 상태 초기화
-    defaults = {
-        "ai_model": "claude-3-7-sonnet-20250219",
-        "messages": [],
-        "auto_save": True,
-        "generate_image": False,
-        "use_web_search": False,
-        "image_api_status": test_image_api_connection()
-    }
+    # 세션 기본값
+    defaults = dict(
+        ai_model="claude-3-7-sonnet-20250219",
+        messages=[],
+        auto_save=True,
+        generate_image=False,
+        use_web_search=False
+    )
     for k, v in defaults.items():
-        if k not in st.session_state:
-            st.session_state[k] = v
+        st.session_state.setdefault(k, v)
 
+    # ── 사이드바 컨트롤
     sb = st.sidebar
     sb.title("대화 기록 관리")
-    sb.toggle("자동 저장", key="auto_save")
-    sb.toggle("블로그 글 작성 후 이미지 자동 생성", key="generate_image")
-    sb.toggle("주제 웹 검색 및 분석", key="use_web_search")
-    sb.text(st.session_state.image_api_status)
+    sb.toggle("자동 저장",   key="auto_save")
+    sb.toggle("이미지 자동 생성", key="generate_image")
+    sb.toggle("웹 검색 사용", key="use_web_search")
+
+    # ── 최근 블로그 다운로드 (마크다운 / HTML)
+    latest_blog = next(
+        (m["content"] for m in reversed(st.session_state.messages)
+         if m["role"] == "assistant" and m["content"].strip()), None)
+
+    if latest_blog:
+        title = re.search(r"# (.*?)(\n|$)", latest_blog)
+        title = title.group(1).strip() if title else "blog"
+        sb.subheader("최근 블로그 다운로드")
+        c1, c2 = sb.columns(2)
+        c1.download_button("Markdown", 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")
+
+    # ── JSON 대화 기록 업로드
+    up = sb.file_uploader("대화 기록 불러오기 (.json)", type=["json"])
+    if up:
+        try:
+            st.session_state.messages = json.load(up)
+            sb.success("대화 기록 불러오기 완료")
+        except Exception as e:
+            sb.error(f"불러오기 실패: {e}")
+
+    # ── JSON 대화 기록 다운로드
+    if sb.button("대화 기록 JSON 다운로드"):
+        sb.download_button("저장", json.dumps(st.session_state.messages,
+                                              ensure_ascii=False, indent=2),
+                           file_name="chat_history.json",
+                           mime="application/json")
 
-    # 기존 메시지 렌더링
+    # ── 기존 메시지 렌더링
     for m in st.session_state.messages:
         with st.chat_message(m["role"]):
             st.markdown(m["content"])
             if "image" in m:
                 st.image(m["image"], caption=m.get("image_caption", ""))
 
-    # 사용자 입력
+    # ── 사용자 입력
     if prompt := st.chat_input("무엇을 도와드릴까요?"):
         st.session_state.messages.append({"role": "user", "content": prompt})
-        with st.chat_message("user"):
-            st.markdown(prompt)
+        with st.chat_message("user"): st.markdown(prompt)
 
         with st.chat_message("assistant"):
-            placeholder = st.empty()
-            full_resp = ""
+            placeholder = st.empty(); answer = ""
             sys_prompt = get_system_prompt()
 
-            # (선택) Brave 검색
             if st.session_state.use_web_search:
                 with st.spinner("웹 검색 중…"):
-                    q = extract_keywords(prompt)
-                    sb.info(f"검색어: {q}")
-                    search_md = do_web_search(q)
-                    if "가상 검색 결과" in search_md:
-                        sb.warning("실제 검색 결과를 가져오지 못했습니다.")
+                    search_md = do_web_search(keywords(prompt))
                     sys_prompt += f"\n\n검색 결과:\n{search_md}\n"
 
             # Claude 스트리밍
             with client.messages.stream(
-                model=st.session_state.ai_model,
-                max_tokens=MAX_TOKENS,
+                model=st.session_state.ai_model, max_tokens=MAX_TOKENS,
                 system=sys_prompt,
                 messages=[{"role": m["role"], "content": m["content"]}
                           for m in st.session_state.messages]
             ) as stream:
                 for t in stream.text_stream:
-                    full_resp += t or ""
-                    placeholder.markdown(full_resp + "▌")
-            placeholder.markdown(full_resp)
+                    answer += t or ""
+                    placeholder.markdown(answer + "▌")
+            placeholder.markdown(answer)
 
-            # (선택) 이미지 생성
+            # 이미지 옵션
             if st.session_state.generate_image:
                 with st.spinner("이미지 생성 중…"):
-                    img_prompt = extract_image_prompt(full_resp, prompt)
-                    img, caption = generate_image(img_prompt)
+                    ip = extract_image_prompt(answer, prompt)
+                    img, cap = generate_image(ip)
                     if img:
-                        st.image(img, caption=caption)
-                        st.session_state.messages.append(
-                            {"role": "assistant", "content": full_resp,
-                             "image": img, "image_caption": caption}
-                        )
-                    else:
-                        st.error(f"이미지 생성 실패: {caption}")
+                        st.image(img, caption=cap)
                         st.session_state.messages.append(
-                            {"role": "assistant", "content": full_resp}
-                        )
-            else:
+                            {"role": "assistant", "content": answer,
+                             "image": img, "image_caption": cap})
+                        answer_entry_saved = True
+            if not st.session_state.generate_image:
                 st.session_state.messages.append(
-                    {"role": "assistant", "content": full_resp}
-                )
+                    {"role": "assistant", "content": answer})
 
-            # 다운로드 버튼
-            st.subheader("이 블로그 다운로드:")
-            c1, c2 = st.columns(2)
-            c1.download_button("마크다운", full_resp,
+            # 본문 다운로드 버튼 (MD / HTML)
+            st.subheader("이 블로그 다운로드")
+            b1, b2 = st.columns(2)
+            b1.download_button("Markdown", answer,
                                file_name=f"{prompt[:30]}.md", mime="text/markdown")
-            html = convert_md_to_html(full_resp, prompt[:30])
-            c2.download_button("HTML", html,
+            b2.download_button("HTML", md_to_html(answer, prompt[:30]),
                                file_name=f"{prompt[:30]}.html", mime="text/html")
 
-    # 자동 저장
+    # ── 자동 백업 저장
     if st.session_state.auto_save and st.session_state.messages:
         try:
-            fname = f"chat_history_{datetime.now():%Y%m%d_%H%M%S}.json"
-            with open(fname, "w", encoding="utf-8") as f:
-                json.dump(st.session_state.messages, f, ensure_ascii=False, indent=2)
+            fn = f"chat_history_auto_{datetime.now():%Y%m%d_%H%M%S}.json"
+            with open(fn, "w", encoding="utf-8") as fp:
+                json.dump(st.session_state.messages, fp,
+                          ensure_ascii=False, indent=2)
         except Exception as e:
-            sb.error(f"자동 저장 오류: {e}")
+            logging.error(f"자동 저장 실패: {e}")
 
-# ───────────────────────────── 9) main ────────────────────────────────────────
-def main():
-    chatbot_interface()
+# ──────────────────────────────── main / requirements ──────────────────────
+def main(): ginigen_app()
 
 if __name__ == "__main__":
-    # requirements.txt 생성
+    # requirements.txt 동적 생성
     with open("requirements.txt", "w") as f:
         f.write("\n".join([
             "streamlit>=1.31.0",