multimodal-ai-engine / database.py
pathananas's picture
database.py
3ab7835 verified
raw
history blame contribute delete
996 Bytes
import os
from pymongo import MongoClient
from datetime import datetime
# ======================================
# MongoDB Connection
# ======================================
MONGO_URI = os.getenv("MONGO_URI", "mongodb://localhost:27017")
client = MongoClient(MONGO_URI)
db = client["multimodal_db"]
analysis_collection = db["analysis_history"]
# ======================================
# Save Analysis
# ======================================
def save_analysis(data):
data["created_at"] = datetime.utcnow()
analysis_collection.insert_one(data)
# ======================================
# Get History
# ======================================
def get_history(limit=20):
return list(
analysis_collection
.find({}, {"_id": 0})
.sort("created_at", -1)
.limit(limit)
)
# ======================================
# Clear History
# ======================================
def clear_history_db():
analysis_collection.delete_many({})
#comit