Update app.py

app.py

@@ -17,6 +17,7 @@ from transformers import AutoProcessor
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
 from io import BytesIO
+from pymongo import MongoClient
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), './colpali-main')))
 
@@ -165,6 +166,39 @@ async def recommendation(file: UploadFile = File(...), k: int = 10):
     response.headers['Content-Disposition'] = 'attachment; filename="results.pdf"'
 
     return response
-    
+
+def get_database():
+ 
+   # Provide the mongodb atlas url to connect python to mongodb using pymongo
+   CONNECTION_STRING = "mongodb://localhost:27017"
+ 
+   # Create a connection using MongoClient. You can import MongoClient or use pymongo.MongoClient
+   client = MongoClient(CONNECTION_STRING)
+ 
+   # Create the database for our example (we will use the same database throughout the tutorial
+   return client['cvquest-colpali']
+
+dbname = get_database()
+collection_name = dbname["files"]
+
+item_1 = {
+  "_id" : "U1IT00001",
+  "item_name" : "Blender",
+  "max_discount" : "10%",
+  "batch_number" : "RR450020FRG",
+  "price" : 340,
+  "category" : "kitchen appliance"
+}
+
+item_2 = {
+  "_id" : "U1IT00002",
+  "item_name" : "Egg",
+  "category" : "food",
+  "quantity" : 12,
+  "price" : 36,
+  "item_description" : "brown country eggs"
+}
+collection_name.insert_many([item_1,item_2])
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)