Spaces:
Running
Running
File size: 486 Bytes
8dcd1f3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import os
from pymongo import MongoClient
# Replace with your actual username and password
MONGO_URI = "mongodb+srv://root:[email protected]/uspark_db?retryWrites=true&w=majority"
client = MongoClient(MONGO_URI)
db = client["uspark_db"]
def save_chat_session(session_id: str, conversation_data: dict):
db.chatbot.insert_one({"session_id": session_id, **conversation_data})
def save_medseg_result(result_data: dict):
db.medseg.insert_one(result_data)
|