#!/bin/bash

# CPU Temperature
cpu_temp=$(sensors | awk '/^Tctl:/ {print $2}' | tr -d '+°C')

# CPU Load (1-minute average)
cpu_load=$(awk '{print $1}' /proc/loadavg)

# GPU Temperature
gpu_temp=$(nvidia-smi --query-gpu=temperature.gpu --format=csv,noheader,nounits 2>/dev/null)

# GPU Load
gpu_load=$(nvidia-smi --query-gpu=utilization.gpu --format=csv,noheader,nounits 2>/dev/null)

# Colors for CPU Load
if (( $(echo "$cpu_load < 1.0" | bc -l) )); then
    cpu_color="#50FA7B"  # Green
elif (( $(echo "$cpu_load < 2.0" | bc -l) )); then
    cpu_color="#F1FA8C"  # Yellow
else
    cpu_color="#FF5555"  # Red
fi

# Colors for GPU Load
if (( $(echo "$gpu_load < 50.0" | bc -l) )); then
    gpu_color="#50FA7B"  # Green
elif (( $(echo "$gpu_load < 75.0" | bc -l) )); then
    gpu_color="#F1FA8C"  # Yellow
else
    gpu_color="#FF5555"  # Red
fi

# Output
echo -e "<span color=\"$cpu_color\">  CPU: ${cpu_temp}°C, Load: ${cpu_load}</span> | <span color=\"$gpu_color\">  GPU: ${gpu_temp}°C, Load: ${gpu_load}%</span>"
echo
echo "#FFFFFF"  # Default color for fallback (ignored if markup is enabled)