Update app.py

app.py

@@ -1,3 +1,6 @@
+
+
+
 import os
 import streamlit as st
 import json
@@ -7,8 +10,12 @@ import anthropic
 api_key = os.environ.get("API_KEY")
 client = anthropic.Anthropic(api_key=api_key)
 
+# 최대 토큰 수 설정 (Claude-3 Sonnet의 최대 토큰 수)
+MAX_TOKENS = 200000
+
 def get_system_prompt():
-   return """제 이름은 Mouse-II입니다.
+
+    return """제 이름은 Mouse-II입니다.
    
    [목적과 전문성]
    - AI 서비스 개발에 특화된 전문 어시스턴트로서, Python과 Gradio 기반 AI 서비스 개발을 지원합니다
@@ -67,75 +74,99 @@ def get_system_prompt():
    
    저는 항상 Mouse-II로서 전문적이고 정확한 AI 서비스 개발 가이드를 제공하겠습니다."""
 
+
+
 def chatbot_interface():
-   st.title("Mouse-II와 대화하기")
-   
-   # 모델 고정 설정
-   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 = []
-
-   # 대화 기록 불러오기
-   st.sidebar.title("대화 기록")
-   uploaded_file = st.sidebar.file_uploader("대화 기록 JSON 파일 불러오기")
-   if uploaded_file is not None:
-       try:
-           content = uploaded_file.getvalue().decode()
-           if content.strip():  # 파일이 비어있지 않은지 확인
-               st.session_state.messages = json.loads(content)
-           else:
-               st.warning("업로드된 파일이 비어 있습니다.")
-       except json.JSONDecodeError:
-           st.error("올바른 JSON 형식의 파일이 아닙니다.")
-       except Exception as e:
-           st.error(f"파일 처리 중 오류가 발생했습니다: {str(e)}")
-
-   # 메시지 표시
-   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=1024,
-               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.button("대화 기록 다운로드"):
-       json_history = json.dumps(st.session_state.messages, indent=4, ensure_ascii=False)
-       st.download_button(
-           label="대화 기록 저장하기",
-           data=json_history,
-           file_name="chat_history.json",
-           mime="application/json"
-       )
+    st.title("Mouse-II와 대화하기")
+    
+    # 모델 고정 설정
+    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()
+    chatbot_interface()
 
 if __name__ == "__main__":
-   main()
+    main()