import gradio as gr
import openai

def generate_completion(user_prompt):
    hidden_context = " "
    prompt = hidden_context + user_prompt
    response = openai.Completion.create(
        model="curie:ft-topwow-llc:cancerdemamas-2023-10-07-02-25-45",
        prompt=prompt,
        max_tokens=15,
        temperature=0.5
    )
    return response.choices[0].text.strip()
iface = gr.Interface(
        fn=generate_completion,
        inputs=gr.components.Textbox(lines=5, placeholder='Escriba en este espacio los síntomas'),
        outputs='text',
        title="Predictor de potenciales síntomas de cáncer de mamas",
        description="En el espacio se debe escribir los malestares y/o dolencias que se han presentado y la plataforma hará una predicción, de forma potencial, si una persona pudiera presentar síntomas de cáncer de mamas. No reemplaza al médico, es una inteligencia artificial entrenada mediante la técnica de fine tunning con millones de parámetros",
         )

iface.launch(share=True)