Made changes to database.py

app/database.py

@@ -1,31 +1,14 @@
 import os
-from datetime import datetime
 from pymongo import MongoClient
 
-# Updated MongoDB URI
-MONGO_URI = "mongodb+srv://pranaychamala:T0UkrSFvPYhUULjr@uspark.azl7t.mongodb.net/?retryWrites=true&w=majority&tls=true"
+# Replace with your actual username and password
+MONGO_URI = "mongodb+srv://root:loXFr21Z68AxC7Br@medseg.kugvh.mongodb.net/uspark_db?retryWrites=true&w=majority&tls=true"
 
-# Connect to MongoDB
 client = MongoClient(MONGO_URI)
-db = client["Uspark"]  # Assuming the database name is still 'uspark_db'
+db = client["uspark_db"]
 
 def save_chat_session(session_id: str, conversation_data: dict):
-    """
-    Save a chatbot conversation session with a timestamp.
-    """
-    document = {
-        "session_id": session_id,
-        "conversation_data": conversation_data,
-        "timestamp": datetime.utcnow()
-    }
-    db.chatbot.insert_one(document)
+    db.chatbot.insert_one({"session_id": session_id, **conversation_data})
 
 def save_medseg_result(result_data: dict):
-    """
-    Save a MedSeg result with a timestamp.
-    """
-    document = {
-        **result_data,
-        "timestamp": datetime.utcnow()
-    }
-    db.medseg.insert_one(document)
+    db.medseg.insert_one(result_data)