Update app.py

app.py

@@ -6,7 +6,6 @@ import spaces
 from transformers import AutoModelForImageSegmentation
 import torch
 from torchvision import transforms
-import uuid  # Thêm thư viện uuid để tạo UUID
 
 torch.set_float32_matmul_precision(["high", "highest"][0])
 
@@ -22,20 +21,16 @@ transform_image = transforms.Compose(
     ]
 )
 
-def create_session_folder():
-    """Tạo thư mục duy nhất dựa trên UUID cho mỗi phiên làm việc."""
-    session_id = str(uuid.uuid4())  # Tạo UUID duy nhất
-    session_folder = os.path.join('output_images', session_id)
-    os.makedirs(session_folder, exist_ok=True)
-    return session_folder, session_id
+output_folder = 'output_images'
+if not os.path.exists(output_folder):
+    os.makedirs(output_folder)
 
 def fn(image):
-    session_folder, session_id = create_session_folder()
     im = load_img(image, output_type="pil")
     im = im.convert("RGB")
     origin = im.copy()
     image = process(im)    
-    image_path = os.path.join(session_folder, "no_bg_image.png")
+    image_path = os.path.join(output_folder, "no_bg_image.png")
     image.save(image_path)
     return (image, origin), image_path
 
@@ -53,8 +48,7 @@ def process(image):
     return image
   
 def process_file(f):
-    session_folder, session_id = create_session_folder()
-    name_path = os.path.join(session_folder, os.path.basename(f).rsplit(".", 1)[0] + ".png")
+    name_path = f.rsplit(".",1)[0]+".png"
     im = load_img(f, output_type="pil")
     im = im.convert("RGB")
     transparent = process(im)
@@ -64,10 +58,11 @@ def process_file(f):
 slider1 = ImageSlider(label="RMBG-2.0", type="pil")
 slider2 = ImageSlider(label="RMBG-2.0", type="pil")
 image = gr.Image(label="Upload an image")
-image2 = gr.Image(label="Upload an image", type="filepath")
+image2 = gr.Image(label="Upload an image",type="filepath")
 text = gr.Textbox(label="Paste an image URL")
 png_file = gr.File(label="output png file")
 
+
 chameleon = load_img("giraffe.jpg", output_type="pil")
 
 url = "http://farm9.staticflickr.com/8488/8228323072_76eeddfea3_z.jpg"
@@ -79,6 +74,7 @@ tab1 = gr.Interface(
 tab2 = gr.Interface(fn, inputs=text, outputs=[slider2, gr.File(label="output png file")], examples=[url], api_name="text")
 tab3 = gr.Interface(process_file, inputs=image2, outputs=png_file, examples=["giraffe.jpg"], api_name="png")
 
+
 demo = gr.TabbedInterface(
     [tab1, tab2], ["input image", "input url"], title="RMBG-2.0 for background removal"
 )