Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -138,16 +138,33 @@ def remove_text_dynamic_fill(img_path, output_path):
|
|
138 |
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
139 |
cv2.imwrite(output_path, image)
|
140 |
|
|
|
|
|
141 |
def process_folder(input_files):
|
142 |
temp_output = tempfile.mkdtemp()
|
|
|
143 |
for file in input_files:
|
144 |
filename = os.path.basename(file.name)
|
145 |
output_path = os.path.join(temp_output, filename)
|
146 |
remove_text_dynamic_fill(file.name, output_path)
|
147 |
-
|
|
|
|
|
|
|
|
|
|
|
148 |
return zip_path
|
149 |
|
150 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
151 |
|
152 |
|
153 |
|
|
|
138 |
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
|
139 |
cv2.imwrite(output_path, image)
|
140 |
|
141 |
+
import uuid
|
142 |
+
|
143 |
def process_folder(input_files):
|
144 |
temp_output = tempfile.mkdtemp()
|
145 |
+
|
146 |
for file in input_files:
|
147 |
filename = os.path.basename(file.name)
|
148 |
output_path = os.path.join(temp_output, filename)
|
149 |
remove_text_dynamic_fill(file.name, output_path)
|
150 |
+
|
151 |
+
unique_name = str(uuid.uuid4())[:8]
|
152 |
+
zip_path = os.path.join("/tmp", f"cleaned_output_{unique_name}.zip")
|
153 |
+
shutil.make_archive(zip_path.replace(".zip", ""), 'zip', temp_output)
|
154 |
+
|
155 |
+
delayed_cleanup(zip_path)
|
156 |
return zip_path
|
157 |
|
158 |
|
159 |
+
import threading
|
160 |
+
import time
|
161 |
+
|
162 |
+
def delayed_cleanup(path, delay=30):
|
163 |
+
def cleanup():
|
164 |
+
time.sleep(delay)
|
165 |
+
if os.path.exists(path):
|
166 |
+
os.remove(path)
|
167 |
+
threading.Thread(target=cleanup).start()
|
168 |
|
169 |
|
170 |
|