app.py

import os
import streamlit as st
import json
import anthropic

# API 설정
api_key = os.environ.get("API_KEY")
client = anthropic.Anthropic(api_key=api_key)

# 최대 토큰 수 설정 (Claude-3 Sonnet의 최대 토큰 수)
MAX_TOKENS = 7999

def get_system_prompt():

    return """


## 🌟 전문 블로거 시스템 프롬프트

### ✅ 단계별 공식 프롬프트

다음 8가지 단계를 정확히 순서대로 따라 블로그를 작성하세요:

### 1. 인사와 공감으로 시작하기
- 친근한 말투로 독자에게 인사하며 글의 주제에 대한 관심 유발
- 독자의 현실적 고민이나 공감 요소를 질문 형태로 제시

### 2. 문제를 명확히 제시하기
- 독자가 겪고 있는 구체적이고 현실적인 문제를 명확히 짚어내기
- 문제의 심각성 또는 긴급성을 강조하여 관심도를 높이기

### 3. 문제 원인을 분석하여 신뢰 구축하기
- 문제 발생 원인을 논리적으로 설명
- 독자가 쉽게 이해할 수 있도록 명확하고 구체적으로 기술
- 필요한 경우 데이터, 예시, 사례 제시

### 4. 구체적 해결책을 제시하기
- 문제를 해결할 수 있는 구체적이고 실천 가능한 방법을 단계적으로 제공
- 명확한 팁, 전략, 가이드라인 형태로 제시하여 즉시 적용 가능하게 하기

### 5. 사회적 증거 제시하기
- 실제 성공 사례, 후기, 사용자 경험담, 데이터 등을 포함
- 독자가 신뢰할 수 있도록 구체적이고 사실적으로 기술

### 6. 행동 유도(CTA: Call To Action)하기
- 지금 당장 독자가 취해야 할 행동을 구체적으로 제안
- "지금 바로", "오늘부터", "즉시"와 같은 긴급한 표현을 사용하여 독자의 행동을 촉진

### 7. 제한 조건 또는 경고를 추가하여 진정성 강화하기
- 이 글이나 해결책이 모든 사람에게 맞지 않을 수 있음을 언급
- 독자에게 진정성과 희소성을 전달하여 신뢰도를 높이기

### 8. 감사 인사 및 후속 연결 안내하기
- 독자에게 읽어줘서 감사하다는 진심을 표현
- 다음 포스팅 예고 또는 댓글, 구독 요청으로 자연스러운 연결 유도

---

### 🚩 작성 스타일 팁
- 친근하고 인간적인 톤 유지
- 구어체와 질문을 자주 사용하여 독자와의 상호작용 촉진
- 명확한 구조와 구분선(---), 번호 목록, 체크리스트 사용으로 가독성 향상
- 가능하면 실제 사례나 구체적 데이터 포함

    

"""



def chatbot_interface():
    st.title("Ginigen Blog")
    
    # 모델 고정 설정
    if "ai_model" not in st.session_state:
        st.session_state["ai_model"] = "claude-3-5-sonnet-20241022"
    # 세션 상태 초기화
    if "messages" not in st.session_state:
        st.session_state.messages = []
        
    # 자동 저장 기능
    if "auto_save" not in st.session_state:
        st.session_state.auto_save = True

    # 대화 기록 관리 (사이드바)
    st.sidebar.title("대화 기록 관리")
    
    # 자동 저장 토글
    st.session_state.auto_save = st.sidebar.toggle("자동 저장", value=st.session_state.auto_save)
    
    # 대화 기록 불러오기
    uploaded_file = st.sidebar.file_uploader("대화 기록 불러오기", type=['json'])
    if uploaded_file is not None:
        try:
            content = uploaded_file.getvalue().decode()
            if content.strip():
                st.session_state.messages = json.loads(content)
                st.sidebar.success("대화 기록을 성공적으로 불러왔습니다!")
            else:
                st.sidebar.warning("업로드된 파일이 비어 있습니다.")
        except json.JSONDecodeError:
            st.sidebar.error("올바른 JSON 형식의 파일이 아닙니다.")
        except Exception as e:
            st.sidebar.error(f"파일 처리 중 오류가 발생했습니다: {str(e)}")

    # 대화 기록 초기화 버튼
    if st.sidebar.button("대화 기록 초기화"):
        st.session_state.messages = []
        st.sidebar.success("대화 기록이 초기화되었습니다.")

    # 메시지 표시
    for message in st.session_state.messages:
        with st.chat_message(message["role"]):
            st.markdown(message["content"])

    # 사용자 입력
    if prompt := st.chat_input("무엇을 도와드릴까요?"):
        st.session_state.messages.append({"role": "user", "content": prompt})
        with st.chat_message("user"):
            st.markdown(prompt)

        # AI 응답 생성
        with st.chat_message("assistant"):
            message_placeholder = st.empty()
            full_response = ""
            
            # API 호출
            with client.messages.stream(
                max_tokens=MAX_TOKENS,
                system=get_system_prompt(),
                messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
                model=st.session_state["ai_model"]
            ) as stream:
                for text in stream.text_stream:
                    full_response += str(text) if text is not None else ""
                    message_placeholder.markdown(full_response + "▌")
                    
            message_placeholder.markdown(full_response)
            
        st.session_state.messages.append({"role": "assistant", "content": full_response})

        # 자동 저장 기능
        if st.session_state.auto_save:
            try:
                with open('chat_history_auto_save.json', 'w', encoding='utf-8') as f:
                    json.dump(st.session_state.messages, f, ensure_ascii=False, indent=4)
            except Exception as e:
                st.sidebar.error(f"자동 저장 중 오류 발생: {str(e)}")

    # 대화 기록 다운로드
    if st.sidebar.button("대화 기록 다운로드"):
        json_history = json.dumps(st.session_state.messages, indent=4, ensure_ascii=False)
        st.sidebar.download_button(
            label="대화 기록 저장하기",
            data=json_history,
            file_name="chat_history.json",
            mime="application/json"
        )

def main():
    chatbot_interface()

if __name__ == "__main__":
    main()