|
|
| 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)
|
|
|
|
|
| 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()) |
