Update app.py

app.py

@@ -1,5 +1,6 @@
 import numpy as np
 import cv2
+import csv
 import imutils
 import pytesseract
 import pandas as pd
@@ -9,6 +10,13 @@ import os
 from PIL import Image
 import streamlit as st
 
+#
+# Create a directory for storing data
+if not os.path.exists("data"):
+    os.makedirs("data")
+
+
+
 bytes_data = None
 
 img_file_buffer = st.camera_input("Take a picture")
@@ -62,13 +70,14 @@ if img_file_buffer is not None:
     file_path = os.path.join("data", "data.csv")
     if os.path.exists(file_path):
         # Append the data to the existing CSV file
-        with open(file_path, "a") as f:
-            df.to_csv(f, header=False, index=False)
+        with open(file_path, mode='a', newline='') as file:
+            writer = csv.writer(file)
+            if os.stat(file_path).st_size == 0:
+                writer.writerow(df.columns)
+            writer.writerow(df.iloc[0])
     else:
         # Create a new CSV file and write the data
-        with open(file_path, "w") as f:
-            df.to_csv(f, index=False)
-
+        df.to_csv(file_path, index=False)
 
 
     # Print recognized text