测试gradio配置

src/config/consts.py

@@ -42,3 +42,9 @@ INDEX_NAME = 'know'
 # 知识相关资料配置
 # =============================
 KNOWLEDGE_PATH = 'C:\\Users\\15663\\Desktop\\低空经济数据库'
+
+# =============================
+# gradio服务相关配置
+# =============================
+GR_SERVER_NAME = 'localhost'
+GR_PORT = 8888
\ No newline at end of file

src/server/get_similarity.py

@@ -5,7 +5,7 @@ class GetSimilarity:
     def __init__(self, _question, _faiss_db: VectorStore_FAISS):
         self.question = _question
         self.faiss_db = _faiss_db
-        self.similarity_doc = self.faiss_db.join_document(self.faiss_db.get_text_similarity("什么是低空飞行"))
+        self.similarity_doc = self.faiss_db.join_document(self.faiss_db.get_text_similarity(self.question))
 
     def get_similarity_doc(self):
         return self.similarity_doc

test/gradio_text.py

@@ -20,7 +20,9 @@ from src.config.consts import (
     VEC_DB_PORT,
     VEC_DB_USER,
     VEC_DB_DBNAME,
-    SIMILARITY_SHOW_NUMBER
+    SIMILARITY_SHOW_NUMBER,
+    GR_PORT,
+    GR_SERVER_NAME
 )
 from src.server.qa import QA
 
@@ -31,8 +33,9 @@ prompt1 = """'''
 {question}"""
 
 PROMPT1 = PromptTemplate(input_variables=["context", "question"], template=prompt1)
-def main():
 
+
+def main():
     c_db = UPostgresDB(host=CHAT_DB_HOST, database=CHAT_DB_DBNAME, user=CHAT_DB_USER, password=CHAT_DB_PASSWORD,
                        port=CHAT_DB_PORT, )
     vecstore_faiss = VectorStore_FAISS(
@@ -45,16 +48,16 @@ def main():
         reset=False)
     base_llm = ChatERNIESerLLM(
         chat_completion=ChatCompletion(ak="pT7sV1smp4AeDl0LjyZuHBV9", sk="b3N0ibo1IKTLZlSs7weZc8jdR0oHjyMu"))
-    my_chat = QA(PROMPT1, base_llm, {"temperature": 0.9}, ['context', 'question'], _db=c_db, _chat_id='2', _faiss_db=vecstore_faiss)
+    my_chat = QA(PROMPT1, base_llm, {"temperature": 0.9}, ['context', 'question'], _db=c_db, _chat_id='2',
+                 _faiss_db=vecstore_faiss)
     with gr.Blocks() as demo:
         with gr.Row():
-            in1 = gr.Textbox(show_label=True, lines=10, visible=False)
-            in2 = gr.Textbox(show_label=True, lines=10)
+            inn = gr.Textbox(show_label=True, lines=10)
         with gr.Row():
             qabtn = gr.Button("SUBMIT")
         out = gr.Textbox(show_label=True, lines=10)
-        qabtn.click(my_chat.async_chat, [in2], [out])
-    demo.queue().launch(share=False, inbrowser=True, server_name="192.168.100.76", server_port=8888)
+        qabtn.click(my_chat.async_chat, [inn], [out])
+    demo.queue().launch(share=False, inbrowser=True, server_name=GR_SERVER_NAME, server_port=GR_PORT)
 
 
 if __name__ == "__main__":