Spaces:
Running
Running
import os | |
import streamlit as st | |
from Pages.createVisual import CREATEVISUALS | |
from Pages.AboutMe import ABOUTUS | |
from Pages.bgImages import BGIMAGES | |
from Pages.createdVisuals import CREATEDIMAGES | |
from Pages.createVideo import CREATEGIF | |
from Pages.objectRecognize import DETECTIMAGE | |
from Pages.adScript import CREATEAD | |
st.set_page_config( | |
page_title="LumiereIQ", | |
page_icon="📸", | |
layout="wide", # 'centered' or 'wide' | |
initial_sidebar_state="expanded" # 'auto', 'expanded', 'collapsed' | |
) | |
def MAIN(): | |
weights_dir = 'weights' | |
weights_file = os.path.join(weights_dir, 'sam_vit_h_4b8939.pth') | |
# Check if the weights file already exists | |
if not os.path.exists(weights_file): | |
# Create the directory if it doesn't exist | |
if not os.path.exists(weights_dir): | |
os.makedirs(weights_dir) | |
# Define the download command | |
download_command = f"wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth -P {weights_dir}" | |
# Execute the download command | |
os.system(download_command) | |
print("Weights downloaded.") | |
else: | |
print("Weights file already exists. No download needed.") | |
# Verify the file is downloaded | |
file_path = os.path.expanduser("~/weights/sam_vit_h_4b8939.pth") | |
if os.path.exists(file_path): | |
print("File downloaded successfully.") | |
else: | |
print("File download failed.") | |
# Rest of your application code | |
st.sidebar.title('LumiereIQ') | |
app = st.sidebar.selectbox('', ['Create Visuals','Background Images','Recognize Object', 'Create Script', 'Created Images','Create Video', 'About Me']) | |
if app == "Create Visuals": | |
CREATEVISUALS() | |
elif app == "About Me": | |
ABOUTUS() | |
elif app == "Background Images": | |
BGIMAGES() | |
elif app == "Create Video": | |
CREATEGIF() | |
elif app == "Created Images": | |
CREATEDIMAGES() | |
elif app == "Recognize Object": | |
DETECTIMAGE() | |
elif app == "Create Script": | |
CREATEAD() | |
MAIN() |