Update app.py

app.py

@@ -15,8 +15,8 @@ import socket
 from send_email_user import send_user_email
 
 # if not os.path.isdir('images'):
-os.mkdir('images')
-print("create folder--->")
+# os.mkdir('images')
+# print("create folder--->")
 def get_device_ip_address():
 
     if os.name == "nt":
@@ -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,34 +115,34 @@ def generate_ocr(Method,img):
 """
 Save generated details
 """
-# def save_details(Method,text_output,img):
-#     method = []
-#     img_path = []
-#     text = []
-#     input_img = ''
-#     hostname = ''
-#     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 = []
+    input_img = ''
+    hostname = ''
+    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(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
 
 """