update helper

helper.py

@@ -208,7 +208,7 @@ def add_comma_after_pattern_ti(text):
     modified_text = pattern.sub(lambda x: x.group() + ',', text)
     return modified_text
 
-def save_image(img):
+def save_image(img, flag):
     path = "./tmp/"
     
     # Check if the input is a string (file path) and load the image if it is
@@ -218,9 +218,10 @@ def save_image(img):
     # Ensure the Hugging Face path exists locally
     if not os.path.exists(path):
         os.makedirs(path)
-    
+    if flag:
+        flag = "_" + flag
     # Generate a unique filename
-    unique_name = str(uuid.uuid4()) + ".webp"
+    unique_name = str(uuid.uuid4()) + flag + ".webp"
     unique_name = os.path.join(path, unique_name)
     
     # Convert the image to WebP format