Update api.py

api.py

@@ -1,13 +1,21 @@
 # Importing the required libraries
 import cv2
-from fastapi import FastAPI, File, UploadFile, HTTPException, Response, Form
+from fastapi import FastAPI, File, UploadFile, HTTPException, Form, Depends
 from fastapi.middleware.cors import CORSMiddleware
 import numpy as np
 import json
+import datetime
 import base64
+from motor.motor_asyncio import AsyncIOMotorClient
+from pymongo.server_api import ServerApi
 
 app = FastAPI()
 
+# MongoDB connection settings
+MONGODB_URL = "mongodb+srv://rohanshaw-1-authdata:[email protected]/?retryWrites=true&w=majority"
+DATABASE_NAME = "AuthData"
+COLLECTION_NAME = "inkognitoUsers"
+
 # Allowing CORS (Cross-Origin Resource Sharing)
 app.add_middleware(
     CORSMiddleware,
@@ -17,6 +25,13 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# MongoDB client initialization
+async def get_database_client():
+    client = AsyncIOMotorClient(MONGODB_URL)
+    database = client[DATABASE_NAME]
+    yield database
+    client.close()
+
 # @app.on_event("startup")
 # def save_openapi_json():
 #     openapi_data = app.openapi()
@@ -139,5 +154,34 @@ async def retrieve_message(image: UploadFile = File(...)):
         decoded_message = show_data(img)
 
         return {"decoded_message": decoded_message}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+    
+@app.post("/save_user_data")
+async def save_user_data(name: str = Form(...), email: str = Form(...), database: AsyncIOMotorClient = Depends(get_database_client)):
+    try:
+        # Save user data to MongoDB
+        users_collection = database[COLLECTION_NAME]
+        user_data = {
+            "name": name,
+            "email": email,
+            "first_login": datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+        }
+        result = await users_collection.insert_one(user_data)
+
+        # Return success response
+        return {"message": "User data saved successfully"}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/get_all_users")
+async def get_all_users(database: AsyncIOMotorClient = Depends(get_database_client)):
+    try:
+        # Retrieve all users from MongoDB
+        users_collection = database[COLLECTION_NAME]
+        all_users = await users_collection.find({}, {"_id": 0}).to_list(length=None)
+
+        # Return the list of all users
+        return {"users": all_users}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))