Spaces:
Sleeping
Sleeping
File size: 2,508 Bytes
69c20e8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
import os
import sys
import subprocess
import platform
def main():
print("🚀 Starting Merch Flow AI...")
# 1. Auto-Detect OS and define venv paths
system = platform.system()
venv_dir = "venv"
if system == "Windows":
bin_dir = "Scripts"
python_exe = "python.exe"
uvicorn_exe = "uvicorn.exe"
else:
bin_dir = "bin"
python_exe = "python"
uvicorn_exe = "uvicorn"
venv_python = os.path.join(venv_dir, bin_dir, python_exe)
venv_uvicorn = os.path.join(venv_dir, bin_dir, uvicorn_exe)
# 2. Check if venv exists
if not os.path.exists(venv_python):
print("❌ Error: Virtual environment not found!")
print("Please run 'setup_env.py' first to initialize the environment.")
sys.exit(1)
# 3. Detect Main Server File
server_files = [
("app.py", "uvicorn"),
("main.py", "uvicorn"),
("streamlit_app.py", "streamlit")
]
target_file = None
runner = None
for filename, r_type in server_files:
if os.path.exists(filename):
target_file = filename
runner = r_type
break
if not target_file:
print("❌ Error: No server file found (looked for app.py, main.py, streamlit_app.py).")
sys.exit(1)
print(f"✅ Found {target_file}, launching with {runner}...")
# 4. Launch Application
try:
if runner == "uvicorn":
# Module name is filename without extension
module_name = os.path.splitext(target_file)[0]
# Construct command: uvicorn main:app --reload
cmd = [venv_uvicorn, f"{module_name}:app", "--reload"]
subprocess.run(cmd, check=True)
elif runner == "streamlit":
# Construct command: streamlit run streamlit_app.py
# Note: streamlit executable might be different, usually <venv>/bin/streamlit
streamlit_exe = os.path.join(venv_dir, bin_dir, "streamlit")
cmd = [streamlit_exe, "run", target_file]
subprocess.run(cmd, check=True)
else:
# Fallback for generic python scripts
cmd = [venv_python, target_file]
subprocess.run(cmd, check=True)
except KeyboardInterrupt:
print("\n🛑 Server stopped by user.")
except Exception as e:
print(f"\n❌ Failed to start server: {e}")
sys.exit(1)
if __name__ == "__main__":
main()
|