rag / app.py
shao-shuai
remove db
b0e8ea2
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))