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:esquizofrenia-2023-08-31-20-33-36",
        prompt=prompt,
        max_tokens=10,
        temperature=0,
       
    )

    return response.choices[0].text.strip()
  

iface = gr.Interface(
    fn=generate_completion,
    inputs=gr.components.Textbox(lines=5, placeholder='Escribe aquí la frase o las palabras '),
    outputs='text',
    title="Detector de potenciales síntomas de esquizofrenia",
    description="Plataforma que detecta potenciales síntomas de esquizofrenia. Escriba la frase y le suma la flecha  -> mas dos espacios, ejemplo: Veo monstruos en la noche  ->  La plataforma está basada en redes neuronales transformer usando la técnica de transfer learning y el proceso de fine tunning",
   )

iface.launch()