Update app.py

app.py

@@ -6,6 +6,7 @@ from PIL import Image
 from huggingface_hub import HfApi
 from os import listdir
 from os.path import isfile, join
+from io import BytesIO
 
 api = HfApi()
 repo_id = "wolf1997/receipt_scanner"  
@@ -304,14 +305,22 @@ def receipt_editor_app():
                 'receipt_data': st.session_state.edited_receipt
             })
             
-            with open("new_receipt_data/receipt_data_list.json", "w") as f:
-                json.dump(data_list, f, indent=2)
-            api.upload_file(path_or_fileobj=data_list, repo_id=repo_id,  path_in_repo="new_receipt_data/receipt_data_list.json")
+            # with open("new_receipt_data/receipt_data_list.json", "w") as f:
+            #     json.dump(data_list, f, indent=2)
+            json_string = json.dumps(data_list)
+            bytes_data = json_string.encode('utf-8')
+            api.upload_file(path_or_fileobj=bytes_data, repo_id=repo_id,  path_in_repo="new_receipt_data/receipt_data_list.json")
             # Save image if available
+
+            
             if st.session_state.receipt_image:
-                save_path = f'new_images/{st.session_state.image_name}'
-                st.session_state.receipt_image.save(save_path)
-                api.upload_file(path_or_fileobj=st.session_state.receipt_image, repo_id=repo_id,  path_in_repo=f'new_images/{st.session_state.image_name}')
+                # save_path = f'new_images/{st.session_state.image_name}'
+                # st.session_state.receipt_image.save(save_path)
+                buffered=BytesIO()
+                image=st.session_state.receipt_image
+                image.save(buffered, format='JPEG')
+                image_data = base64.b64encode(buffered.getvalue()).decode("utf-8")
+                api.upload_file(path_or_fileobj=image_data, repo_id=repo_id,  path_in_repo=f'new_images/{st.session_state.image_name}')
                 st.success(f"Receipt data and image saved successfully!")
             else:
                 st.success("Receipt data saved (no image available)")