Upload app.py

app.py

@@ -1,5 +1,6 @@
-
 import json
+import os
+
 import gradio as gr
 import requests
 from lagent.schema import AgentStatusCode
@@ -10,18 +11,21 @@ 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.'''
+    '''
+    Reset the chatbot memory.
+    '''
     history_planner = []
     history_searcher = []
     if PLANNER_HISTORY:
         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
+            AgentStatusCode.STREAM_ING, AgentStatusCode.ANSWER_ING
     ]:
         gr_history[-1][1] = agent_return['response']
     elif agent_return['state'] == AgentStatusCode.PLUGIN_START:
@@ -31,7 +35,8 @@ 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```'  # noqa: E501
+            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]
@@ -42,11 +47,13 @@ def format_response(gr_history, agent_return):
         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'):
+                                             decode_unicode=False,
+                                             delimiter=b'\n'):
             if chunk:
                 decoded = chunk.decode('utf-8')
                 if decoded == '\r':
@@ -92,31 +99,39 @@ def predict(history_planner, history_searcher):
             yield history_planner, history_searcher
     return history_planner, history_searcher
 
+
 examples = [
-    ["Find legal precedents in contract law."],
-    ["What are the top 10 e-commerce websites?"],
-    ["Generate a report on global climate change."],
+    ['Find legal precedents in contract law.'],
+    ['What are the top 10 e-commerce websites?'],
+    ['Generate a report on global climate change.'],
 ]
 
-import os
-css_path = os.path.join(os.path.dirname(__file__), "css", "gradio_front.css")
-
+css_path = os.path.join(os.path.dirname(__file__), 'css', 'gradio_front.css')
 with gr.Blocks(css=css_path) as demo:
-    with gr.Column(elem_classes="chat-box"):
+    with gr.Column(elem_classes='chat-box'):
         gr.HTML("""<h1 align="center">MindSearch Gradio Demo</h1>""")
-        gr.HTML("""<p style="text-align: center; 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(
+            """<p style="text-align: center; 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;">
-        <a href="https://github.com/InternLM/MindSearch" style="margin-right: 15px; text-decoration: none; color: #4A90E2;" target="_blank">🔗 GitHub</a>
-        <a href="https://arxiv.org/abs/2407.20183" style="margin-right: 15px; text-decoration: none; color: #4A90E2;" target="_blank">📄 Arxiv</a>
-        <a href="https://huggingface.co/papers/2407.20183" style="margin-right: 15px; text-decoration: none; color: #4A90E2;" target="_blank">📚 Hugging Face Papers</a>
-        <a href="https://huggingface.co/spaces/internlm/MindSearch" style="text-decoration: none; color: #4A90E2;" target="_blank">🤗 Hugging Face Demo</a>
+        <a href="https://github.com/InternLM/MindSearch" style="margin-right: 15px;
+         text-decoration: none; color: #4A90E2;" target="_blank">🔗 GitHub</a>
+        <a href="https://arxiv.org/abs/2407.20183" style="margin-right: 15px;
+         text-decoration: none; color: #4A90E2;" target="_blank">📄 Arxiv</a>
+        <a href="https://huggingface.co/papers/2407.20183" style="margin-right:
+         15px; text-decoration: none; color: #4A90E2;" target="_blank">📚 Hugging Face Papers</a>
+        <a href="https://huggingface.co/spaces/internlm/MindSearch"
+         style="text-decoration: none; color: #4A90E2;" target="_blank">🤗 Hugging Face Demo</a>
         </div>""")
-        gr.HTML("""
-        <h1 align='right'><img src='https://raw.githubusercontent.com/InternLM/MindSearch/98fd84d566fe9e3adc5028727f72f2944098fd05/assets/logo.svg' alt='MindSearch Logo1' class="logo"></h1>
-        """)
+    gr.HTML("""
+       <h1 align='right'><img
+        src=
+        'https://raw.githubusercontent.com/InternLM/MindSearch/98fd84d566fe9e3adc5028727f72f2944098fd05/assets/logo.svg'
+         alt='MindSearch Logo1' class="logo"></h1> """)
 
     with gr.Row():
         with gr.Column(scale=10):
@@ -127,31 +142,33 @@ with gr.Blocks(css=css_path) as demo:
                                          show_copy_button=True,
                                          bubble_full_width=False,
                                          render_markdown=True,
-                                         elem_classes="chatbot-container")
+                                         elem_classes='chatbot-container')
                 with gr.Column():
                     searcher = gr.Chatbot(label='searcher',
                                           show_label=True,
                                           show_copy_button=True,
                                           bubble_full_width=False,
                                           render_markdown=True,
-                                          elem_classes="chatbot-container")
+                                          elem_classes='chatbot-container')
 
-            with gr.Row(elem_classes="chat-box"):
+            with gr.Row(elem_classes='chat-box'):
                 # Text input area
-                user_input = gr.Textbox(
-                    show_label=False,
-                    placeholder="Type your message...",
-                    lines=1,
-                    container=False,
-                    elem_classes="editor"
-                )
+                user_input = gr.Textbox(show_label=False,
+                                        placeholder='Type your message...',
+                                        lines=1,
+                                        container=False,
+                                        elem_classes='editor')
                 # Buttons (now in the same Row)
-                submitBtn = gr.Button("submit", variant="primary", elem_classes="toolbarButton")
-                clearBtn = gr.Button("clear", variant="secondary", elem_classes="toolbarButton")
-
-            with gr.Row(elem_classes="examples-container"):
-                examples_component = gr.Examples(examples, inputs=user_input,
-                                                  label="Try these examples:")
+                submitBtn = gr.Button('submit',
+                                      variant='primary',
+                                      elem_classes='toolbarButton')
+                clearBtn = gr.Button('clear',
+                                     variant='secondary',
+                                     elem_classes='toolbarButton')
+            with gr.Row(elem_classes='examples-container'):
+                examples_component = gr.Examples(examples,
+                                                 inputs=user_input,
+                                                 label='Try these examples:')
 
     def user(query, history):
         return '', history + [[query, '']]
@@ -160,13 +177,13 @@ with gr.Blocks(css=css_path) as demo:
         return user(example[0], planner.value)
 
     submitBtn.click(user, [user_input, planner], [user_input, planner],
-                     queue=False).then(predict, [planner, searcher],
-                                        [planner, searcher])
+                    queue=False).then(predict, [planner, searcher],
+                                      [planner, searcher])
     clearBtn.click(rst_mem, [planner, searcher], [planner, searcher],
-                    queue=False)
+                   queue=False)
 
 demo.queue()
 demo.launch(server_name='0.0.0.0',
             server_port=7860,
             inbrowser=True,
-            share=True)
+            share=True)