|
|
| import os
|
| import sys
|
| import subprocess
|
| from pathlib import Path
|
| import shutil
|
|
|
| def setup_environment():
|
| """Set up the environment for the application."""
|
| print("Setting up environment...")
|
|
|
|
|
| if not os.path.exists('.env'):
|
| shutil.copy('.env.example', '.env')
|
| print("Created .env file from .env.example. Please update with your settings.")
|
|
|
|
|
| instance_path = os.getenv('INSTANCE_PATH', '/tmp/instance')
|
| Path(instance_path).mkdir(parents=True, exist_ok=True)
|
| print(f"Created instance directory at {instance_path}")
|
|
|
|
|
| models_dir = Path('app/models')
|
| models_dir.mkdir(parents=True, exist_ok=True)
|
| print("Created models directory")
|
|
|
| return True
|
|
|
| def setup_database():
|
| """Set up the database for the application."""
|
| print("Setting up database...")
|
| try:
|
| subprocess.run([sys.executable, 'manage.py', 'init-db'], check=True)
|
| print("Database initialized successfully.")
|
| return True
|
| except subprocess.CalledProcessError:
|
| print("Error initializing database. Please check your database configuration.")
|
| return False
|
|
|
| def download_yolo_models():
|
| """Download YOLOv4 models for object detection."""
|
| print("Downloading YOLOv4 models...")
|
| models_dir = Path('app/models')
|
|
|
|
|
| weights_path = models_dir / "yolov4.weights"
|
| config_path = models_dir / "yolov4.cfg"
|
|
|
| if weights_path.exists() and config_path.exists():
|
| print("YOLOv4 models already exist. Skipping download.")
|
| return True
|
|
|
| try:
|
|
|
| subprocess.run([
|
| "wget",
|
| "https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights",
|
| "-O", str(weights_path)
|
| ], check=True)
|
|
|
|
|
| subprocess.run([
|
| "wget",
|
| "https://raw.githubusercontent.com/AlexeyAB/darknet/master/cfg/yolov4.cfg",
|
| "-O", str(config_path)
|
| ], check=True)
|
|
|
| print("YOLOv4 models downloaded successfully.")
|
| return True
|
| except subprocess.CalledProcessError:
|
| print("Error downloading YOLOv4 models. Please download them manually.")
|
| return False
|
|
|
| def main():
|
| """Main setup function."""
|
| print("Setting up Facebook Ad Analytics...")
|
|
|
|
|
| if not setup_environment():
|
| print("Error setting up environment. Exiting.")
|
| return
|
|
|
|
|
| if not setup_database():
|
| print("Error setting up database. Exiting.")
|
| return
|
|
|
|
|
| if not download_yolo_models():
|
| print("Error downloading YOLO models. You may need to download them manually.")
|
|
|
| print("\nSetup completed successfully!")
|
| print("To run the application, use: python manage.py run")
|
| print("For production deployment, use Docker: docker-compose up -d")
|
|
|
| if __name__ == "__main__":
|
| main() |
