Update api/data_extractor.py

api/data_extractor.py

@@ -21,6 +21,9 @@ collection = db.products
 print(f"collection is {collection}")
 
 
+def encode_image(uploaded_file):
+    return base64.b64encode(uploaded_file.read()).decode('utf-8')
+
 def extract_information(images_list: List[Any]) -> Dict[str, Any]:
     global openai_client
     print(f"DEBUG - openai_client : {openai_client}")
@@ -59,8 +62,10 @@ Your goal will be to extract information from these images to populate the schem
         # Convert valid images to byte streams for API processing
         image_message = [
             {
-                "type": "image",
-                "image": {"bytes": base64.b64encode(uploaded_file.read()).decode('utf-8')}
+                "type": "image_url",
+                "image_url": {
+                "url":  f"data:image/jpeg;base64,{encode_image(uploaded_file)}"
+                }
             }
             for uploaded_file in valid_image_files
         ]