| 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( |
| |
| ) |
|
|
| |
| messages = [{"role": "system", "content": "Translate the patent text from Chinese to English."}, {"role": "user", "content": None}] |
|
|
| |
| def translate(text_cn): |
| messages[1]["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 |
|
|
| |
| |
|
|
| 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)) |