import gradio as gr
import openai
import os

openai.api_key = os.getenv("OPENAI_API_KEY")



def generate_completion(user_prompt):
    hidden_context = " "
    prompt = hidden_context + user_prompt
    response = openai.Completion.create(
        model="davinci:ft-topwow-llc:tumorgastrico-2023-09-14-20-11-13",
        prompt=prompt,
        max_tokens=30,
        temperature=0,
        stop=["_END"]
    )


    return response.choices[0].text.strip()
iface = gr.Interface(fn=generate_completion,
                     inputs=gr.inputs.Textbox(lines=5, placeholder='Escriba la frase aqui'),
                     outputs='text',
                     title="Detector de potenciales síntomas de tumor gastrico",
                     description="Plataforma que detecta potenciales síntomas de tumor gástrico. Escriba la frase y le suma la flecha  -> mas dos espacios, ejemplo: Tengo un intenso dolor en la boca del estomago  -> ",
                     input_labels="frase",
                     output_labels="prediccion")

iface.launch()