main.py

from flask import Flask, jsonify
from flask_cors import CORS
from thread_sys import ThreadSys

app = Flask(__name__)
CORS(app)
app.config['SECRET_KEY'] = 'secret!'
sys_info = ThreadSys()


@app.route('/')
def entry_page():
    cpu_used = sys_info.cpu_used
    memory_total = sys_info.memory_total
    memory_used = sys_info.memory_used
    rx_speed = sys_info.rx_speed
    tx_speed = sys_info.tx_speed

    return """
    CPU LOAD: {}%
    TOTAL MEMORY: {} GiB
    MEMORY USED: {} GiB
    RX_SPEED: {} Mbps
    TX_SPEED: {} Mbps""".format(cpu_used, memory_total, memory_used, rx_speed, tx_speed)


@app.route('/info')
def info():
    data = {'cpu_used': sys_info.cpu_used,
            'memory_total': sys_info.memory_total,
            'memory_used': sys_info.memory_used,
            'rx_speed': sys_info.rx_speed,
            'tx_speed': sys_info.tx_speed}
    return jsonify(data)

#
# def get_available_gpus():
#     local_device_protos = device_lib.list_local_devices()
#     gpus = [x.name for x in local_device_protos if x.device_type == 'GPU']
#     return len(gpus)

if __name__ == '__main__':
    app.run(debug=False, port=5000, host='0.0.0.0')