float (heatmap)

app.py

@@ -78,7 +78,7 @@ def prepare_heatmap_data(data):
     print("DEBUG: Data type received in prepare_heatmap_data:", type(data))
     print("DEBUG: Data content:", data)
 
-    heatmap_data = pd.DataFrame("", index=range(len(data)), columns=id2label.values())
+    heatmap_data = pd.DataFrame(0.0, index=range(len(data)), columns=id2label.values())
 
     for idx, row in enumerate(data):
         print(f"DEBUG: Item at index {idx}: {row} (type: {type(row)})")
@@ -89,7 +89,7 @@ def prepare_heatmap_data(data):
         confidences = row["emotions"].tolist()
         for idy, confidence in enumerate(confidences):
             emotion = id2label[idy]
-            heatmap_data.at[idx, emotion] = f"{round(100 * confidence, 2)}%"
+            heatmap_data.at[idx, emotion] = round(confidence, 4)
 
     heatmap_data.index = [item["sentence"][:18]+"..." for item in data]
     return heatmap_data