Update app.py

app.py

@@ -14,8 +14,8 @@ from paddleocr import PaddleOCR
 import socket
 from send_email_user import send_user_email
 
-# if not os.path.isdir('images'):
-#     os.mkdir('images')
+if not os.path.isdir('images'):
+    os.mkdir('images')
 
 def get_device_ip_address():
 
@@ -101,7 +101,7 @@ def generate_ocr(Method,img):
             text_output = ocr_with_keras(img)
         if Method == 'PaddleOCR':
             text_output = ocr_with_paddle(img)
-        # save_details(Method,text_output,img)
+        save_details(Method,text_output,img)
 
         return text_output
         # hostname = socket.gethostname()
@@ -115,32 +115,32 @@ def generate_ocr(Method,img):
 """
 Save generated details
 """
-# def save_details(Method,text_output,img):
-#     method = []
-#     img_path = []
-#     text = []
-#     picture_path = "image.jpg"    
-#     curr_datetime = datetime.now().strftime('%Y-%m-%d %H-%M-%S')
-#     if text_output:
-#         splitted_path = os.path.splitext(picture_path)
-#         modified_picture_path = splitted_path[0] + curr_datetime + splitted_path[1]
-#         cv2.imwrite('images/'+ modified_picture_path, img)
-#         input_img = 'images/'+ modified_picture_path
-#         try:
-#             df = pd.read_csv("AllDetails.csv")
-#             df2 = {'method': Method, 'input_img': input_img, 'generated_text': text_output}
-#             df = df.append(df2, ignore_index = True)
-#             df.to_csv("AllDetails.csv", index=False)
-#         except:
-#             method.append(Method)
-#             img_path.append(input_img)
-#             text.append(text_output)
-#             dict = {'method': method, 'input_img': img_path, 'generated_text': text}
-#             df = pd.DataFrame(dict,index=None)
-#             df.to_csv("AllDetails.csv")
+def save_details(Method,text_output,img):
+    method = []
+    img_path = []
+    text = []
+    picture_path = "image.jpg"    
+    curr_datetime = datetime.now().strftime('%Y-%m-%d %H-%M-%S')
+    if text_output:
+        splitted_path = os.path.splitext(picture_path)
+        modified_picture_path = splitted_path[0] + curr_datetime + splitted_path[1]
+        cv2.imwrite('images/'+ modified_picture_path, img)
+        input_img = 'images/'+ modified_picture_path
+        try:
+            df = pd.read_csv("AllDetails.csv")
+            df2 = {'method': Method, 'input_img': input_img, 'generated_text': text_output}
+            df = df.append(df2, ignore_index = True)
+            df.to_csv("AllDetails.csv", index=False)
+        except:
+            method.append(Method)
+            img_path.append(input_img)
+            text.append(text_output)
+            dict = {'method': method, 'input_img': img_path, 'generated_text': text}
+            df = pd.DataFrame(dict,index=None)
+            df.to_csv("AllDetails.csv")
 
-#         hostname = get_device_ip_address()
-#     return send_user_email(input_img,hostname,text_output,Method)
+        hostname = get_device_ip_address()
+    return send_user_email(input_img,hostname,text_output,Method)
     # return hostname
 
 """