import gradio as gr
from openai import OpenAI
import os

from supabase import create_client, Client

url: str = os.environ.get("SUPABASE_URL")
key: str = os.environ.get("SUPABASE_KEY")
supabase: Client = create_client(url, key)

username = os.environ.get("USERNAME")
passwd = os.environ.get("PASSWD")

client = OpenAI(
    # defaults to os.environ.get("OPENAI_API_KEY")
)

# set the context
messages = [{"role": "system", "content": "Translate the patent text from Chinese to English."}, {"role": "user", "content": None}]

# get the user inputs
def translate(text_cn):
    messages[1]["content"] = text_cn
    # messages.append({"role": "user", "content": text_cn})
    response = client.chat.completions.create(
        model="ft:gpt-3.5-turbo-1106:personal::8urfRayz",
        messages=messages,
        temperature=0.01,
        max_tokens=4096,
        top_p=1,
        frequency_penalty=0,
        presence_penalty=0
    )

    response = response.choices[0].message.content

    # write text data to database
    # supabase.table('gradio_text').insert({"text_cn": text_cn, "text_en": response}).execute()

    return response

iface = gr.Interface(fn=translate, inputs="text", outputs="text", title="😀 Patent Translate Bot", description="A simple bot for patent translation")

iface.launch(auth=(username, passwd))