Update app.py

app.py

@@ -39,10 +39,26 @@ prompt_template = """
     The above content is a form of biological background knowledge. Please answer the questions according to the above content. Please be sure to answer the questions according to the background knowledge and attach the doi number of the information source when answering.
     Question: {question}
     Answer in English:"""
-    PROMPT = PromptTemplate(
-        template=prompt_template, input_variables=["context", "question"]
+PROMPT = PromptTemplate(
+    template=prompt_template, input_variables=["context", "question"]
+)
+chain_type_kwargs = {"prompt": PROMPT}
+retriever = db.as_retriever()
+qa = RetrievalQA.from_chain_type(
+        llm=llm,
+        chain_type="stuff",
+        retriever=retriever,
+        chain_type_kwargs=chain_type_kwargs,
+        return_source_documents=True
     )
-    chain_type_kwargs = {"prompt": PROMPT}
+
+def chat(question, history):
+    if len(history) == 0:
+        response = qa.invoke(question)['result']
+    else:
+        response = qwen_api(question, gradio_history=history)
+    return response
+
 
 def respond(
     message,