Update app.py

app.py

@@ -72,12 +72,10 @@ def generate_response(
         if system_message.strip():
             messages.append(format_message("system", system_message))
         
-        # 대화 히스토리 추가
+        # 대화 히스토리 추가 (이미 사용자 메시지가 포함되어 있음)
         formatted_history = get_formatted_history(history)
-        messages.extend(formatted_history)
         
-        # 현재 메시지와 파일 내용 준비
-        current_content = message
+        # 파일 내용이 있는 경우 마지막 사용자 메시지에 추가
         if files:
             file_contents = []
             for file in files:
@@ -85,10 +83,12 @@ def generate_response(
                 if content:
                     file_contents.append(f"파일 내용:\n{content}")
             if file_contents:
-                current_content = current_content + "\n\n" + "\n\n".join(file_contents)
+                if formatted_history and formatted_history[-1]["role"] == "user":
+                    formatted_history[-1]["content"] += "\n\n" + "\n\n".join(file_contents)
+                else:
+                    formatted_history.append(format_message("user", "\n\n".join(file_contents)))
         
-        # 현재 메시지 추가
-        messages.append(format_message("user", current_content))
+        messages.extend(formatted_history)
         
         # 디버깅: API 요청 내용 출력
         st.write("API 요청 메시지:", json.dumps(messages, ensure_ascii=False, indent=2))
@@ -145,14 +145,14 @@ def main():
             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("무엇을 알고 싶으신가요?"):
-        # 사용자 메시지 추가
+        # 사용자 메시지 추가 (한번만 추가)
         user_message = format_message("user", prompt)
         st.session_state.messages.append(user_message)
         
@@ -184,4 +184,4 @@ def main():
         st.session_state.messages.append(assistant_message)
 
 if __name__ == "__main__":
-    main()
+    main()