Update app.py

app.py

@@ -1049,10 +1049,10 @@ def continue_writing(history, system_message, max_tokens, temperature, top_p):
     5. 필요한 경우 추가적인 이미지 프롬프트 생성
     """
 
+    # 메시지 구조 수정
     messages = [
         {"role": "system", "content": system_message},
-        {"role": "assistant", "content": last_response},
-        {"role": "user", "content": continue_prompt}
+        {"role": "user", "content": continue_prompt}  # 사용자 메시지로 시작
     ]
 
     try:
@@ -1069,13 +1069,16 @@ def continue_writing(history, system_message, max_tokens, temperature, top_p):
                 token = message.choices[0].delta.content
                 if token:
                     full_response += token
-                    history.append(("계속 작성", full_response))
-                    yield history
+                    # 이전 대화 기록을 유지하면서 새로운 응답 추가
+                    new_history = history.copy()
+                    new_history.append(("계속 작성", full_response))
+                    yield new_history
 
     except Exception as e:
         error_message = f"계속 작성 중 오류가 발생했습니다: {str(e)}"
-        history.append(("오류", error_message))
-        yield history
+        new_history = history.copy()
+        new_history.append(("오류", error_message))
+        yield new_history
 
     return history