Update data_extractor.py

data_extractor.py

@@ -136,12 +136,13 @@ def extract_data(image_links):
         # Call the extraction function
         extracted_data = extract_information(image_links)
         print(f"extracted data : {extracted_data} ")
+        print(f"extracted data : {type(extracted_data)} ")
         
         if extracted_data:
-            ans = json.loads(extracted_data)
+            #ans = json.loads(extracted_data)
             # Store in MongoDB
-            collection.insert_one(ans)
-            return ans
+            collection.insert_one(extracted_data)
+            return extracted_data
         else:
             return {"error": "Failed to extract information"}