from flask import Flask, jsonify, render_template
import psutil

app = Flask(__name__)

def get_system_info():
    memory = psutil.virtual_memory()
    disk = psutil.disk_usage('/')
    return {
        "memory": {
            "total": memory.total,
            "available": memory.available,
            "used": memory.used,
            "percent": memory.percent,
        },
        "disk": {
            "total": disk.total,
            "used": disk.used,
            "free": disk.free,
            "percent": disk.percent,
        }
    }

@app.route('/')
def index():
    return render_template('index.html')

@app.route('/system_info')
def system_info():
    return jsonify(get_system_info())

if __name__ == '__main__':
    app.run(debug=True)