import gradio as gr
from SynonymEditor import SynonymEditor


def replace_synonyms(api_key, text, temperature):
    model_engine = "text-davinci-003"
    max_tokens = 500
    editor = SynonymEditor(api_key, model_engine, max_tokens, temperature)
    return editor._edit_text(text)


api_key = gr.inputs.Textbox(label="API Key", lines=1, default="")

input_text = gr.inputs.Textbox(
    label="Input Text", lines=10, default="Enter your text here.")

temperature_slider = gr.inputs.Slider(
    label="Temperature", minimum=0, maximum=1, default=0.7, step=0.1)

output_text = gr.outputs.Textbox(label="Output Text")

io = gr.Interface(fn=replace_synonyms, inputs=[api_key, input_text, temperature_slider], outputs=output_text, title="Synonym Replacer",
                  description="Replace words in a text with their synonyms.")
io.launch()