app.py

import os
import openai
import gradio as gr
from dotenv import load_dotenv

load_dotenv()
openai.api_key = os.getenv('OPENAI_KEY')
LIMITE_COPYS = 5
usage_state = {}


def openai_completion(prompt, request: gr.Request):
    ipaddress = request.client.host

    # if ipaddress in usage_state.keys() and usage_state[ipaddress] >= LIMITE_COPYS:
    #     return {output: "", total_tries_lbl: "Limite atingido"}
    #
    # if ipaddress not in usage_state.keys():
    #     usage_state[ipaddress] = 1
    # else:
    #     usage_state[ipaddress] += 1

    prompt = 'Crie um texto promocional, em português brasileiro, para ' + prompt
    response = openai.Completion.create(
        model='text-davinci-002',
        # model="text-ada-001",
        prompt=prompt,
        temperature=0.7,
        max_tokens=256,
        top_p=1,
        frequency_penalty=0,
        presence_penalty=0
    )

    return {output: response['choices'][0]['text'].strip()}


with gr.Blocks() as demo:
    prompt = gr.Textbox(label='Crie um texto promocional para...')
    output = gr.Textbox(label="Copy gerado")
    # total_tries_lbl = gr.Label(label="Copys gerados")
    greet_btn = gr.Button("Gerar copy")
    greet_btn.click(fn=openai_completion, inputs=prompt, outputs=output)

demo.launch(share=False, auth=("startse", "St@rtS3**"))