import streamlit as st
from streamlit_option_menu import option_menu
from page1 import text
from page2 import image
from page3 import sdxl

def main():


    with st.sidebar:
        selection = option_menu(
        menu_title="OpenCopilot",
        options=["Vanilla Chat", "Chat with Image", "Image Creator"],
        icons=["pencil", "chat", "image"],
        menu_icon="bi bi-brilliance",
        default_index=0
        )

    if selection == "Vanilla Chat":
        text()

    elif selection == "Vision Chat":
        image()

    elif selection == "Image Creator":
        sdxl()

   


if __name__ == '__main__':
    main()