File size: 1,324 Bytes
6b26269
33eb293
2d0b7de
6b26269
e5fab6f
 
 
 
 
 
b10ae43
 
 
33eb293
 
 
 
 
267a387
33eb293
 
 
e306d43
 
33eb293
6cf9502
e306d43
33eb293
 
 
 
6e716bf
33eb293
 
 
e5fab6f
 
b0e8ea2
e5fab6f
33eb293
 
281ffc8
e5fab6f
b10ae43
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
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))