Create heatmap.py

tools/heatmap.py

@@ -0,0 +1,26 @@
+import pandas as pd
+import matplotlib.pyplot as plt
+import numpy as np
+
+
+def generate_heatmap(df: pd.DataFrame):
+    if df is None or df.empty:
+        return None
+
+    heat = np.zeros((20, 30))
+
+    for _, row in df.iterrows():
+        a = int(row["Suggested Aisle"])
+        r = int(row["Suggested Rack"])
+        if 1 <= a <= 20 and 1 <= r <= 30:
+            heat[a - 1, r - 1] += 1
+
+    plt.figure(figsize=(6, 4))
+    plt.imshow(heat, cmap="hot", interpolation="nearest")
+    plt.colorbar()
+    plt.title("Warehouse Heatmap")
+    img_path = "/tmp/heatmap.png"
+    plt.savefig(img_path)
+    plt.close()
+
+    return img_path