import os
from dataset_loader import load_dataset

if not os.path.isdir('agent_function'):
    load_dataset()

import gradio as gr
from agent_function.agent import Agent

iface = gr.Interface(
    fn=Agent().ask,
    inputs=gr.components.Textbox(lines=7, label="Enter your text"),
    outputs="text",
    title="LGMS Cybersecurity ChatBot",
    description="""Understand Cybersecurity with LLM Chatbot. 1 to 1 Q&A (No memory)"""
    )

iface.launch(share=False)