Update app.py

app.py

@@ -104,8 +104,7 @@ with gr.Blocks(css='style.css') as demo:
         retry_button = gr.Button('🔄  Retry', variant='secondary')
         undo_button = gr.Button('↩️ Undo', variant='secondary')
         clear_button = gr.Button('🗑️  Clear', variant='secondary')
-    
-    gr.Markdown(LICENSE)
+
     saved_input = gr.State()
 
     with gr.Accordion(label='Advanced options', open=False):
@@ -124,7 +123,7 @@ with gr.Blocks(css='style.css') as demo:
             minimum=0.1,
             maximum=4.0,
             step=0.1,
-            value=0.1,
+            value=0.2,
         )
         top_p = gr.Slider(
             label='Top-p (nucleus sampling)',
@@ -141,6 +140,7 @@ with gr.Blocks(css='style.css') as demo:
             value=10,
         )
 
+    gr.Markdown(LICENSE)
 
     textbox.submit(
         fn=clear_and_save_textbox,
@@ -205,4 +205,53 @@ with gr.Blocks(css='style.css') as demo:
         outputs=chatbot,
         api_name=False,
     )
+
+    retry_button.click(
+        fn=delete_prev_fn,
+        inputs=chatbot,
+        outputs=[chatbot, saved_input],
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=display_input,
+        inputs=[saved_input, chatbot],
+        outputs=chatbot,
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=generate,
+        inputs=[
+            saved_input,
+            chatbot,
+            system_prompt,
+            max_new_tokens,
+            temperature,
+            top_p,
+            top_k,
+        ],
+        outputs=chatbot,
+        api_name=False,
+    )
+
+    undo_button.click(
+        fn=delete_prev_fn,
+        inputs=chatbot,
+        outputs=[chatbot, saved_input],
+        api_name=False,
+        queue=False,
+    ).then(
+        fn=lambda x: x,
+        inputs=[saved_input],
+        outputs=textbox,
+        api_name=False,
+        queue=False,
+    )
+
+    clear_button.click(
+        fn=lambda: ([], ''),
+        outputs=[chatbot, saved_input],
+        queue=False,
+        api_name=False,
+    )
+
 demo.queue(max_size=20).launch()