Update services/utils.py

services/utils.py

@@ -65,14 +65,15 @@ def strcuture_document_data(raw_text:str)->dict:
         # )
         
         doc_query= (
-            "The response should only be in JSON format very Strictly and it should have these keys brand , total_cost , location, purchase_category,brand_category , Date."    
+            "Extract and return strictly a JSON object containing only the following keys: brand, total_cost, location, purchase_category, brand_category, Date. 
+        Ensure that if a value is not present in the OCR text, it is returned as null."    
         )
         
         parser = PydanticOutputParser(pydantic_object=Candidate)
         
         prompt = PromptTemplate(
             template="""Your primary goal is to take my receipt OCR text and then return back a parsable json.
-            Below is the receipt OCR:.\n {raw_text} \n These are the format instructions telling you to convert the data into json :\n {format_instructions}\n Follow the below instrcution very strictly:\n {query} \n""",
+            Below is the receipt OCR:.\n {raw_text} \n These are the format instructions telling you to convert the data into json :\n {format_instructions}\nDo not include descriptions or explanations from the Candidate class in the JSON output. The response must be a valid JSON object.\n Follow the below instrcution very strictly:\n {query} \n""",
             input_variables=["query"],
             partial_variables={"format_instructions": parser.get_format_instructions(),"raw_text":raw_text},
         )