update

app.py

@@ -8,6 +8,8 @@ from PIL import Image
 import sys, os
 from rembg import remove
 from utils import functions
+from utils.functions import complete, clean, clean_by_name, get_random_name
+
 import os, sys
 import argparse
 
@@ -18,43 +20,8 @@ logger.setLevel(DEBUG)
 logger.addHandler(handler)
 logger.propagate = False
 
-# logger.debug('variable')
-# logger.info('system')
-
-
-
-def complete(work_dir):
-    work_dir = work_dir
-    # logger.debug(f"complete :", work_dir)
-    return work_dir
-
-
-def clean(text_output):
-    # logger.debug(f"text_output : {text_output}")
-
-    if text_output!="idle_state":
-        logger.info(f"clean up : {text_output}.zip")
-        os.remove(f"{text_output}.zip")
-        return "idle_state"
-    else:
-        logger.info(f"reset")
-        return "idle_state"
-
-
-
-def clean_by_name(text_output):
-    # logger.debug(f"text_output : {text_output}")
-    if text_output!="idle_state":
-        text_output, dir_name = text_output.split("+")
-        logger.info(f"clean up : {dir_name}.zip")
-        os.remove(f"{dir_name}.zip")
-        return "idle_state"
-    else:
-        logger.info(f"reset")
-        return "idle_state"
-    
-    
 
+        
 def run_rembg(img):
     output = remove(img)
     output_pil = Image.fromarray(output)
@@ -66,7 +33,7 @@ def run_rembg(img):
 
 
 def from_zip(inputs):
-    work_dir = functions.get_random_name()
+    work_dir = get_random_name()
     os.makedirs(work_dir, exist_ok=True)
 
     image_data_dict = {}
@@ -90,7 +57,7 @@ def from_zip(inputs):
                         image_files.append(image_name)
                         
                         image_array = np.array(image)
-                        logger.debug(f"image name : {image_name}")
+                        # logger.debug(f"image name : {image_name}")
                         category_dir = image_name.split("/")[0]
                         # image_name = image_name.split("/")[1]
                         os.makedirs(f"{work_dir}/{category_dir}", exist_ok=True)

utils/functions.py

@@ -1,4 +1,11 @@
-import random
+import os, random
+from logging import getLogger, StreamHandler, DEBUG
+logger = getLogger(__name__)
+handler = StreamHandler(); handler.setLevel(DEBUG)
+logger.setLevel(DEBUG)
+logger.addHandler(handler)
+logger.propagate = False
+
 
 def get_random_name():
     famous_painters = [
@@ -31,3 +38,33 @@ def get_random_name():
 
     return random_painter+str(rand_num)
 
+
+def complete(work_dir):
+    work_dir = work_dir
+    # logger.debug(f"complete :", work_dir)
+    return work_dir
+
+
+def clean(text_output):
+    # logger.debug(f"text_output : {text_output}")
+
+    if text_output!="idle_state":
+        logger.info(f"clean up : {text_output}.zip")
+        os.remove(f"{text_output}.zip")
+        return "idle_state"
+    else:
+        logger.info(f"reset")
+        return "idle_state"
+
+
+
+def clean_by_name(text_output):
+    # logger.debug(f"text_output : {text_output}")
+    if text_output!="idle_state":
+        text_output, dir_name = text_output.split("+")
+        logger.info(f"clean up : {dir_name}.zip")
+        os.remove(f"{dir_name}.zip")
+        return "idle_state"
+    else:
+        logger.info(f"reset")
+        return "idle_state"