Upload app.py

app.py

@@ -690,7 +690,95 @@ def mindAsk(
         past_key_values,
     )                    
                     
-                    
+from bs4 import BeautifulSoup
+import requests
+
+def scrape_text(url, proxies) -> str:
+    """从网页抓取文本，限制为前500个字符
+
+    参数:
+        url (str): 要抓取文本的网址
+
+    返回:
+        str: 抓取到的文本，最多为前500个字符
+    """
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.61 Safari/537.36',
+        'Content-Type': 'text/plain',
+    }
+    try: 
+        response = requests.get(url, headers=headers, proxies=proxies, timeout=8)
+        if response.encoding == "ISO-8859-1":
+            response.encoding = response.apparent_encoding
+    except: 
+        return "无法连接到该网页"
+    soup = BeautifulSoup(response.text, "html.parser")
+    for script in soup(["script", "style"]):
+        script.extract()
+    text = soup.get_text()
+    # 截取文本，限制最多500个字符
+    text = text[:500]
+    return text
+
+# 定义函数：联网搜索并更新聊天界面
+def GGSearch(
+    user_input, chatbot, max_length, top_p, temperature, history, past_key_values
+):
+    global GGSearchins  # 确保 GGSearchins 在函数内部可用
+
+    # 使用用户输入进行联网搜索
+    url = f"https://www.google.com/search?q={user_input}"
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.61 Safari/537.36'
+    }
+
+    try:
+        response = requests.get(url, headers=headers)
+        soup = BeautifulSoup(response.content, 'html.parser')
+        search_results = []
+        for g in soup.find_all('div', class_='g'):
+            anchors = g.find_all('a')
+            if anchors:
+                link = anchors[0]['href']
+                if link.startswith('/url?q='):
+                    link = link[7:]
+                if not link.startswith('http'):
+                    continue
+                search_results.append(link)
+    except:
+        search_results = []
+
+    # 限制搜索结果数量为3
+    search_results = search_results[:5]
+
+    # 从搜索结果抓取文本并存储到 GGSearchins
+    scraped_texts = []
+    for link in search_results:
+        scraped_text = scrape_text(link, proxies=None)  # 假设抓取不需要代理
+        scraped_texts.append(scraped_text)
+
+    # 将抓取到的文本拼接并存储到 GGSearchins
+    GGSearchins = "\n".join(scraped_text for scraped_text in scraped_texts)
+
+    # 更新聊天界面和历史记录
+    chatbot.append(("联网搜索结果：", GGSearchins))
+    history.append(("联网搜索结果：", GGSearchins))
+
+    user_input = txtSumins+GGSearchins
+
+    # 继续正常的 GPT 对话流程
+    yield from predict(
+        RETRY_FLAG,
+        user_input,
+        chatbot,
+        max_length,
+        top_p,
+        temperature,
+        history=[],        
+        past_key_values=None,
+    )
+
+                   
                     
  
 with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) as demo:
@@ -758,6 +846,7 @@ with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) a
                 with gr.Row():
                     fitnessAskBtn = gr.Button("🥼健康咨询", variant="primary")
                     mindAskBtn = gr.Button("😶‍🌫️心理咨询", variant="primary")
+                    GGSearchBtn = gr.Button("🐞联网搜索", variant="primary")
                                   
         with gr.Column(scale=1):
             gr.HTML("""<h3 align="center">🍀您好，除健康咨询和心理咨询外，其它功能使用前，请先清空历史，并输入问题。🍀</h3>""")
@@ -1090,6 +1179,23 @@ with gr.Blocks(title="🐰Bofan Ai🐰", theme=gr.themes.Soft(text_size="sm")) a
         outputs=[chatbot, history, past_key_values],
     )
     
+    GGSearchBtn.click(
+    reset_state, outputs=[chatbot, history, past_key_values], show_progress="full"
+    )
+    GGSearchBtn.click(
+        GGSearch,
+        inputs=[
+            user_input,
+            chatbot,
+            max_length,
+            top_p,
+            temperature,
+            history,
+            past_key_values,
+        ],
+        # outputs = [chatbot, history, last_user_message, user_message]
+        outputs=[chatbot, history, past_key_values],
+    )    
     deleteBtn.click(delete_last_turn, [chatbot, history], [chatbot, history])
 
     with gr.Accordion("Example inputs", open=False):