文件上传功能

app.py

@@ -54,11 +54,14 @@ def doc_vectorize(texts):
     return docsearch
 
 #文本被拆解储存在数组texts中
+# raw_file = 
 texts = doc_split("./资治通鉴_1_残缺.txt")
-title = "资治通鉴"
 docsearch = doc_vectorize((texts))
 
-def openai_reply(word1, word2, word3, temp):
+def openai_reply(word1, word2, word3, temp, file):
+    if file:
+        texts = doc_split(file)
+        docsearch = doc_vectorize((texts))
     words = word1 + "*****" + word2 + "*****" + word3
     # 文本相似查找，最终结果是一个列表
     docs = docsearch.similarity_search(words)
@@ -121,7 +124,7 @@ with gr.Blocks() as demo:
         
         btn.click(
             openai_reply,
-            inputs= [raw_text_1, raw_text_2, raw_text_3, temp],
+            inputs= [raw_text_1, raw_text_2, raw_text_3, temp, file],
             outputs= [answer, reference, shijian],
         )