import gradio as gr
import time

def echo(message, history, system_prompt, tokens):
    response = f"System prompt: {system_prompt}\n Message: {message}."
    for i in range(min(len(response), int(tokens))):
        time.sleep(0.05)
        yield response[: i+1]

with gr.Blocks(theme="citrus") as demo:
    system_prompt = gr.Textbox("You are helpful AI.", label="System Prompt")
    slider = gr.Slider(100, 500, render=False)

    gr.ChatInterface(
        echo, additional_inputs=[system_prompt, slider], type="messages", save_history=True, examples=[["hi there"], ["What is 2+2?"], ["Is 9.90 > 9.11?"]]
    )

demo.launch()