import gradio as gr
import os
import time
from google.generativeai import palm

palm.configure(api_key=os.environ.get("palm_key"))

defaults = {
    'model': 'models/chat-bison-001',
    'temperature': 0.25,
    'candidate_count': 1,
    'top_k': 40,
    'top_p': 0.95,
}

context = "You're a computer failure assistant"

examples = [
    [
        "Hey my computer is broken",
        "Hey, what is the issue with your computer?"
    ]
]

messages = ["NEXT REQUEST"]

response = palm.chat(
    **defaults,
    context=context,
    examples=examples,
    messages=messages
)


def respond(msg):
    messages.append(msg)
    bot_message = palm.chat(
        **defaults,
        context=context,
        examples=examples,
        messages=messages
    )
    time.sleep(2)
    return bot_message['message'], messages


iface = gr.Interface(
    fn=respond,
    inputs="text",
    outputs="text",
    title="Chatbot",
    description="Interact with a chatbot."
)

iface.launch()