Upload app.py

app.py

@@ -15,6 +15,7 @@ import torch
 import os
 import numpy as np
 import json
+import time
 
 cache_dir = '/data/KB'
 os.makedirs(cache_dir, exist_ok=True)
@@ -65,7 +66,7 @@ def add_pdf_gradio(pdf_file_list, progress=gr.Progress()):
 
     print(pdf_file_list)
     
-    pdf_file_list = sorted(pdf_file_list)
+    pdf_file_list = sorted(pdf_file_list, key=lambda x: os.path.basename(x))
     knowledge_base_name = str(int(time.time()))
     this_cache_dir = os.path.join(cache_dir, knowledge_base_name)
     os.makedirs(this_cache_dir, exist_ok=True)