Update app.py

app.py

@@ -12,6 +12,7 @@ from huggingface_hub import Repository
 from datetime import datetime
 import scipy.ndimage.interpolation as inter
 import easyocr
+from datasets import load_dataset
 from PIL import Image
 from paddleocr import PaddleOCR
 import socket
@@ -20,30 +21,32 @@ from huggingface_hub import HfApi
 import smtplib
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
-mydataset_name = "pragnakalp/OCR-img-to-text"
-print("$$$$$$$$$$$$$$$$$$$$$$$$$$$$",type(mydataset_name))
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN,mydataset_name)
-
-
-# # DATASET_REPO_URL = "https://huggingface.co/datasets/pragnakalp/OCR-img-to-text"
-# DATA_FILENAME = "ocr_data.csv"
-# DATA_FILE = os.path.join("ocr_data", DATA_FILENAME)
-# HF_TOKEN = os.environ.get("HF_TOKEN")
-# DATASET_REPO_ID = "pragnakalp/OCR-img-to-text"
-# print("is none?", HF_TOKEN is None)
-# try:
-#     hf_hub_download(
-#         repo_id=DATASET_REPO_ID,
-#         filename=DATA_FILENAME,
-#         cache_dir=DATA_DIRNAME,
-#         force_filename=DATA_FILENAME
-#     )
-# except:
-#     print("file not found")
-
-# repo = Repository(
-#     local_dir="ocr_data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
-# )
+# mydataset_name = "pragnakalp/OCR-img-to-text"
+# print("$$$$$$$$$$$$$$$$$$$$$$$$$$$$",type(mydataset_name))
+# hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN,mydataset_name)
+
+
+# DATASET_REPO_URL = "https://huggingface.co/datasets/pragnakalp/OCR-img-to-text"
+DATA_FILENAME = "ocr_data.csv"
+DATA_FILE = os.path.join("ocr_data", DATA_FILENAME)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+DATASET_REPO_ID = "pragnakalp/OCR-img-to-text"
+print("is none?", HF_TOKEN is None)
+try:
+    hf_hub_download(
+        repo_id=DATASET_REPO_ID,
+        filename=DATA_FILENAME,
+        cache_dir=DATA_DIRNAME,
+        force_filename=DATA_FILENAME
+    )
+except:
+    print("file not found")
+
+repo = Repository(
+    local_dir="ocr_data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+)
+
+dataset = load_dataset("imagefolder", data_dir="pragnakalp/OCR-img-to-text", drop_labels=False)
 
 def get_device_ip_address():
 
@@ -128,31 +131,31 @@ def generate_ocr(Method,img):
         if Method == 'PaddleOCR':
             text_output = ocr_with_paddle(img)
 
-        # with open(DATA_FILE, "a") as csvfile:
-        #     writer = csv.DictWriter(csvfile, fieldnames=["method", "image", "generated_text"])
-        #     writer.writerow(
-        #         {"method": Method, "image": img, "generated_text": text_output}
-        #     )
-        # commit_url = repo.push_to_hub()
-        # print(commit_url)
+        with open(DATA_FILE, "a") as csvfile:
+            writer = csv.DictWriter(csvfile, fieldnames=["method", "image", "generated_text"])
+            writer.writerow(
+                {"method": Method, "image": img, "generated_text": text_output}
+            )
+        commit_url = repo.push_to_hub()
+        print(commit_url)
         
-        # save_details(Method,text_output,img)
-        # sender="pragnakalp.dev33@gmail.com"
-        # password="httscgatatbbxxur"
-        # reciever="pragnakalp.dev35@gmail.com"
-
-        # s = smtplib.SMTP('smtp.gmail.com', 587)
-        # s.starttls()
-        # s.ehlo()
-        # s.login(sender,password)
-
-        # message = """Subject : Appointment Booking\n\n
-        #         Hello,
-        # Your OCR generated successfully"""
-        # s.sendmail(sender, reciever, message)
-        # s.quit()
-        # mailsend=1
-        # print("Send mail successfully")
+        save_details(Method,text_output,img)
+        sender="pragnakalp.dev33@gmail.com"
+        password="httscgatatbbxxur"
+        reciever="pragnakalp.dev35@gmail.com"
+
+        s = smtplib.SMTP('smtp.gmail.com', 587)
+        s.starttls()
+        s.ehlo()
+        s.login(sender,password)
+
+        message = """Subject : Appointment Booking\n\n
+                Hello,
+        Your OCR generated successfully"""
+        s.sendmail(sender, reciever, message)
+        s.quit()
+        mailsend=1
+        print("Send mail successfully")
         return text_output
     
     except Exception as e:
@@ -220,8 +223,8 @@ demo = gr.Interface(
     title="Optical Character Recognition",
     description="Try OCR with different methods",
     css=".gradio-container {background-color: lightgray} #radio_div {background-color: #FFD8B4; font-size: 40px;}",
-    allow_flagging = "manual",
-    flagging_dir = "flagged",
-	flagging_callback=hf_writer
+    allow_flagging = "manual"
+ #    flagging_dir = "flagged",
+	# flagging_callback=hf_writer
 )
 demo.launch(enable_queue = False)