update

main.py

@@ -10,8 +10,7 @@ from pathlib import Path
 import platform
 import time
 import tempfile
-# import uuid
-import shortuuid
+import hashlib
 
 from project_settings import project_path, log_directory
 import log
@@ -64,6 +63,15 @@ def build_html_output(s: str, style: str = "result_item_success"):
     """
 
 
+def md5_encrypt(text: str) -> str:
+    md = hashlib.md5()
+
+    md.update(text.encode())
+
+    result = md.hexdigest()
+    return result
+
+
 @torch.no_grad()
 def process(
     language: str,
@@ -106,16 +114,14 @@ def process(
     if len(repo_id.parts) == 1:
         repo_name = repo_id.parts[-1]
         if len(repo_name) > 40:
-            # repo_name = str(uuid.uuid4())
-            repo_name = str(shortuuid.uuid())
+            repo_name = md5_encrypt(repo_name)
         # repo_name = repo_name[:40]
         folder = repo_name
     elif len(repo_id.parts) == 2:
         repo_supplier = repo_id.parts[-2]
         repo_name = repo_id.parts[-1]
         if len(repo_name) > 40:
-            # repo_name = str(uuid.uuid4())
-            repo_name = str(shortuuid.uuid())
+            repo_name = md5_encrypt(repo_name)
         # repo_name = repo_name[:40]
         folder = "{}/{}".format(repo_supplier, repo_name)
     else: