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:braintumor-2023-08-30-01-15-51",
        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=' Insert the arrow -> after the description of the symptom, for example "I have a bad headache at night ->"'),
                     outputs='text',
                     title="Brain tumor",
                     description="Platform to detect potential brain tumor symptoms, does not replace medical professionals. ",
                     input_labels="Phrases",
                     output_labels="symptoms")
iface.launch()