Spaces:
Running
Running
| import streamlit as st | |
| import replicate | |
| from src.page1 import page1 | |
| from src.page2 import page2 | |
| from src.page3 import page3 | |
| from src.page4 import page4 | |
| from src.page5 import page5 | |
| from src.page6 import page6 | |
| from src.page7 import page7 | |
| from src.page8 import page8 | |
| # import openai | |
| import os | |
| from dotenv import load_dotenv | |
| load_dotenv() | |
| REPLICATE_API_TOKEN = os.environ.get("REPLICATE_API_TOKEN") | |
| replicate = replicate.Client(api_token=REPLICATE_API_TOKEN) | |
| st.set_page_config(layout="wide") | |
| # openai.api_key = os.getenv("OPENAI_API_KEY") | |
| pages = { | |
| # "Entry point": page1, | |
| # "Text to image": page2, | |
| # "Image variation": page3, | |
| # "Image edit": page4, | |
| "Text to Image":page5, | |
| "Image to Text": page6, | |
| "Image to Video":page7, | |
| "Text to Video": page8 | |
| } | |
| # Create the selectbox in the sidebar | |
| page = st.sidebar.selectbox("Select a page", list(pages.keys())) | |
| # Display the selected page | |
| pages[page]() |