update

app.py

@@ -46,7 +46,7 @@ with gr.Blocks() as demo:
     chatbot = gr.Chatbot(avatar_images=("assets/man.png", "assets/bot.png"))
     with gr.Row():
         with gr.Column(scale=4):
-            # user_input = gr.Textbox(show_label=False, placeholder="Input...", lines=10)
+            generated_text = gr.Textbox(show_label=False, placeholder="...", lines=10, visible=False)
             with gr.Row():
                 generate_btn = gr.Button("🤔️ Generate")
                 retry_btn = gr.Button("🔄  Regenerate")
@@ -86,8 +86,8 @@ with gr.Blocks() as demo:
 
     clear_btn.click(reset_state, inputs=[system], outputs=[chatbot, history], show_progress="full")
 
-    generate_btn.click(generate, [chatbot, history], outputs=[chatbot, history],
-                             show_progress="full")
+    generate_btn.click(generate, [chatbot, history], outputs=[generated_text, chatbot, history],
+                       show_progress="full")
 
     slider_max_new_tokens.change(set_max_tokens, inputs=[slider_max_new_tokens])
 

app_util.py

@@ -49,17 +49,17 @@ def generate_response(query, chatbot, history):
 
     if history[-1]["role"] != "user":
         gr.Warning('You should generate or type user-input first.')
-        yield chatbot, history
+        yield None, chatbot, history
     else:
         streamer = bot.generate_response(history, stream=True)
         for response in streamer:
             chatbot[-1] = (query, response)
-            yield chatbot, history
+            yield response, chatbot, history
 
         history.append({"role": "assistant", "content": response})
         print(f"chatbot is {chatbot}")
         print(f"history is {history}")
-        yield chatbot, history
+        yield response, chatbot, history
 
 
 def generate(chatbot, history):
@@ -68,8 +68,8 @@ def generate(chatbot, history):
     elif history[-1]["role"] in ["user", "system"]:
         return generate_query(chatbot, history)
     else:
-
-        gr.Warning("")
+        gr.Warning("bug")
+        return
 
 
 def regenerate():