Hugging Face's logo

Spaces:
mrwabnalas40
/
Test
Runtime error

App Files Community
Test
File size: 2,150 Bytes 
eebf941
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
import logging
import socket
import threading
import psutil
import GPUtil
from flask import Flask, render_template
from flask_socketio import SocketIO, emit
from peer_discovery import PEERS
from peer_discovery import PORT


logging.basicConfig(level=logging.INFO)

app = Flask(__name__)
socketio = SocketIO(app, cors_allowed_origins="*")

node_id = socket.gethostname()
connected_peers_data = {}

# ─────────────── Ψ¬Ω…ΨΉ Ψ¨ΩŠΨ§Ω†Ψ§Ψͺ Ψ§Ω„Ψ­Ω…Ω„ ───────────────
def get_system_status():
    cpu = psutil.cpu_percent()
    ram = psutil.virtual_memory().percent
    gpu_load = 0
    try:
        gpus = GPUtil.getGPUs()
        if gpus:
            gpu_load = gpus[0].load * 100
    except:
        pass
    return {"cpu": cpu, "ram": ram, "gpu": gpu_load}

# ─────────────── Ψ₯Ψ±Ψ³Ψ§Ω„ Ψ§Ω„ΨͺحديثاΨͺ ───────────────
def broadcast_status():
    while True:
        status = get_system_status()
        socketio.emit("status_update", {node_id: status}, broadcast=True)
        socketio.sleep(5)

# ─────────────── ءفحة Ψ§Ω„ΩˆΨ§Ψ¬Ω‡Ψ© ───────────────
@app.route("/")
def index():
    return render_template("dashboard.html", peers=connected_peers_data)

# ─────────────── Ψ§Ψ³ΨͺΩ‚Ψ¨Ψ§Ω„ ΨͺحديثاΨͺ Ψ§Ω„Ψ­Ψ§Ω„Ψ© ───────────────
@socketio.on("status_update")
def handle_status_update(data):
    connected_peers_data.update(data)
    emit("update_peers", connected_peers_data, broadcast=True)

# ─────────────── Ψ―Ψ±Ψ―Ψ΄Ψ© ───────────────
@socketio.on("send_message")
def handle_message(data):
    emit("receive_message", data, broadcast=True)

# ─────────────── ΨͺΨ΄ΨΊΩŠΩ„ ───────────────
if __name__ == "__main__":
    threading.Thread(target=broadcast_status).start()
    logging.info(f"πŸš€ ΨͺΨ΄ΨΊΩŠΩ„ Dashboard ΨΉΩ„Ω‰ {node_id}")
    socketio.run(app, host="0.0.0.0", port=7000)