Update app.py

app.py

@@ -71,26 +71,48 @@ def render_message(history):
     return messages_html
 
 with gr.Blocks() as demo:
-    chatbot_output = gr.HTML()  
-    msg_input = gr.Textbox(show_label=False, placeholder="Type your message here...")
+    gr.Markdown("## P-MSQ Chat Interface with Profile Pictures")  
+    gr.Markdown("""
+    Welcome to the **P-MSQ** (Messaging Service Query) chat interface! 
+    You are interacting with a friendly AI chatbot that can assist you in various situations.
+    Use the text input box below to start chatting.
+    """)
+
+    with gr.Box():  
+        chatbot_output = gr.HTML()  
+
+    with gr.Row():
+        msg_input = gr.Textbox(show_label=False, placeholder="Type your message here...", lines=2)
+        send_btn = gr.Button("Send")
+        regen_btn = gr.Button("Regenerate")
 
     system_message = gr.Textbox(value="You are P-MSQ (Messaging Service Query), a friendly AI Chatbot that can help in any situations.", label="System message")
+    
+    gr.Markdown("### Settings")  
     max_tokens = gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max new tokens")
     top_p = gr.Slider(minimum=0, maximum=2, value=0.8, step=0.1, label="Top P")
     temperature = gr.Slider(minimum=0.1, maximum=1, value=0.7, step=0.1, label="Temperature")
 
     history_state = gr.State([])  
+    last_message_state = gr.State("")  
+
     def user_interaction(message, history, system_message, max_tokens, top_p, temperature):
         history, assistant_reply = respond(message, history, system_message, max_tokens, top_p, temperature)
+        return render_message(history), history, "", message  
+
+    def regenerate_response(history, last_message, system_message, max_tokens, top_p, temperature):
+        if last_message:
+            history, assistant_reply = respond(last_message, history, system_message, max_tokens, top_p, temperature)
+            return render_message(history), history
         return render_message(history), history  
 
-    msg_input.submit(user_interaction, 
-                     inputs=[msg_input, history_state, system_message, max_tokens, top_p, temperature], 
-                     outputs=[chatbot_output, history_state])
+    send_btn.click(user_interaction, 
+                   inputs=[msg_input, history_state, system_message, max_tokens, top_p, temperature], 
+                   outputs=[chatbot_output, history_state, msg_input, last_message_state])
 
-    gr.Markdown("## Chat Interface with Profile Pictures")
-    chatbot_output
-    msg_input
+    regen_btn.click(regenerate_response, 
+                    inputs=[history_state, last_message_state, system_message, max_tokens, top_p, temperature], 
+                    outputs=[chatbot_output, history_state])
 
 if __name__ == "__main__":
     demo.launch()