Spaces:
Running
Running
| 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 |