File size: 2,168 Bytes
ead2510 f03e4f0 ead2510 f03e4f0 ead2510 f03e4f0 ead2510 f03e4f0 |
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 |
#!/usr/bin/env python3
import os
import sys
import subprocess
import time
from pathlib import Path
def run_command(cmd):
print(f"Running command: {' '.join(cmd)}")
process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
for line in process.stdout:
sys.stdout.write(line)
sys.stdout.flush()
process.wait()
return process.returncode
def check_required_files():
required_files = [
"pyscout_api.py",
"deepinfra_client.py",
"proxy_finder.py",
"db_helper.py",
"hf_utils.py",
]
for file in required_files:
if not Path(file).exists():
print(f"ERROR: Required file '{file}' not found!")
return False
return True
def wait_for_mongodb():
import time
import pymongo
mongo_uri = os.environ.get("MONGODB_URI")
if not mongo_uri:
print("MongoDB URI not found in environment variables, skipping connection check")
return True
max_attempts = 30
for attempt in range(max_attempts):
try:
client = pymongo.MongoClient(mongo_uri, serverSelectionTimeoutMS=5000)
client.admin.command('ping')
print(f"MongoDB connection successful after {attempt+1} attempts")
return True
except Exception as e:
print(f"Attempt {attempt+1}/{max_attempts}: MongoDB not yet available. Waiting... ({str(e)})")
time.sleep(2)
print("ERROR: Failed to connect to MongoDB after multiple attempts")
return False
def main():
if not check_required_files():
sys.exit(1)
# Always run in API mode with fixed host and port
print("Starting PyScoutAI API server")
wait_for_mongodb()
host = os.environ.get("HOST", "0.0.0.0")
port = int(os.environ.get("PORT", "7860"))
cmd = [
"uvicorn",
"pyscout_api:app",
"--host", host,
"--port", str(port)
]
return run_command(cmd)
if __name__ == "__main__":
sys.exit(main()) |