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


def start_ui():
    chatbot_interface = gradio.Interface(
        fn=chatbot.chat,
        inputs=["text",
                gradio.inputs.Dropdown(
                    vehicle_options, label="Select Vehicle Model"),
                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="Owner's Manual",
        description="Ask your vehicle manual and get a response.",
        examples=[["What are the different features of the dashboard console?", "S-Class", 2, 0.5],
                  ["What is flacon?", "S-Class", 3, 0.8],
                  ["What is hyperscreen?", "EQS", 2, 0.2],
                  ["Where can I find my vin?", "EQS", 3, 0.3]
                  ]
    )

    chatbot_interface.launch()

if __name__ == '__main__':
    start_ui()