Evidence_prediction / evidences_db.py
itzbhav's picture
Upload 6 files
dcc5005 verified
from pymongo import MongoClient
# Connect to MongoDB
client = MongoClient('mongodb://localhost:27017/')
db = client['risk_prediction_db']
evidence_collection = db['evidences']
# Example evidence records to insert
sample_evidences = [
{
"username": "admin",
"image_filename": "evidence_image_1.png",
"temperature": 25.3,
"humidity": 60.5,
"vibration": 1, # 1 means "Yes"
"light_intensity": 400.0,
"battery_level": 85.0,
"latitude": 37.7749,
"longitude": -122.4194,
"risk_prediction": "High Risk"
},
{
"username": "user1",
"image_filename": "evidence_image_2.png",
"temperature": 22.1,
"humidity": 55.3,
"vibration": 0, # 0 means "No"
"light_intensity": 500.0,
"battery_level": 90.0,
"latitude": 40.7128,
"longitude": -74.0060,
"risk_prediction": "Low Risk"
},
{
"username": "user2",
"image_filename": "evidence_image_3.png",
"temperature": 35.7,
"humidity": 75.1,
"vibration": 1, # 1 means "Yes"
"light_intensity": 300.0,
"battery_level": 65.0,
"latitude": 51.5074,
"longitude": -0.1278,
"risk_prediction": "High Risk"
}
]
# Insert evidence records into MongoDB
evidence_collection.insert_many(sample_evidences)
print("inserted")