app.py

import gradio as gr
import openai


def get_answer_chatgpt(chat_history, message, system_message=None):
    messages = []
    if system_message:
        TEMPLATE_SYSTEM = {"role": "system", "content": system_message}
        message.append(TEMPLATE_SYSTEM)

    TEMPLATE_USER = {"role": "user"}
    TEMPLATE_BOT = {"role": "assistant"}

    # add historical conversation
    for pair in chat_history:
        TEMPLATE_USER["content"] = pair[0]
        TEMPLATE_BOT["content"] = pair[1]
        messages.append(TEMPLATE_USER)
        messages.append(TEMPLATE_BOT)

    # add latest input
    TEMPLATE_USER["content"] = message
    messages.append(TEMPLATE_USER)
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages
    )

    return {
        "completion": response["choices"][0]["message"]["content"],
        "usage_prompt_token": response["usage"]["prompt_tokens"],
        "usage_completion_token": response["usage"]["completion_tokens"]
    }


def reset_textbox():
    return gr.update(value='')


with gr.Blocks() as demo:
    gr.Markdown() 
    system_prompt = gr.Textbox(placeholder="You are helpful assistence", lines=5, label="System Message")
    chatbot = gr.Chatbot()
    message = gr.Textbox(placeholder="Halo kak, aku mau bertanya", label="Chat Here")
    with gr.Row():
        with gr.Column(scale=5):
            clear = gr.Button("Clear")
        with gr.Column(scale=5):
            retry = gr.Button("Retry")
    send = gr.Button("Send")

    def respond(chat_history, message):
        response = get_answer_chatgpt(chat_history, message)
        response = response["completion"]

        return chat_history + [[message, response]]

    message.submit(respond, [chatbot, message], chatbot)

    clear.click(lambda: None, None, chatbot, queue=False)
    send.click(respond, [chatbot, message], chatbot)
    send.click(reset_textbox, [], [message])
    message.submit(reset_textbox, [], [message])

    retry.click(respond, [chatbot, message], chatbot)
    retry.click(reset_textbox, [], [message])

    demo.launch(debug=True)