From 01ac065d7e519ba9521c9feca66bedfc597e4a64 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Wed, 11 Jan 2023 01:33:57 -0300
Subject: [PATCH] Implement Continue button

---
 server.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/server.py b/server.py
index 1c5782a..28ab261 100644
--- a/server.py
+++ b/server.py
@@ -189,7 +189,7 @@ if args.notebook:
 elif args.chat:
     history = []
 
-    def chatbot(text, temperature, max_length, inference_settings, selected_model, name1, name2, context):
+    def chatbot_wrapper(text, temperature, max_length, inference_settings, selected_model, name1, name2, context):
         question = context+'\n\n'
         for i in range(len(history)):
             question += f"{name1}: {history[i][0][3:-5].strip()}\n"
@@ -208,7 +208,6 @@ elif args.chat:
 
     with gr.Blocks(css=css+".h-\[40vh\] {height: 50vh}", analytics_enabled=False) as interface:
         gr.Markdown(description)
-
         with gr.Row():
             with gr.Column():
                 with gr.Row():
@@ -227,16 +226,20 @@ elif args.chat:
                 btn = gr.Button("Generate")
                 btn2 = gr.Button("Clear history")
 
-        btn.click(chatbot, [textbox, temp_slider, length_slider, preset_menu, model_menu, name1, name2, context], display1, show_progress=True)
-        textbox.submit(chatbot, [textbox, temp_slider, length_slider, preset_menu, model_menu, name1, name2, context], display1, show_progress=True)
+        btn.click(chatbot_wrapper, [textbox, temp_slider, length_slider, preset_menu, model_menu, name1, name2, context], display1, show_progress=True)
+        textbox.submit(chatbot_wrapper, [textbox, temp_slider, length_slider, preset_menu, model_menu, name1, name2, context], display1, show_progress=True)
         btn2.click(clear)
         btn.click(lambda x: "", textbox, textbox, show_progress=False)
         textbox.submit(lambda x: "", textbox, textbox, show_progress=False)
         btn2.click(lambda x: "", display1, display1)
 else:
+
+    def continue_wrapper(question, temperature, max_length, inference_settings, selected_model):
+        a, b, c = generate_reply(question, temperature, max_length, inference_settings, selected_model)
+        return a, a, b, c
+
     with gr.Blocks(css=css, analytics_enabled=False) as interface:
         gr.Markdown(description)
-
         with gr.Row():
             with gr.Column():
                 textbox = gr.Textbox(value=default_text, lines=15, label='Input')
@@ -245,6 +248,7 @@ else:
                 preset_menu = gr.Dropdown(choices=available_presets, value="NovelAI-Sphinx Moth", label='Preset')
                 model_menu = gr.Dropdown(choices=available_models, value=model_name, label='Model')
                 btn = gr.Button("Generate")
+                cont = gr.Button("Continue")
             with gr.Column():
                 with gr.Tab('Raw'):
                     output_textbox = gr.Textbox(value=default_text, lines=15, label='Output')
@@ -254,6 +258,7 @@ else:
                     html = gr.HTML()
 
         btn.click(generate_reply, [textbox, temp_slider, length_slider, preset_menu, model_menu], [output_textbox, markdown, html], show_progress=True)
+        cont.click(continue_wrapper, [output_textbox, temp_slider, length_slider, preset_menu, model_menu], [output_textbox, textbox, markdown, html], show_progress=True)
         textbox.submit(generate_reply, [textbox, temp_slider, length_slider, preset_menu, model_menu], [output_textbox, markdown, html], show_progress=True)
 
 if args.no_listen: