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

def main():


    with st.sidebar:
        selection = option_menu(
        menu_title="ForgeStudio",
        options=["Home", "Vanilla Chat", "Chat with Image", "Image Creator"],
        icons=["house", "pencil", "chat", "image"],
        menu_icon="cast",
        default_index=1
        )

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

    elif selection == "Chat with Image":
        image()

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

    elif selection == "Home":
        details()


if __name__ == '__main__':
    main()