Update app.py

app.py

@@ -46,7 +46,8 @@ def search_web(query):
     if results:
         for result in results:
             web_content += result['body']
-    return web_content
+    # return web_content
+    return ''  # 4097 tokens limit
 
 def chat_gpt(input, use_web, history=None):
     if history is None:
@@ -88,7 +89,7 @@ def chat_gpt(input, use_web, history=None):
     
     result = qa({"query": input, "chat_history": history})
     logger.info("chatgpt:", input,result)
-    return result["answer"]
+    return result["result"]
 
 def predict(input,
             large_language_model,
@@ -103,10 +104,12 @@ def predict(input,
     if history == None:
         history = []
 
-    if large_language_model == "gpt-3.5-turbo":
+    if large_language_model == "GPT-3.5-turbo":
         resp  = chat_gpt(input, use_web, history)
     elif large_language_model == "ChatGLM-6B-int4":
         resp = chat_glm(input, history)
+    elif large_language_model == "Search Web":
+        resp = search_web(input)
 
     history.append((input, resp))
     return '', history, history
@@ -126,7 +129,7 @@ with block as demo:
             model_choose = gr.Accordion("模型选择")
             with model_choose:
                 large_language_model = gr.Dropdown(
-                    ["ChatGLM-6B-int4","gpt-3.5-turbo"],
+                    ["ChatGLM-6B-int4","GPT-3.5-turbo","Search Web"],
                     label="large language model",
                     value="ChatGLM-6B-int4")
             use_web = gr.Radio(["True", "False"],