Update app.py

app.py

@@ -127,14 +127,14 @@ def extract_invoice_data(file_data, content_type, json_schema):
 
         try:
             parsed_content = json.loads(cleaned_content)
-            return parsed_content
+            return parsed_content,base64dataresp
         except json.JSONDecodeError as e:
             logger.error(f"JSON Parse Error: {e}")
-            return None
+            return None,base64dataresp
 
     except Exception as e:
         logger.error(f"Error in data extraction: {e}")
-        return {"error": str(e)}
+        return {"error": str(e)},base64dataresp
 
 def get_content_type_from_s3(file_key):
     """Fetch the content type (MIME type) of a file stored in S3."""