Update app.py

app.py

@@ -345,7 +345,10 @@ def create_interface():
 
 def chat_interface(prompt, history, agent_name, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty):
     generated_text = generate(prompt, history, agent_name, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty)
-    return generated_text, ""  # Return the chatbot response and an empty string for the message textbox
+    chatbot_response = []
+    for response in generated_text:
+        chatbot_response.append([prompt, response])
+    return chatbot_response, ""  # Return the chatbot response and an empty string for the message textbox
 
 iface = gr.Blocks()
 
@@ -361,7 +364,7 @@ with iface:
     temperature = gr.Slider(label="Temperature", value=0.9, minimum=0.0, maximum=1.0, step=0.05)
     max_new_tokens = gr.Slider(label="Max new tokens", value=1048 * 10, minimum=0, maximum=1048 * 10, step=64)
     top_p = gr.Slider(label="Top-p (nucleus sampling)", value=0.90, minimum=0.0, maximum=1, step=0.05)
-    repetition_penalty = gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05)
+    repetition_penalty = gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum 2.0, step=0.05)
 
     msg.submit(chat_interface, 
                [msg, chatbot, agent_dropdown, sys_prompt, temperature, max_new_tokens, top_p, repetition_penalty],