Update app.py

app.py

@@ -150,11 +150,11 @@ def chat_with_model(prompt, system_prompt, chatbot_display, internal_history,lan
     internal_history.append({"role": "assistant", "content": response_text})
     
     # 7. Cập nhật lịch sử hiển thị (gr.Chatbot)
-    chatbot_display.append([prompt, translated,response_text])
+    chatbot_display.append([prompt, translated])
 
     # 8. Trả về cả hai để cập nhật UI
     # (chuỗi rỗng "" để xóa nội dung trong ô prompt_box)
-    return "", chatbot_display, internal_history
+    return "", chatbot_display, internal_history, response_text, prompt
 
 def clear_chat():
     """Xóa lịch sử."""
@@ -187,6 +187,18 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
                 placeholder="Nhập tin nhắn của bạn và nhấn Enter...",
                 lines=1
             )
+
+            prompt = gr.Textbox(
+                label="Your Message",
+                placeholder="Nhập tin nhắn của bạn và nhấn Enter...",
+                lines=1
+            )
+
+            response = gr.Textbox(
+                label="Your Message",
+                placeholder="Nhập tin nhắn của bạn và nhấn Enter...",
+                lines=1
+            )           
             
             with gr.Row():
                 clear_button = gr.Button("Clear Chat")
@@ -207,14 +219,14 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
     prompt_box.submit(
         fn=chat_with_model,
         inputs=[prompt_box, system_prompt_box, chatbot_display, internal_history,lang],
-        outputs=[prompt_box, chatbot_display, internal_history]
+        outputs=[prompt_box, chatbot_display, internal_history, response, prompt]
     )
 
     # Khi người dùng nhấn nút "Send" (ẩn)
     submit_button.click(
         fn=chat_with_model,
         inputs=[prompt_box, system_prompt_box, chatbot_display, internal_history,lang],
-        outputs=[prompt_box, chatbot_display, internal_history]
+        outputs=[prompt_box, chatbot_display, internal_history, response, prompt]
     )
     
     # Khi người dùng nhấn nút "Clear Chat"