Update app.py

app.py

@@ -51,17 +51,18 @@ def respond(message, history, system_message, max_tokens, temperature, top_p, se
     except Exception as e:
         yield truncated_history + [(message, f"오류 발생: {str(e)}")]
 
-def continue_writing(history, system_message, max_tokens, temperature, top_p, model):
+def continue_writing(message, history, system_message, max_tokens, temperature, top_p, selected_model):
     if not history:
-        return [("시스템", "대화 내역이 없습니다.")]
-    
+        yield [("시스템", "대화 내역이 없습니다.")]
+        return
+
     truncated_history = truncate_history(history)
-    last_user_message = truncated_history[-1][0]
     last_assistant_message = truncated_history[-1][1]
     
     prompt = f"이전 대화를 간단히 요약하고 이어서 작성해주세요. 마지막 응답: {last_assistant_message[:100]}..."
     
-    return respond(prompt, truncated_history[:-1], system_message, max_tokens, temperature, top_p, model)
+    async for response in respond(prompt, truncated_history[:-1], system_message, max_tokens, temperature, top_p, selected_model):
+        yield response
 
 def stop_generation():
     stop_event.set()