update app.py

app.py

@@ -10,6 +10,7 @@ os.system("python -m mindsearch.app --lang cn --model_format internlm_silicon &"
 PLANNER_HISTORY = []
 SEARCHER_HISTORY = []
 
+
 def rst_mem(history_planner: list, history_searcher: list):
     '''
     Reset the chatbot memory.
@@ -20,8 +21,11 @@ def rst_mem(history_planner: list, history_searcher: list):
         PLANNER_HISTORY.clear()
     return history_planner, history_searcher
 
+
 def format_response(gr_history, agent_return):
-    if agent_return['state'] in [AgentStatusCode.STREAM_ING, AgentStatusCode.ANSWER_ING]:
+    if agent_return['state'] in [
+            AgentStatusCode.STREAM_ING, AgentStatusCode.ANSWER_ING
+    ]:
         gr_history[-1][1] = agent_return['response']
     elif agent_return['state'] == AgentStatusCode.PLUGIN_START:
         thought = gr_history[-1][1].split('```')[0]
@@ -30,18 +34,25 @@ def format_response(gr_history, agent_return):
     elif agent_return['state'] == AgentStatusCode.PLUGIN_END:
         thought = gr_history[-1][1].split('```')[0]
         if isinstance(agent_return['response'], dict):
-            gr_history[-1][1] = thought + '\n' + f'```json\n{json.dumps(agent_return["response"], ensure_ascii=False, indent=4)}\n```'
+            gr_history[-1][
+                1] = thought + '\n' + f'```json\n{json.dumps(agent_return["response"], ensure_ascii=False, indent=4)}\n```'  # noqa: E501
     elif agent_return['state'] == AgentStatusCode.PLUGIN_RETURN:
         assert agent_return['inner_steps'][-1]['role'] == 'environment'
         item = agent_return['inner_steps'][-1]
-        gr_history.append([None, f"```json\n{json.dumps(item['content'], ensure_ascii=False, indent=4)}\n```"])
+        gr_history.append([
+            None,
+            f"```json\n{json.dumps(item['content'], ensure_ascii=False, indent=4)}\n```"
+        ])
         gr_history.append([None, ''])
     return
 
+
 def predict(history_planner, history_searcher):
 
     def streaming(raw_response):
-        for chunk in raw_response.iter_lines(chunk_size=8192, decode_unicode=False, delimiter=b'\n'):
+        for chunk in raw_response.iter_lines(chunk_size=8192,
+                                             decode_unicode=False,
+                                             delimiter=b'\n'):
             if chunk:
                 decoded = chunk.decode('utf-8')
                 if decoded == '\r':
@@ -60,7 +71,11 @@ def predict(history_planner, history_searcher):
     url = 'http://localhost:8002/solve'
     headers = {'Content-Type': 'application/json'}
     data = {'inputs': PLANNER_HISTORY}
-    raw_response = requests.post(url, headers=headers, data=json.dumps(data), timeout=20, stream=True)
+    raw_response = requests.post(url,
+                                 headers=headers,
+                                 data=json.dumps(data),
+                                 timeout=20,
+                                 stream=True)
 
     for resp in streaming(raw_response):
         agent_return, node_name = resp
@@ -83,29 +98,30 @@ def predict(history_planner, history_searcher):
             yield history_planner, history_searcher
     return history_planner, history_searcher
 
-with gr.Blocks(css=".button-primary { background-color: #4CAF50; color: white; border-radius: 8px; border: none; padding: 10px 20px; font-size: 16px; cursor: pointer; transition: background-color 0.3s ease; } .button-primary:hover { background-color: #45a049; } .button-secondary { background-color: #f44336; color: white; border-radius: 8px; border: none; padding: 10px 20px; font-size: 16px; cursor: pointer; transition: background-color 0.3s ease; } .button-secondary:hover { background-color: #e53935; }") as demo:
-    gr.HTML("""<h1 align="center" style="font-family: 'Arial', sans-serif; color: #4A90E2;">MindSearch Gradio Demo</h1>""")
-    gr.HTML("""<p style="text-align: center; font-family: Arial, sans-serif; color: #333; max-width: 800px; margin: 0 auto;">MindSearch is an open-source AI Search Engine Framework with Perplexity.ai Pro performance. You can deploy your own Perplexity.ai-style search engine using either closed-source LLMs (GPT, Claude) or open-source LLMs (InternLM2.5-7b-chat).</p>""")
+
+with gr.Blocks() as demo:
+    gr.HTML("""<h1 align="center">MindSearch Gradio Demo</h1>""")
+    gr.HTML("""<p style="text-align: center; max-width: 800px; font-family: Arial, sans-serif;">MindSearch is an open-source AI Search Engine Framework with Perplexity.ai Pro performance. You can deploy your own Perplexity.ai-style search engine using either closed-source LLMs (GPT, Claude) or open-source LLMs (InternLM2.5-7b-chat).</p>""")
     gr.HTML("""
-    <div style="text-align: center; font-size: 16px; margin-bottom: 20px;">
-        <a href="https://github.com/InternLM/MindSearch" style="margin-right: 15px; text-decoration: none; color: #4A90E2; font-weight: bold;">🔗 GitHub</a>
-        <a href="https://arxiv.org/abs/2407.20183" style="margin-right: 15px; text-decoration: none; color: #4A90E2; font-weight: bold;">📄 Arxiv</a>
-        <a href="https://huggingface.co/papers/2407.20183" style="margin-right: 15px; text-decoration: none; color: #4A90E2; font-weight: bold;">📚 Hugging Face Papers</a>
-        <a href="https://huggingface.co/spaces/internlm/MindSearch" style="text-decoration: none; color: #4A90E2; font-weight: bold;">🤗 Hugging Face Demo</a>
+    <div style="text-align: center; font-size: 16px;">
+        <a href="https://github.com/InternLM/MindSearch" style="margin-right: 15px; text-decoration: none; color: #4A90E2;">🔗 GitHub</a>
+        <a href="https://arxiv.org/abs/2407.20183" style="margin-right: 15px; text-decoration: none; color: #4A90E2;">📄 Arxiv</a>
+        <a href="https://huggingface.co/papers/2407.20183" style="margin-right: 15px; text-decoration: none; color: #4A90E2;">📚 Hugging Face Papers</a>
+        <a href="https://huggingface.co/spaces/internlm/MindSearch" style="text-decoration: none; color: #4A90E2;">🤗 Hugging Face Demo</a>
     </div>
     """)
     with gr.Row():
         with gr.Column(scale=10):
             with gr.Row():
                 with gr.Column():
-                    planner = gr.Chatbot(label='Planner',
+                    planner = gr.Chatbot(label='planner',
                                          height=700,
                                          show_label=True,
                                          show_copy_button=True,
                                          bubble_full_width=False,
                                          render_markdown=True)
                 with gr.Column():
-                    searcher = gr.Chatbot(label='Searcher',
+                    searcher = gr.Chatbot(label='searcher',
                                           height=700,
                                           show_label=True,
                                           show_copy_button=True,
@@ -115,13 +131,12 @@ with gr.Blocks(css=".button-primary { background-color: #4CAF50; color: white; b
                 user_input = gr.Textbox(show_label=False,
                                         placeholder='帮我搜索一下 InternLM 开源体系',
                                         lines=5,
-                                        container=False,
-                                        css="border-radius: 8px; border: 1px solid #ddd; padding: 10px;")
+                                        container=False)
             with gr.Row():
                 with gr.Column(scale=2):
-                    submitBtn = gr.Button('Submit', css="button-primary")
+                    submitBtn = gr.Button('Submit')
                 with gr.Column(scale=1, min_width=20):
-                    emptyBtn = gr.Button('Clear History', css="button-secondary")
+                    emptyBtn = gr.Button('Clear History')
 
     def user(query, history):
         return '', history + [[query, '']]
@@ -136,4 +151,4 @@ demo.queue()
 demo.launch(server_name='0.0.0.0',
             server_port=7860,
             inbrowser=True,
-            share=True)
+            share=True)