Update prediction.py

prediction.py

@@ -76,16 +76,23 @@ def is_valid_smiles(smiles):
     return mol is not None
 
 # Function to save prediction to MongoDB
-def save_to_db(smiles_input, predictions):
-    db = get_database()
-    collection = db["polymers"]  # your collection
+from datetime import datetime
+
+def save_to_db(smiles, predictions):
+    # Convert all prediction values to native Python float
+    predictions_clean = {k: float(v) for k, v in predictions.items()}
+    
     doc = {
-        "smiles": smiles_input,
-        "predictions": predictions,
-        "timestamp": datetime.datetime.utcnow()
+        "smiles": smiles,
+        "predictions": predictions_clean,
+        "timestamp": datetime.now()
     }
+
+    db = get_database()
+    collection = db["polymer_predictions"]
     collection.insert_one(doc)
 
+
 # Prediction Page UI
 def show():
     st.markdown("<h1 style='text-align: center; color: #4CAF50;'>🔬 Polymer Property Prediction</h1>", unsafe_allow_html=True)