import gradio as gr
import os

from headline_gen.Control import ServerCntrl, Generate
os.environ['JAVAHOME'] = 'usr/bin/java'

Server = ServerCntrl("Start")

def MainFun(Article):
    return Generate(Article, Server)


interface = gr.Interface(fn=MainFun, inputs="text", outputs=["text"], title="News Headline Generator")
interface.launch()