Add confidence_score field to save_data function for enhanced data tracking

main.py

@@ -345,7 +345,8 @@ async def save_data(
             "retrieved_text": [],
             "model_type": [],
             "reaction": [],
-            "timestamp": []
+            "timestamp": [],
+            "confidence_score": []
         }
         
         # Add each item to the data dict
@@ -357,6 +358,7 @@ async def save_data(
             data["model_type"].append(item.model_type)
             data["reaction"].append(item.reaction)
             data["timestamp"].append(datetime.now(timezone.utc).isoformat().replace('+00:00', 'Z'))
+            data["confidence_score"].append(item.get("confidence_score", 0.0))
         
         try:
             # Load existing dataset and merge
@@ -369,7 +371,10 @@ async def save_data(
             # Add new data
             for key in data:
                 if key not in existing_data:
-                    existing_data[key] = ["" if key in ["timestamp"] else None] * len(next(iter(existing_data.values())))
+                    existing_data[key] = [
+                         "" if key in ["timestamp"] else 
+                         0.0 if key in ["confidence_score"] else None
+                     ] * len(next(iter(existing_data.values())))
                 existing_data[key].extend(data[key])
         
         except Exception as e: