Update app.py

app.py

@@ -28,15 +28,27 @@ def user(message, history):
     history.append([message, ""])
     return "", history
 
+def regenerate(_chatbot, _task_history):
+    if not _task_history:
+        yield _chatbot
+        return
+    item = _task_history.pop(-1)
+    _chatbot.pop(-1)
+    yield from make_prediction(item[0])  # Assuming make_prediction is the function you want to use for regeneration
+
+
 def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
     history = history or []
     
+    # Use BASE_SYSTEM_MESSAGE if system_message is empty
+    system_message_to_use = system_message if system_message.strip() else BASE_SYSTEM_MESSAGE
+    
     # A última mensagem do usuário
     user_prompt = history[-1][0] if history else ""
     
     # Preparar a entrada para o modelo
     prompt_template = f'''system
-{system_message.strip()}
+{system_message_to_use.strip()}
 user
 {user_prompt}
 assistant
@@ -96,6 +108,7 @@ with gr.Blocks(css=CSS) as demo:
         submit = gr.Button(value="Send message", variant="secondary", scale=1)
         clear = gr.Button(value="New topic", variant="secondary", scale=0)
         stop = gr.Button(value="Stop", variant="secondary", scale=0)
+        regen_btn = gr.Button(value="Regenerate", variant="secondary", scale=0)
     with gr.Accordion("Show Model Parameters", open=False):
         with gr.Row():
             with gr.Column():
@@ -113,11 +126,22 @@ with gr.Blocks(css=CSS) as demo:
     clear.click(lambda: None, None, chatbot, queue=False)
 
     submit_click_event = submit.click(
-        fn=user, inputs=[message, chat_history_state], outputs=[message, chat_history_state], queue=True
+    fn=user, inputs=[message, chat_history_state], outputs=[message, chat_history_state], queue=True
     ).then(
         fn=chat, inputs=[chat_history_state, system_msg, max_tokens, temperature, top_p, top_k, repetition_penalty], outputs=[chatbot, chat_history_state, message], queue=True
     )
+    
+    # Corrected the clear button click event
+    clear.click(
+        fn=clear_chat, inputs=[chat_history_state, message], outputs=[chat_history_state, message], queue=False
+    )
+    
+    # Stop button remains the same
     stop.click(fn=None, inputs=None, outputs=None, cancels=[submit_click_event], queue=False)
 
+    regen_btn.click(
+        fn=regenerate, inputs=[chatbot, chat_history_state], outputs=[chatbot], queue=True
+    )
+
 demo.queue(max_size=128, concurrency_count=2)
 demo.launch()