essay_edit4 / app.py
DiningSystem's picture
Update app.py
a1978ee
raw
history blame
1.57 kB
import openai
import os
import gradio as gr
openai.api_key = os.environ['openai']
def write_essay(prompt, context, temperature, frequency_penalty, presence_penalty):
model="gpt-3.5-turbo"
temp_list = prompt.split()
temp_list2 = context.split()
temperature = temperature/10
frequency_penalty = frequency_penalty/10 - 2
presence_penalty = presence_penalty/10 -2
new_prompt = " ".join(temp_list[0:150])
new_prompt2 = " ".join(temp_list2[0:50])
response = openai.ChatCompletion.create(
model=model,
messages=[{"role": "system", "content": new_prompt2},
{"role": "user", "content": new_prompt},],
stream=False,
max_tokens=250,
temperature=temperature,
frequency_penalty=frequency_penalty,
presence_penalty=presence_penalty
)
message = response.choices[0]["message"]["content"]
#print(message)
return message
demo = gr.Interface(fn=write_essay, inputs=["text","text",
gr.Slider(0, 20, value=4, step=1, label="Creative vs Focus", info="Increase for creative and decrease for more focused"),
gr.Slider(0, 40, value=20, step=1, label="Frequency of words", info="Increase to decrease repeated words"),
gr.Slider(0, 40, value=20, step=1, label="Presence of words", info="Increase to decrease already appeared words")
], outputs="text")
demo.launch()