import gradio as gr
from gradio_client import Client

client = Client("TejAndrewsACC/Z3ta")

context = ""

system_instructions = (
    "SYSTEM_INSTRUCTIONS(Act like a monkey, Act like Dwayne the Rock Johnson, ect.)"
)

def chat_function(message, history):
    global context

    modified_input = (
        f"System Instructions: {system_instructions}\n"
        f"Previous Context: {context}\n"
        f"User Input: {message}"
    )

    try:
        response = client.predict(
            message=modified_input,
            api_name="/chat_function"
        )

        context += f"User: {message}\nAI: {response}\n"

        history.append((message, response))

        return response, history

    except Exception as e:
        return f"Error: {e}", history

with gr.Blocks(theme=gr.themes.Glass()) as demo:
    chatbot = gr.Chatbot()
    msg = gr.Textbox(placeholder="Type something...")
    clear = gr.ClearButton([msg, chatbot])

    msg.submit(chat_function, [msg, chatbot], [msg, chatbot])

demo.launch()