Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 import fitz
 import re
 import gradio as gr
-
+import time
 
 
 
@@ -64,16 +64,28 @@ llm = OpenAI(temperature=0.5,max_tokens=1024)
 
 
 
+
+prompt='''
+请根据给出的法律条文回答问题，给出适当的法律建议。回答时要说出你引用的法律条文是第几条，并说出引用的每一条是哪部法律中的。
+引用的法律条文不要超过两条，回答尽量简明扼要
+如果问题与搜索结果无关，就仅回答"该问题与青少年法律无关"即可。
+'''
+
+
 def generate_answer(question):
-    prompt='''
-    请根据给出的法律条文回答问题，给出适当的法律建议。回答时要说出你引用的法律条文是第几条，并说出引用的每一条是哪部法律中的。
-    引用的法律条文不要超过两条，回答尽量简明扼要
-    如果问题与搜索结果无关，就仅回答"该问题与青少年法律无关"即可。
-    '''
+    start_time = time.time()
     most_relevant_texts = vectordb.max_marginal_relevance_search(question, k=5)
+    print(most_relevant_texts)
     chain = load_qa_chain(llm)
     answer = chain.run(input_documents=most_relevant_texts, question=question+prompt)
-    return answer
+    
+    run_time = time.time() - start_time
+    return (answer,
+            '\n'.join([t.page_content for t in most_relevant_texts]),
+            run_time
+           )                  
+
+
     
 def ask_api(question):
     
@@ -84,22 +96,40 @@ def ask_api(question):
 
 
 
-title = '青少年法律科普问答'
-description = """ 本bot旨在根据中华人民共和国的法律回答有关青少年的问题，目前囊括的法律有\n
-《未成年人保护法》\n
-《义务教育法》\n
-《预防未成年人犯罪法》\n
-《妇女儿童权益保护法》
-"""
 
 
 
-gr.Interface(
-    title=title,
-    description=description,
-    fn=ask_api, 
-    inputs=gr.Textbox(label="请输入与青少年法律相关的问题",lines=2), 
-    outputs=gr.outputs.Textbox(label="参考回答"), 
-    examples=[["未成年遭受网络欺凌该怎么办？"],['年满多少岁的儿童应当接受义务教育？'],['若发现离家出走的未成年人，应如何处理？']]).launch()
+with gr.Blocks() as demo:
+    gr.Markdown(
+    """
+    # 青少年法律科普问答
+    本bot旨在根据中华人民共和国的法律回答有关青少年的问题，目前囊括的法律有
+
+   《未成年人保护法》
+
+   《义务教育法》
+
+   《预防未成年人犯罪法》
+
+   《妇女儿童权益保护法》
+    """)
+    
+    with gr.Column():
+        text_input = gr.Textbox(label='请输入与青少年法律相关的问题')
+        text_button = gr.Button("提交")
+        gr.Examples(
+        [["未成年遭受网络欺凌该怎么办？"],['年满多少岁的儿童应当接受义务教育？'],['遇到离家出走的未成年应该如何处理？']],
+        [text_input],
+        label='示例问题'
+
+    )
+        text_output = [gr.Textbox(label='参考回答'),
+                       gr.Textbox(label='相关法律原文'),
+                       gr.Number(label="运行时长(s)")]
+        
+
+    text_button.click(generate_answer, inputs=text_input, outputs=text_output)
+
+demo.launch()