import argparse
import gradio as gr
from ui import chat

def main(args):
    demo = gr.ChatInterface(
        fn=chat, 
        examples=["hello", "how are you?", "What is Large Language Model?"], 
        title="Space of Gradio ➕ Text Generation Inference",
        multimodal=False
    )
    
    demo.queue().launch(server_name="0.0.0.0", server_port=args.port)

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="This is my Gradio app's description")
    parser.add_argument("--port", type=int, default=7860, help="Port to expose Gradio app")
    
    args = parser.parse_args()    
    main(args)