import os
from dataset_loader import load_dataset

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

import gradio as gr
from mystartup_data.agent import Agent

iface = gr.Interface(
    fn=Agent().ask,
    inputs=gr.components.Textbox(lines=7, label="Enter your text"),
    outputs="text",
    title="MyStartup Bot",
    description="""This is a chatbot to answer your question regarding the MyStartup platform. 
\nNotes: This chatbot does not implement the memory module, ask specific question and don't expect it remembers the last conversation."""
    )
iface.launch(share=False)