Update plot.py

plot.py

@@ -1,31 +1,28 @@
-# plot.py
 import matplotlib.pyplot as plt
 import numpy as np
-from io import BytesIO
+import io
 from PIL import Image
 
-def plot_radar_chart(config):
-    labels = list(config.keys())
-    values = list(config.values())
+def plot_radar_chart(traits):
+    labels = list(traits.keys())
+    values = list(traits.values())
 
-    # Convert boolean to numeric for plotting
-    values = [1 if v is True else 0 if v is False else v for v in values]
+    values.append(values[0])  # repeat first to close circle
+    labels.append(labels[0])
 
     angles = np.linspace(0, 2 * np.pi, len(labels), endpoint=False).tolist()
-    values += values[:1]
     angles += angles[:1]
 
-    fig, ax = plt.subplots(figsize=(6, 6), subplot_kw={'polar': True})
-    ax.plot(angles, values, 'o-', linewidth=2)
+    fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(polar=True))
+    ax.plot(angles, values, "o-", linewidth=2)
     ax.fill(angles, values, alpha=0.25)
+    ax.set_yticklabels([])
+    ax.set_xticks(angles[:-1])
+    ax.set_xticklabels(labels)
+    ax.set_title("Trait Radar Chart", y=1.1)
 
-    ax.set_thetagrids(np.degrees(angles[:-1]), labels)
-    ax.set_title("Trait Radar Chart", fontsize=14)
-    ax.grid(True)
-
-    buf = BytesIO()
-    plt.tight_layout()
-    plt.savefig(buf, format="png")
+    buf = io.BytesIO()
+    plt.savefig(buf, format="png", bbox_inches="tight")
     plt.close(fig)
     buf.seek(0)
     return Image.open(buf)