import gradio as gr
from local_jarvis_model import LocalJarvisModel

jarvis = LocalJarvisModel()

def chat(message, history):
    return jarvis.reply(message)

demo = gr.ChatInterface(
    fn=chat,
    title="Jarvis AI",
    description="Custom-built AI chatbot"
)

demo.launch(share=True)