Spaces:
Runtime error
Runtime error
| #!/usr/bin/env python3 | |
| """ | |
| Create distributable package for SafetyMaster Pro | |
| Creates a ZIP file with all necessary components for easy sharing | |
| """ | |
| import os | |
| import shutil | |
| import zipfile | |
| from pathlib import Path | |
| import datetime | |
| def create_distribution_package(): | |
| """Create a complete distribution package.""" | |
| print("📦 Creating SafetyMaster Pro Distribution Package") | |
| print("=" * 50) | |
| # Create distribution folder | |
| timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S") | |
| dist_name = f"SafetyMasterPro_v1.0_{timestamp}" | |
| dist_folder = f"{dist_name}" | |
| if os.path.exists(dist_folder): | |
| shutil.rmtree(dist_folder) | |
| os.makedirs(dist_folder) | |
| print(f"📁 Created distribution folder: {dist_folder}") | |
| # Files to include in distribution | |
| files_to_copy = [ | |
| 'web_interface.py', | |
| 'safety_detector.py', | |
| 'camera_manager.py', | |
| 'config.py', | |
| 'requirements.txt', | |
| 'README.md', | |
| 'high_fps_test.py', | |
| 'test_improved_detection.py', | |
| 'test_camera.py', | |
| ] | |
| # Copy Python files | |
| print("📄 Copying Python files...") | |
| for file in files_to_copy: | |
| if os.path.exists(file): | |
| shutil.copy2(file, dist_folder) | |
| print(f" ✅ {file}") | |
| else: | |
| print(f" ⚠️ {file} not found") | |
| # Copy model files | |
| print("🤖 Copying AI model files...") | |
| model_files = list(Path('.').glob('*.pt')) | |
| for model_file in model_files: | |
| shutil.copy2(model_file, dist_folder) | |
| print(f" ✅ {model_file.name}") | |
| # Copy templates folder | |
| if os.path.exists('templates'): | |
| print("🎨 Copying templates...") | |
| shutil.copytree('templates', os.path.join(dist_folder, 'templates')) | |
| print(" ✅ templates/ folder") | |
| # Copy test files | |
| test_files = [ | |
| 'test_websocket.html', | |
| 'demo.py', | |
| 'demo_simple.py', | |
| ] | |
| print("🧪 Copying test files...") | |
| for file in test_files: | |
| if os.path.exists(file): | |
| shutil.copy2(file, dist_folder) | |
| print(f" ✅ {file}") | |
| # Create startup scripts | |
| create_startup_scripts(dist_folder) | |
| # Create user guide | |
| create_user_guide(dist_folder) | |
| # Create ZIP package | |
| zip_filename = f"{dist_name}.zip" | |
| print(f"🗜️ Creating ZIP package: {zip_filename}") | |
| with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf: | |
| for root, dirs, files in os.walk(dist_folder): | |
| for file in files: | |
| file_path = os.path.join(root, file) | |
| arc_name = os.path.relpath(file_path, dist_folder) | |
| zipf.write(file_path, arc_name) | |
| # Get package size | |
| zip_size = os.path.getsize(zip_filename) / (1024 * 1024) # MB | |
| print(f"\n🎉 Package created successfully!") | |
| print(f"📦 Package: {zip_filename}") | |
| print(f"📏 Size: {zip_size:.1f} MB") | |
| print(f"📁 Folder: {dist_folder}/") | |
| return zip_filename, dist_folder | |
| def create_startup_scripts(dist_folder): | |
| """Create easy startup scripts for users.""" | |
| print("🚀 Creating startup scripts...") | |
| # Windows batch script | |
| windows_script = '''@echo off | |
| title SafetyMaster Pro | |
| echo. | |
| echo ███████╗ █████╗ ███████╗███████╗████████╗██╗ ██╗ | |
| echo ██╔════╝██╔══██╗██╔════╝██╔════╝╚══██╔══╝╚██╗ ██╔╝ | |
| echo ███████╗███████║█████╗ █████╗ ██║ ╚████╔╝ | |
| echo ╚════██║██╔══██║██╔══╝ ██╔══╝ ██║ ╚██╔╝ | |
| echo ███████║██║ ██║██║ ███████╗ ██║ ██║ | |
| echo ╚══════╝╚═╝ ╚═╝╚═╝ ╚══════╝ ╚═╝ ╚═╝ | |
| echo. | |
| echo MASTER PRO v1.0 | |
| echo Real-time AI Safety Equipment Detection | |
| echo. | |
| echo Checking Python installation... | |
| python --version >nul 2>&1 | |
| if errorlevel 1 ( | |
| echo ❌ ERROR: Python is not installed or not in PATH | |
| echo Please install Python 3.8+ from https://python.org | |
| echo. | |
| pause | |
| exit /b 1 | |
| ) | |
| echo ✅ Python found! | |
| echo. | |
| echo Installing dependencies (first time only)... | |
| pip install -r requirements.txt >nul 2>&1 | |
| echo. | |
| echo 🚀 Starting SafetyMaster Pro... | |
| echo 🌐 Web interface will open at: http://localhost:8080 | |
| echo 📹 Make sure your camera is connected | |
| echo. | |
| echo Press Ctrl+C to stop the application | |
| echo. | |
| python web_interface.py | |
| pause | |
| ''' | |
| # Unix shell script | |
| unix_script = '''#!/bin/bash | |
| clear | |
| echo | |
| echo " ███████╗ █████╗ ███████╗███████╗████████╗██╗ ██╗" | |
| echo " ██╔════╝██╔══██╗██╔════╝██╔════╝╚══██╔══╝╚██╗ ██╔╝" | |
| echo " ███████╗███████║█████╗ █████╗ ██║ ╚████╔╝ " | |
| echo " ╚════██║██╔══██║██╔══╝ ██╔══╝ ██║ ╚██╔╝ " | |
| echo " ███████║██║ ██║██║ ███████╗ ██║ ██║ " | |
| echo " ╚══════╝╚═╝ ╚═╝╚═╝ ╚══════╝ ╚═╝ ╚═╝ " | |
| echo | |
| echo " MASTER PRO v1.0" | |
| echo " Real-time AI Safety Equipment Detection" | |
| echo | |
| echo "Checking Python installation..." | |
| if ! command -v python3 &> /dev/null; then | |
| echo "❌ ERROR: Python 3 is not installed" | |
| echo "Please install Python 3.8+ from your package manager" | |
| exit 1 | |
| fi | |
| echo "✅ Python found!" | |
| echo | |
| echo "Installing dependencies (first time only)..." | |
| pip3 install -r requirements.txt > /dev/null 2>&1 | |
| echo | |
| echo "🚀 Starting SafetyMaster Pro..." | |
| echo "🌐 Web interface will open at: http://localhost:8080" | |
| echo "📹 Make sure your camera is connected" | |
| echo | |
| echo "Press Ctrl+C to stop the application" | |
| echo | |
| python3 web_interface.py | |
| ''' | |
| # Write scripts | |
| with open(os.path.join(dist_folder, 'START_SafetyMaster.bat'), 'w') as f: | |
| f.write(windows_script) | |
| with open(os.path.join(dist_folder, 'START_SafetyMaster.sh'), 'w') as f: | |
| f.write(unix_script) | |
| # Make shell script executable | |
| os.chmod(os.path.join(dist_folder, 'START_SafetyMaster.sh'), 0o755) | |
| print(" ✅ START_SafetyMaster.bat (Windows)") | |
| print(" ✅ START_SafetyMaster.sh (Unix/Linux/Mac)") | |
| def create_user_guide(dist_folder): | |
| """Create a comprehensive user guide.""" | |
| print("📖 Creating user guide...") | |
| user_guide = '''# SafetyMaster Pro v1.0 - User Guide | |
| ## 🚀 Quick Start | |
| ### Windows Users: | |
| 1. Double-click `START_SafetyMaster.bat` | |
| 2. Wait for installation to complete | |
| 3. Open your web browser to: http://localhost:8080 | |
| ### Mac/Linux Users: | |
| 1. Open terminal in this folder | |
| 2. Run: `./START_SafetyMaster.sh` | |
| 3. Open your web browser to: http://localhost:8080 | |
| ## 📋 Requirements | |
| - **Python 3.8+** (Download from https://python.org) | |
| - **Webcam or USB camera** | |
| - **Internet connection** (for first-time model download) | |
| - **4GB RAM minimum** (8GB recommended) | |
| ## 🎯 Features | |
| - **Real-time PPE Detection**: Hard hats, safety vests, face masks | |
| - **High-Performance**: Optimized for 30+ FPS | |
| - **Web Interface**: Modern, responsive dashboard | |
| - **Violation Alerts**: Real-time safety compliance monitoring | |
| - **Multi-Platform**: Windows, Mac, Linux support | |
| ## 🔧 Manual Installation | |
| If the automatic scripts don't work: | |
| ```bash | |
| # Install dependencies | |
| pip install -r requirements.txt | |
| # Run the application | |
| python web_interface.py | |
| ``` | |
| ## 🎮 Controls | |
| - **Start Monitoring**: Click "Start Monitoring" in the web interface | |
| - **Stop Monitoring**: Click "Stop Monitoring" | |
| - **Fullscreen**: Click the fullscreen button for immersive view | |
| - **Settings**: Adjust camera source and detection settings | |
| ## 🎨 Interface Features | |
| - **Live Video Feed**: Real-time camera with AI detection overlays | |
| - **Statistics Panel**: People count, compliance rate, violations | |
| - **Violation Log**: Real-time alerts with timestamps | |
| - **FPS Counter**: Performance monitoring | |
| - **Responsive Design**: Works on desktop, tablet, mobile | |
| ## 🔍 Detection Classes | |
| The AI model detects: | |
| - ✅ **Hard Hat** (Green boxes when worn) | |
| - ✅ **Safety Vest** (Yellow boxes when worn) | |
| - ✅ **Face Mask** (Blue boxes when worn) | |
| - ❌ **Violations** (Red person boxes when equipment missing) | |
| ## ⚡ Performance Tips | |
| - **Close other applications** for better performance | |
| - **Use good lighting** for better detection accuracy | |
| - **Position camera** to clearly see people and equipment | |
| - **Stable internet** for model downloads | |
| ## 🐛 Troubleshooting | |
| ### Camera Issues: | |
| - Check camera permissions | |
| - Try different camera source (0, 1, 2...) | |
| - Restart the application | |
| ### Performance Issues: | |
| - Close other applications | |
| - Lower camera resolution | |
| - Check system requirements | |
| ### Installation Issues: | |
| - Update Python to latest version | |
| - Run as administrator (Windows) | |
| - Check internet connection | |
| ## 📞 Support | |
| For technical support or questions: | |
| - Check the README.md file | |
| - Review error messages in the console | |
| - Ensure all requirements are met | |
| ## 🔒 Privacy | |
| - All processing is done locally on your computer | |
| - No data is sent to external servers | |
| - Camera feed stays on your device | |
| --- | |
| **SafetyMaster Pro v1.0** - Professional AI-powered safety monitoring | |
| ''' | |
| with open(os.path.join(dist_folder, 'USER_GUIDE.md'), 'w') as f: | |
| f.write(user_guide) | |
| print(" ✅ USER_GUIDE.md") | |
| def main(): | |
| """Main packaging process.""" | |
| try: | |
| zip_file, dist_folder = create_distribution_package() | |
| print(f"\n📋 Distribution Package Summary:") | |
| print(f" 📦 ZIP File: {zip_file}") | |
| print(f" 📁 Folder: {dist_folder}/") | |
| print(f" 🚀 Startup: START_SafetyMaster.bat/.sh") | |
| print(f" 📖 Guide: USER_GUIDE.md") | |
| print(f"\n✅ Ready to share!") | |
| print(f" Users can simply:") | |
| print(f" 1. Extract the ZIP file") | |
| print(f" 2. Run the startup script") | |
| print(f" 3. Open http://localhost:8080") | |
| except Exception as e: | |
| print(f"❌ Error creating package: {e}") | |
| if __name__ == "__main__": | |
| main() |