Upload 8 files

app.py

@@ -92,7 +92,8 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                     "🧹 新的对话",
                 )
                 retryBtn = gr.Button("🔄 重新生成")
-                delLastBtn = gr.Button("🗑️ 删除一条对话")
+                delFirstBtn = gr.Button("🗑️ 删除最旧对话")
+                delLastBtn = gr.Button("🗑️ 删除最新对话")
                 reduceTokenBtn = gr.Button("♻️ 总结对话")
 
         with gr.Column():
@@ -306,6 +307,12 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     )
     retryBtn.click(**get_usage_args)
 
+    delFirstBtn.click(
+        delete_first_conversation,
+        [history, token_count],
+        [history, token_count, status_display],
+    )
+
     delLastBtn.click(
         delete_last_conversation,
         [chatbot, history, token_count],

modules/llama_func.py

@@ -17,9 +17,11 @@ from modules.presets import *
 from modules.utils import *
 
 def get_index_name(file_src):
-    index_name = ""
+    index_name = []
     for file in file_src:
-        index_name += os.path.basename(file.name)
+        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
 

modules/utils.py

@@ -153,6 +153,16 @@ def construct_assistant(text):
 def construct_token_message(token, stream=False):
     return f"Token 计数: {token}"
 
+def delete_first_conversation(history, previous_token_count):
+    if history:
+        del history[:2]
+        del previous_token_count[0]
+    return (
+        history,
+        previous_token_count,
+        construct_token_message(sum(previous_token_count)),
+    )
+
 
 def delete_last_conversation(chatbot, history, previous_token_count):
     if len(chatbot) > 0 and standard_error_msg in chatbot[-1][1]: