from InnovationHub.llm.vector_store import *
from InnovationHub.llm.chain import *


def start_ui():
    chatbot_interface = gradio.Interface(
        fn=chat,
        inputs=["text",
                gradio.inputs.Dropdown(vehicle_options, label="Select a Mercedes-Benz Owner's Manual")
                #gradio.inputs.Slider(minimum=1, maximum=10, step=1, label="k"),
                #gradio.inputs.Slider(minimum=0.0, maximum=1.0, step=0.01, label="Temperature")
                ],
        outputs="text",
        title="Mercedes-Benz Owner's Manual",
        description="Ask questions to get answers from the Mercedes-Benz Owner's manual. <br/><font size='-2'><u>Disclaimer:</u> THIS IS NOT OFFICIAL AND MAY NOT BE AVAILABLE ALL THE TIME. ALWAYS LOOK AT THE OFFICIAL DOCUMENTATION at https://www.mbusa.com/en/owners/manuals</font>",
        article = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=kaushikdatta.owner-manual" alt="visitor badge"/></center>'
    )

    chatbot_interface.launch()

if __name__ == '__main__':
    start_ui()