|
|
|
|
|
import os
|
|
|
from huggingface_hub import snapshot_download
|
|
|
import sys
|
|
|
|
|
|
|
|
|
|
|
|
os.environ.setdefault("HOME", "/home/user")
|
|
|
|
|
|
os.environ.setdefault("STREAMLIT_SERVER_ENABLE_XSRF_PROTECTION", "false")
|
|
|
os.environ.setdefault("STREAMLIT_SERVER_ENABLE_CORS", "false")
|
|
|
os.environ.setdefault("STREAMLIT_BROWSER_GATHER_USAGE_STATS", "false")
|
|
|
os.environ.setdefault("STREAMLIT_SERVER_MAX_UPLOAD_SIZE", "1024")
|
|
|
|
|
|
|
|
|
try:
|
|
|
cfg_dir = os.path.join(os.environ["HOME"], ".streamlit")
|
|
|
os.makedirs(cfg_dir, exist_ok=True)
|
|
|
cfg_path = os.path.join(cfg_dir, "config.toml")
|
|
|
if not os.path.exists(cfg_path):
|
|
|
with open(cfg_path, "w", encoding="utf-8") as f:
|
|
|
f.write(
|
|
|
"[server]\n"
|
|
|
"enableXsrfProtection = false\n"
|
|
|
"enableCORS = false\n"
|
|
|
"maxUploadSize = 1024\n"
|
|
|
"\n[browser]\n"
|
|
|
"gatherUsageStats = false\n"
|
|
|
)
|
|
|
except Exception:
|
|
|
pass
|
|
|
|
|
|
|
|
|
LOCAL_DIR = snapshot_download(
|
|
|
repo_id="zenefil/veo",
|
|
|
repo_type="space",
|
|
|
use_auth_token=os.getenv("HF_TOKEN"),
|
|
|
)
|
|
|
|
|
|
|
|
|
streamlit_script = os.path.join(LOCAL_DIR, "src", "streamlit_app.py")
|
|
|
if not os.path.exists(streamlit_script):
|
|
|
print(f"File tidak ditemukan: {streamlit_script}")
|
|
|
sys.exit(1)
|
|
|
|
|
|
os.execvp(
|
|
|
"streamlit",
|
|
|
[
|
|
|
"streamlit",
|
|
|
"run",
|
|
|
streamlit_script,
|
|
|
"--server.address", "0.0.0.0",
|
|
|
"--server.port", os.getenv("PORT", "7860"),
|
|
|
],
|
|
|
)
|
|
|
|