from flask import Flask, request, jsonify from flask_cors import CORS import time import logging import json logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) app = Flask(__name__) CORS(app) # MEMORIA SEMPLICE coordinators = {} workers = {} system_stats = { 'connected_users': 1, 'active_workers': 0, 'online_coordinators': 0, 'total_gpu_memory': 0, 'total_requests': 12547, 'system_uptime': int(time.time()) } print("=" * 70) print("🚀 ALLO CENTRAL SERVER - DASHBOARD MODE") print("🌐 http://localhost:5000") print("📊 Dashboard ready with real-time monitoring") print("=" * 70) @app.route('/api/register_coordinator', methods=['POST']) def register_coordinator(): try: data = request.json coord_id = data.get('coordinator_id', 'unknown') print(f"🎯 COORDINATOR REGISTRATION RECEIVED: {coord_id}") print(f"📊 Workers: {data.get('current_workers', 0)}, Memory: {data.get('total_gpu_memory', 0)}MB") # Salva coordinator coordinators[coord_id] = { 'host': data.get('host', 'localhost'), 'port': data.get('port', 8765), 'workers': data.get('current_workers', 0), 'total_memory': data.get('total_gpu_memory', 0), 'performance_score': data.get('performance_score', 0), 'last_seen': time.time(), 'capabilities': data.get('capabilities', {}) } print(f"✅ COORDINATOR REGISTERED: {coord_id}") return jsonify({'status': 'success', 'message': 'Coordinator registered'}) except Exception as e: print(f"❌ COORDINATOR ERROR: {e}") return jsonify({'status': 'error', 'message': str(e)}), 400 @app.route('/api/register_node', methods=['POST']) def register_node(): try: data = request.json worker_id = data.get('node_id', 'unknown') print(f"🎯 WORKER REGISTRATION RECEIVED: {worker_id}") print(f"📊 GPU: {data.get('gpu_name')}, Memory: {data.get('gpu_memory', 0)}MB") # Salva worker workers[worker_id] = { 'coordinator_id': data.get('coordinator_id', 'unknown'), 'gpu_name': data.get('gpu_name', 'Unknown GPU'), 'gpu_memory': data.get('gpu_memory', 0), 'performance_score': data.get('performance_score', 0), 'node_type': data.get('node_type', 'unknown'), 'last_seen': time.time() } print(f"✅ WORKER REGISTERED: {worker_id} - {data.get('gpu_memory', 0)}MB") return jsonify({'status': 'success', 'message': 'Worker registered'}) except Exception as e: print(f"❌ WORKER ERROR: {e}") return jsonify({'status': 'error', 'message': str(e)}), 400 @app.route('/api/system_stats', methods=['GET']) def get_system_stats(): # Calcola statistiche in tempo reale active_workers = len(workers) total_memory = sum(worker['gpu_memory'] for worker in workers.values()) online_coords = len(coordinators) total_performance = sum(worker.get('performance_score', 0) for worker in workers.values()) # Aggiorna statistiche updated_stats = { 'connected_users': 1, 'active_workers': active_workers, 'online_coordinators': online_coords, 'total_gpu_memory': total_memory, 'total_performance': total_performance, 'total_requests': 12547 + active_workers * 100, 'system_uptime': int(time.time() - system_stats['system_uptime']) } if active_workers > 0: print(f"📊 LIVE: {active_workers} workers, {total_memory}MB memory, {online_coords} coordinators") return jsonify({'stats': updated_stats, 'timestamp': time.time()}) @app.route('/api/health', methods=['GET']) def health(): return jsonify({'status': 'online', 'timestamp': time.time()}) @app.route('/api/debug', methods=['GET']) def debug(): return jsonify({ 'coordinators': coordinators, 'workers': workers, 'total_coordinators': len(coordinators), 'total_workers': len(workers), 'total_memory': sum(worker['gpu_memory'] for worker in workers.values()), 'total_performance': sum(worker.get('performance_score', 0) for worker in workers.values()) }) @app.route('/api/coordinator/', methods=['GET']) def get_coordinator(coord_id): if coord_id in coordinators: return jsonify({ 'coordinator': coordinators[coord_id], 'workers': [w for w in workers.values() if w['coordinator_id'] == coord_id] }) return jsonify({'error': 'Coordinator not found'}), 404 @app.route('/') def dashboard(): return ''' 🚀 AllO AI - Distributed Computing Platform

🚀 AllO AI

Next-Generation Distributed AI Computing Platform

LIVE • REAL-TIME • DISTRIBUTED

📡 Connected Coordinators

Loading coordinators...

⚡ Active Workers

Loading workers...
Last update: Loading...
''' if __name__ == '__main__': app.run(host='0.0.0.0', port=5000, debug=True)