修改检测是否同一个文件的方式为检测文件的md5哈希值

modules/llama_func.py

@@ -17,13 +17,17 @@ from modules.presets import *
 from modules.utils import *
 
 def get_index_name(file_src):
-    index_name = []
-    for file in file_src:
-        index_name.append(os.path.basename(file.name))
-    index_name = sorted(index_name)
-    index_name = "".join(index_name)
-    index_name = sha1sum(index_name)
-    return index_name
+    file_paths = [x.name for x in file_src]
+    file_paths.sort(key=lambda x: os.path.basename(x))
+
+    md5_hash = hashlib.md5()
+    for file_path in file_paths:
+        with open(file_path, "rb") as f:
+            while chunk := f.read(8192):
+                md5_hash.update(chunk)
+
+    return md5_hash.hexdigest()
+
 
 def get_documents(file_src):
     documents = []

modules/utils.py

@@ -367,12 +367,6 @@ def submit_key(key):
     return key, msg
 
 
-def sha1sum(filename):
-    sha1 = hashlib.sha1()
-    sha1.update(filename.encode("utf-8"))
-    return sha1.hexdigest()
-
-
 def replace_today(prompt):
     today = datetime.datetime.today().strftime("%Y-%m-%d")
     return prompt.replace("{current_date}", today)