信息读取改为文件

app.py

@@ -3,7 +3,7 @@ from typing import Callable, Optional
 import gradio as gr
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores import Zilliz
-from langchain.document_loaders import WebBaseLoader
+from langchain.document_loaders import TextLoader
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.chains import RetrievalQAWithSourcesChain
 import uuid
@@ -12,10 +12,10 @@ from project.llm.zhipuai_llm import ZhipuAILLM
 chain: Optional[Callable] = None
 
 
-def web_loader(url_list, openai_key, puzhiai_key, zilliz_uri, user, password):
-    if not url_list:
-        return "please enter url list"
-    loader = WebBaseLoader(url_list.split())
+def web_loader(file, openai_key, puzhiai_key, zilliz_uri, user, password):
+    if not file:
+        return "please upload file"
+    loader = TextLoader(file)
     docs = loader.load()
 
     text_splitter = CharacterTextSplitter(chunk_size=1024, chunk_overlap=0)
@@ -88,11 +88,13 @@ if __name__ == "__main__":
         6. 🔍 Click the Generate button to search for the answer to the question. The final answer will be displayed in the question answer text box.
         """
         )
-        url_list_text = gr.Textbox(
-            label="url list",
-            lines=3,
-            placeholder="https://milvus.io/docs/overview.md",
-        )
+        # url_list_text = gr.Textbox(
+        #     label="url list",
+        #     lines=3,
+        #     placeholder="https://milvus.io/docs/overview.md",
+        # )
+        file = gr.File(label='请上传知识库文件',
+                       file_types=['.txt', '.md', '.docx', '.pdf'])
         openai_key_text = gr.Textbox(label="openai api key", type="password", placeholder="sk-******")
         puzhiai_key_text = gr.Textbox(label="puzhi api key", type="password", placeholder="******")
         with gr.Row():
@@ -109,7 +111,7 @@ if __name__ == "__main__":
         loader_btn.click(
             fn=web_loader,
             inputs=[
-                url_list_text,
+                file,
                 openai_key_text,
                 puzhiai_key_text,
                 zilliz_uri_text,