Update app.py

app.py

@@ -73,6 +73,14 @@ def stop_generation():
     stop_event.set()
     return "생성이 중단되었습니다."
 
+def regenerate(history, system_message, max_tokens, temperature, top_p, model):
+    if not history:
+        return "", []
+    last_user_message = history[-1][0]
+    new_history = history[:-1]
+    for _, updated_history in respond(last_user_message, new_history, system_message, max_tokens, temperature, top_p, model):
+        yield "", updated_history
+
 models = {
     "deepseek-ai/DeepSeek-Coder-V2-Instruct": "(한국회사)DeepSeek-Coder-V2-Instruct",
     "meta-llama/Meta-Llama-3.1-8B-Instruct": "Meta-Llama-3.1-8B-Instruct",
@@ -96,7 +104,7 @@ with gr.Blocks() as demo:
         continue_btn = gr.Button("계속 작성", scale=1)
 
     with gr.Row():
-        regenerate = gr.Button("🔄 재생성")
+        regenerate_btn = gr.Button("🔄 재생성")
         stop = gr.Button("🛑 생성 중단")
         clear = gr.Button("🗑️ 대화 내역 지우기")
 
@@ -112,16 +120,15 @@ with gr.Blocks() as demo:
         model = gr.Radio(list(models.keys()), value=list(models.keys())[0], label="언어 모델 선택", info="사용할 언어 모델을 선택하세요")
 
     msg.javascript = """
-    (x) => {
-        const textbox = document.querySelector("#component-3");
+    function(x) {
+        const textbox = document.getElementById('component-3');
+        if (!textbox) return x;
+        
         textbox.addEventListener("keydown", function(e) {
             if (e.key === 'Enter' && !e.shiftKey) {
                 e.preventDefault();
-                return;
-            }
-            if (e.key === 'Enter' && e.shiftKey) {
-                e.preventDefault();
-                document.querySelector("#component-5").click();
+                const sendButton = document.getElementById('component-5');
+                if (sendButton) sendButton.click();
             }
         });
         return x;
@@ -131,11 +138,11 @@ with gr.Blocks() as demo:
     send.click(respond, inputs=[msg, chatbot, system_message, max_tokens, temperature, top_p, model], outputs=[msg, chatbot])
     msg.submit(respond, inputs=[msg, chatbot, system_message, max_tokens, temperature, top_p, model], outputs=[msg, chatbot])
     continue_btn.click(continue_writing, inputs=[chatbot, system_message, max_tokens, temperature, top_p, model], outputs=[msg, chatbot])
-    regenerate.click(lambda h, s, m, t, p, mod: respond(h[-1][0] if h else "", h[:-1], s, m, t, p, mod), inputs=[chatbot, system_message, max_tokens, temperature, top_p, model], outputs=[msg, chatbot])
+    regenerate_btn.click(regenerate, inputs=[chatbot, system_message, max_tokens, temperature, top_p, model], outputs=[msg, chatbot])
     stop.click(stop_generation, inputs=[], outputs=[msg])
     clear.click(lambda: (None, None), outputs=[msg, chatbot])
 
 if __name__ == "__main__":
     if not hf_token:
         print("경고: HF_TOKEN 환경 변수가 설정되지 않았습니다. 일부 모델에 접근할 수 없을 수 있습니다.")
-    demo.launch(share=True)
+    demo.launch()