Update gradio_app.py

gradio_app.py

@@ -4,6 +4,7 @@ import shutil
 import time
 from glob import glob
 from pathlib import Path
+import uuid
 
 import gradio as gr
 import torch
@@ -56,17 +57,37 @@ def get_example_mv_list():
     return mv_list
 
 
-def gen_save_folder(max_size=60):
+# def gen_save_folder(max_size=60):
+#     os.makedirs(SAVE_DIR, exist_ok=True)
+#     exists = set(int(_) for _ in os.listdir(SAVE_DIR) if _.isdigit())
+#     cur_id = min(set(range(max_size)) - exists) if len(exists) < max_size else -1
+#     if os.path.exists(f"{SAVE_DIR}/{(cur_id + 1) % max_size}"):
+#         shutil.rmtree(f"{SAVE_DIR}/{(cur_id + 1) % max_size}")
+#         print(f"remove {SAVE_DIR}/{(cur_id + 1) % max_size} success !!!")
+#     save_folder = f"{SAVE_DIR}/{max(0, cur_id)}"
+#     os.makedirs(save_folder, exist_ok=True)
+#     print(f"mkdir {save_folder} suceess !!!")
+#     return save_folder
+
+def gen_save_folder(max_size=200):
     os.makedirs(SAVE_DIR, exist_ok=True)
-    exists = set(int(_) for _ in os.listdir(SAVE_DIR) if _.isdigit())
-    cur_id = min(set(range(max_size)) - exists) if len(exists) < max_size else -1
-    if os.path.exists(f"{SAVE_DIR}/{(cur_id + 1) % max_size}"):
-        shutil.rmtree(f"{SAVE_DIR}/{(cur_id + 1) % max_size}")
-        print(f"remove {SAVE_DIR}/{(cur_id + 1) % max_size} success !!!")
-    save_folder = f"{SAVE_DIR}/{max(0, cur_id)}"
-    os.makedirs(save_folder, exist_ok=True)
-    print(f"mkdir {save_folder} suceess !!!")
-    return save_folder
+
+    # 获取所有文件夹路径
+    dirs = [f for f in Path(SAVE_DIR).iterdir() if f.is_dir()]
+
+    # 如果文件夹数量超过 max_size，删除创建时间最久的文件夹
+    if len(dirs) >= max_size:
+        # 按创建时间排序，最久的排在前面
+        oldest_dir = min(dirs, key=lambda x: x.stat().st_ctime)
+        shutil.rmtree(oldest_dir)
+        print(f"Removed the oldest folder: {oldest_dir}")
+
+    # 生成一个新的 uuid 文件夹名称
+    new_folder = os.path.join(SAVE_DIR, str(uuid.uuid4()))
+    os.makedirs(new_folder, exist_ok=True)
+    print(f"Created new folder: {new_folder}")
+
+    return new_folder
 
 
 def export_mesh(mesh, save_folder, textured=False, type='glb'):