Update app.py

app.py

@@ -5,6 +5,7 @@ from typing import Iterator
 from PIL import Image
 import base64
 from PyPDF2 import PdfReader
+import json  # 디버깅용 추가
 
 API_KEY = os.getenv("TOGETHER_API_KEY")
 if not API_KEY:
@@ -33,9 +34,28 @@ def process_file(file) -> str:
         st.error(f"파일 처리 중 오류 발생: {str(e)}")
         return ""
 
+def format_message(role: str, content: str) -> dict:
+    """API 메시지 형식에 맞게 메시지를 포맷팅합니다."""
+    return {
+        "role": role,
+        "content": content
+    }
+
+def get_formatted_history(messages: list) -> list:
+    """대화 히스토리를 API 형식에 맞게 변환합니다."""
+    formatted_messages = []
+    for msg in messages:
+        if isinstance(msg, dict) and "role" in msg and "content" in msg:
+            # 역할이 올바른지 확인하고 수정
+            role = msg["role"]
+            if role not in ["system", "user", "assistant"]:
+                role = "user" if role == "human" else "assistant"
+            formatted_messages.append(format_message(role, msg["content"]))
+    return formatted_messages
+
 def generate_response(
     message: str,
-    history: list[dict],  # 히스토리 형식 변경
+    history: list,
     system_message: str,
     max_tokens: int,
     temperature: float,
@@ -50,13 +70,11 @@ def generate_response(
         
         # 시스템 메시지 추가
         if system_message.strip():
-            messages.append({
-                "role": "system",
-                "content": system_message
-            })
+            messages.append(format_message("system", system_message))
         
-        # 대화 히스토리 추가 - 형식 수정
-        messages.extend(history)
+        # 대화 히스토리 추가
+        formatted_history = get_formatted_history(history)
+        messages.extend(formatted_history)
         
         # 현재 메시지와 파일 내용 준비
         current_content = message
@@ -70,10 +88,10 @@ def generate_response(
                 current_content = current_content + "\n\n" + "\n\n".join(file_contents)
         
         # 현재 메시지 추가
-        messages.append({
-            "role": "user",
-            "content": current_content
-        })
+        messages.append(format_message("user", current_content))
+        
+        # 디버깅: API 요청 내용 출력
+        st.write("API 요청 메시지:", json.dumps(messages, ensure_ascii=False, indent=2))
         
         # API 요청
         try:
@@ -94,11 +112,11 @@ def generate_response(
             if "rate limit" in str(e).lower():
                 yield "API 호출 한도에 도달했습니다. 잠시 후 다시 시도해주세요."
             else:
-                st.error(f"API 오류 상세: {str(e)}")  # 디버깅을 위한 오류 출력
+                st.error(f"API 오류 상세: {str(e)}")
                 yield "죄송합니다. 잠시 후 다시 시도해주세요."
                 
     except Exception as e:
-        st.error(f"전체 오류 상세: {str(e)}")  # 디버깅을 위한 오류 출력
+        st.error(f"전체 오류 상세: {str(e)}")
         yield "오류가 발생했습니다. 잠시 후 다시 시도해주세요."
 
 def main():
@@ -107,9 +125,6 @@ def main():
     # 세션 상태 초기화
     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 모델과 대화하세요. 필요한 경우 파일을 업로드할 수 있습니다.")
@@ -138,8 +153,8 @@ def main():
     # 채팅 입력
     if prompt := st.chat_input("무엇을 알고 싶으신가요?"):
         # 사용자 메시지 추가
-        st.session_state.messages.append({"role": "user", "content": prompt})
-        st.session_state.conversation_history.append({"role": "user", "content": prompt})
+        user_message = format_message("user", prompt)
+        st.session_state.messages.append(user_message)
         
         with st.chat_message("user"):
             st.markdown(prompt)
@@ -152,7 +167,7 @@ def main():
             # generate_response 호출
             for response_chunk in generate_response(
                 prompt,
-                st.session_state.conversation_history,  # 수정된 히스토리 전달
+                st.session_state.messages,
                 system_message,
                 max_tokens,
                 temperature,
@@ -165,8 +180,8 @@ 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})
+        assistant_message = format_message("assistant", full_response)
+        st.session_state.messages.append(assistant_message)
 
 if __name__ == "__main__":
     main()