Update main.py

main.py

@@ -44,6 +44,7 @@ def sanitize_filename(filename: str) -> str:
 async def upload_new_images(files: List[UploadFile] = File(...), folder_name: str = Form(...)):
     uploaded_urls = []
     standardized_folder = standardize_category_name(folder_name)
+    detect_faces: bool = Form(True)
     
     try:
         for file in files:
@@ -54,7 +55,7 @@ async def upload_new_images(files: List[UploadFile] = File(...), folder_name: st
                 
             image_url = db.upload_image(temp_path, standardized_folder)
             
-            vectors_to_save = ai.process_image(temp_path, is_query=False)
+            vectors_to_save = ai.process_image(temp_path, is_query=False, detect_faces=detect_faces)
             
             for vec_dict in vectors_to_save:
                 image_id = str(uuid.uuid4())
@@ -72,11 +73,12 @@ async def search_database(file: UploadFile = File(...)):
     try:
         safe_filename = sanitize_filename(file.filename)
         temp_path = f"temp_uploads/query_{safe_filename}"
+        detect_faces: bool = Form(True)
         
         with open(temp_path, "wb") as buffer:
             shutil.copyfileobj(file.file, buffer)
             
-        vectors_to_search = ai.process_image(temp_path, is_query=True)
+        vectors_to_search = ai.process_image(temp_path, is_query=True, detect_faces=detect_faces)
         
         all_results = []
         for vec_dict in vectors_to_search: