import gradio as gr
from src.inference import CodeStylePipeline

pipe=CodeStylePipeline()

def run(code,style):
    return pipe(code,style)

gr.Interface(
    fn=run,
    inputs=[gr.Textbox(lines=10), gr.Dropdown(['clean','competitive','enterprise','legacy'])],
    outputs='text',
    title='CodeStyleDiffuser'
).launch()