add retry button

app.py

@@ -115,6 +115,14 @@ def predict(question, history=[], behavior=[]):
     return "", history, response
 
 
+def retry(question, history=[], behavior=[]):
+    if len(history)<2:
+        return "", history, []
+    question = history[-2]
+    history = history[:-2]
+    return predict(question, history, behavior)
+
+
 with gr.Blocks() as demo:
     
     examples_txt = [
@@ -159,12 +167,14 @@ with gr.Blocks() as demo:
         txt = gr.Textbox(show_label=False, placeholder="输入你想让ChatGPT回答的问题").style(container=False)
         with gr.Row():
             button_gen = gr.Button("Submit")
+            button_rtr = gr.Button("Retry")
             button_clr = gr.Button("Clear")
         
     gr.Examples(examples=examples_bhv, inputs=bhv, label="Examples for setting behavior")
     gr.Examples(examples=examples_txt, inputs=txt, label="Examples for asking question")
     txt.submit(predict, [txt, state, behavior], [txt, state, chatbot])
     button_gen.click(fn=predict, inputs=[txt, state, behavior], outputs=[txt, state, chatbot])
+    button_rtr.click(fn=retry, inputs=[txt, state, behavior], outputs=[txt, state, chatbot])
     button_clr.click(fn=lambda :([],[]), inputs=None, outputs=[chatbot, state])
 
 demo.launch()