Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 import zipfile
 import uuid
 import gradio as gr
-
+import uuid
 
 
 def remove_watermark_area(original_image, text_mask_path):
@@ -60,17 +60,24 @@ def make_zip(image_list):
             zipf.write(image, os.path.basename(image))
     return zip_path
 
+def random_image_name():
+    """Generate a random image name."""
+    return str(uuid.uuid4())[:8]
+
+    
 def process_files(image_files):
     image_list = []
     if len(image_files) == 1:
-        saved_path = os.path.basename(image_files[0])
-        saved_path = f"./temp/{saved_path}"
+        # saved_path = os.path.basename(image_files[0])
+        # saved_path = f"./temp/{saved_path}"
+        saved_path = f"./temp/{random_image_name()}.jpg"
         remove_watermark(image_files[0], saved_path)
         return saved_path, saved_path
     else:
         for image_path in image_files:
-            saved_path = os.path.basename(image_path)
-            saved_path = f"./temp/{saved_path}"
+            # saved_path = os.path.basename(image_path)
+            # saved_path = f"./temp/{saved_path}"
+            saved_path = f"./temp/{random_image_name()}.jpg"
             remove_watermark(image_path, saved_path)
             image_list.append(saved_path)
         zip_path = make_zip(image_list)