File size: 1,202 Bytes
0af7025
b74f28e
d51fa15
d3f2071
d51fa15
 
b74f28e
 
 
 
d51fa15
b74f28e
 
79c3236
b74f28e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
d51fa15
b74f28e
0af7025
6475b08
d51fa15
 
6475b08
 
d51fa15
0af7025
bf93ff7
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
import gradio as gr
import requests
import os 

def function(Textbox,Textbox2):
    target = os.environ.get("target")
    target2 = os.environ.get("target2")
    model = os.environ.get("model")
    hrc = os.environ.get("hrc")
    content = os.environ.get("content")
    if Textbox2 == target:
        payload = {
            "model": "gpt-3.5-turbo",
            "messages": [{"role": "system", "content": content},{"role": "user", "content": Textbox}],
            "temperature" : 1.0,
            "top_p":1.0,
            "n" : 1,
            "stream": False,
            "presence_penalty":0,
            "frequency_penalty":0,
        }
        
        headers = {
            "Content-Type": "application/json",
            "Authorization": f"Bearer {target2}"
        }
        
        response = requests.post(hrc, headers=headers, json=payload, stream=False)
        response = response.json()
        return response["choices"][0]["message"]["content"]
    else:
        return "Failed"

inputs = [
    gr.inputs.Textbox(label="Textbox",type="text"),
    gr.inputs.Textbox(label="Textbox2",type="password")
]

iface = gr.Interface(fn=function, inputs=inputs, outputs="text")

iface.launch()