Update receipt_gen_agent.py

receipt_gen_agent.py

@@ -28,7 +28,7 @@ from os.path import isfile, join
 
 class State(TypedDict):
     prompt: str
-    image_number: int
+    # image_number: int
     image_data: json
     image_byte: list[str]
     eval: dict
@@ -254,18 +254,18 @@ class receipt_agent:
     def receipt_gen(self, images):
         config = {"configurable": {"thread_id": "1"}}
         
-        data_list = [f for f in listdir('new_receipt_data') if isfile(join('new_receipt_data', f))]
-        if not data_list:
-            data_list=[]
-        else:
-            with open(f'new_receipt_data/{data_list[0]}', 'r') as openfile:
-                # Reading from json file
-                data_list = json.load(openfile)
+        # data_list = [f for f in listdir('new_receipt_data') if isfile(join('new_receipt_data', f))]
+        # if not data_list:
+        #     data_list=[]
+        # else:
+        #     with open(f'new_receipt_data/{data_list[0]}', 'r') as openfile:
+        #         # Reading from json file
+        #         data_list = json.load(openfile)
         
         response = self.agent.invoke({
             'prompt': 'analyse this receipt and list the items, return a json',
             'n_retries': 0,
-            'image_number': len(data_list),
+            # 'image_number': len(data_list),
             'image_byte': images  # Now passing the list of images directly
         }, config)