debugging for a while

app.py

@@ -73,15 +73,24 @@ def get_most_probable_label(probs):
     probability = f"{round(100 * probs.max(), 2)}%"
     return label, probability
 
+
 def prepare_heatmap_data(data):
-    heatmap_data = pd.DataFrame(0, index=range(len(data)), columns=id2label.values())
+    print("DEBUG: Data type received in prepare_heatmap_data:", type(data))
+    print("DEBUG: Data content:", data)
+
+    heatmap_data = pd.DataFrame(0.0, index=range(len(data)), columns=id2label.values())
+
     for idx, item in enumerate(data):
-        print(item) # DEBUG
+        print(f"DEBUG: Item at index {idx}: {item} (type: {type(item)})")
+
+        if not isinstance(item, dict):
+            raise TypeError(f"Expected dict, but got {type(item)} at index {idx}: {item}")
+
         confidences = item["emotions"].tolist()
-        print(confidences)
-        for idy, confidence in enumerate(confidences): # it's a numpy array...
+        for idy, confidence in enumerate(confidences):
             emotion = id2label[idy]
             heatmap_data.at[idx, emotion] = confidence
+
     heatmap_data.index = [item["sentence"] for item in data]
     return heatmap_data