Update app.py

app.py

@@ -35,7 +35,7 @@ def process_file(file) -> str:
 
 def generate_response(
     message: str,
-    history: list[tuple[str, str]],
+    history: list[dict],  # 히스토리 형식 변경
     system_message: str,
     max_tokens: int,
     temperature: float,
@@ -48,23 +48,15 @@ def generate_response(
         # 메시지 배열 초기화
         messages = []
         
-        # 시스템 메시지가 있는 경우에만 추가
+        # 시스템 메시지 추가
         if system_message.strip():
             messages.append({
                 "role": "system",
                 "content": system_message
             })
         
-        # 대화 히스토리 추가
-        for user_msg, assistant_msg in history:
-            messages.append({
-                "role": "user",
-                "content": user_msg
-            })
-            messages.append({
-                "role": "assistant",
-                "content": assistant_msg
-            })
+        # 대화 히스토리 추가 - 형식 수정
+        messages.extend(history)
         
         # 현재 메시지와 파일 내용 준비
         current_content = message
@@ -83,19 +75,16 @@ def generate_response(
             "content": current_content
         })
         
-        # API 요청 설정
-        request_params = {
-            "model": "deepseek-ai/DeepSeek-R1",
-            "messages": messages,
-            "max_tokens": max_tokens,
-            "temperature": temperature,
-            "top_p": top_p,
-            "stream": True
-        }
-        
-        # API 호출
+        # API 요청
         try:
-            stream = client.chat.completions.create(**request_params)
+            stream = client.chat.completions.create(
+                model="deepseek-ai/DeepSeek-R1",
+                messages=messages,
+                max_tokens=max_tokens,
+                temperature=temperature,
+                top_p=top_p,
+                stream=True
+            )
             
             for chunk in stream:
                 if hasattr(chunk.choices[0].delta, 'content') and chunk.choices[0].delta.content:
@@ -105,21 +94,22 @@ def generate_response(
             if "rate limit" in str(e).lower():
                 yield "API 호출 한도에 도달했습니다. 잠시 후 다시 시도해주세요."
             else:
-                error_message = str(e)
-                # Together.ai의 오류 응답 분석
-                if "Input validation error" in error_message:
-                    yield "입력 형식이 올바르지 않습니다. 시스템 관리자에게 문의해주세요."
-                else:
-                    yield f"API 호출 중 오류가 발생했습니다: {error_message}"
+                st.error(f"API 오류 상세: {str(e)}")  # 디버깅을 위한 오류 출력
+                yield "죄송합니다. 잠시 후 다시 시도해주세요."
                 
     except Exception as e:
-        yield f"오류가 발생했습니다: {str(e)}"
+        st.error(f"전체 오류 상세: {str(e)}")  # 디버깅을 위한 오류 출력
+        yield "오류가 발생했습니다. 잠시 후 다시 시도해주세요."
 
 def main():
     st.set_page_config(page_title="DeepSeek 채팅", page_icon="💭", layout="wide")
     
+    # 세션 상태 초기화
     if "messages" not in st.session_state:
         st.session_state.messages = []
+    
+    if "conversation_history" not in st.session_state:  # 새로운 대화 히스토리 저장소
+        st.session_state.conversation_history = []
 
     st.title("DeepSeek 채팅")
     st.markdown("DeepSeek AI 모델과 대화하세요. 필요한 경우 파일을 업로드할 수 있습니다.")
@@ -131,34 +121,38 @@ def main():
             value="당신은 깊이 있게 생각하는 AI입니다. 문제를 깊이 고려하고 체계적인 추론 과정을 통해 올바른 해결책을 도출하세요. 반드시 한글로 답변하세요.",
             height=100
         )
-        max_tokens = st.slider("최대 토큰 수", 1, 4096, 2048)  # 토큰 제한 조정
-        temperature = st.slider("온도", 0.0, 2.0, 0.7, 0.1)    # 온도 범위 조정
-        top_p = st.slider("Top-p", 0.0, 1.0, 0.7, 0.1)        # top_p 범위 조정
+        max_tokens = st.slider("최대 토큰 수", 1, 4096, 2048)
+        temperature = st.slider("온도", 0.0, 2.0, 0.7, 0.1)
+        top_p = st.slider("Top-p", 0.0, 1.0, 0.7, 0.1)
         uploaded_file = st.file_uploader(
             "파일 업로드 (선택사항)",
             type=['txt', 'py', 'md', 'pdf', 'png', 'jpg', 'jpeg'],
             accept_multiple_files=True
         )
 
+    # 메시지 표시
     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})
+        st.session_state.conversation_history.append({"role": "user", "content": prompt})
+        
         with st.chat_message("user"):
             st.markdown(prompt)
 
+        # 어시스턴트 응답 생성
         with st.chat_message("assistant"):
             response_placeholder = st.empty()
             full_response = ""
             
-            history = [(msg["content"], next_msg["content"]) 
-                      for msg, next_msg in zip(st.session_state.messages[::2], st.session_state.messages[1::2])]
-            
+            # generate_response 호출
             for response_chunk in generate_response(
                 prompt,
-                history,
+                st.session_state.conversation_history,  # 수정된 히스토리 전달
                 system_message,
                 max_tokens,
                 temperature,
@@ -170,7 +164,9 @@ def main():
             
             response_placeholder.markdown(full_response)
         
+        # 응답 저장
         st.session_state.messages.append({"role": "assistant", "content": full_response})
+        st.session_state.conversation_history.append({"role": "assistant", "content": full_response})
 
 if __name__ == "__main__":
     main()