246 lines
8.5 KiB
Python
246 lines
8.5 KiB
Python
|
import gradio as gr
|
||
|
|
import json
|
||
|
|
from typing import Dict, List, Any
|
||
|
|
import requests
|
||
|
|
|
||
|
|
class SearchUI:
|
||
|
|
def __init__(self):
|
||
|
|
self.api_key = "8af097ae8b587bb0569425058e03e5ef33b4c7b8b1a505053764b62e7e4ab9d6"
|
||
|
|
|
||
|
|
def parse_google_results(self, results: Dict) -> List[Dict]:
|
||
|
|
"""解析Google搜索结果"""
|
||
|
|
parsed = []
|
||
|
|
|
||
|
|
# 解析organic_results
|
||
|
|
if 'organic_results' in results:
|
||
|
|
for result in results['organic_results']:
|
||
|
|
parsed.append({
|
||
|
|
'title': result.get('title', ''),
|
||
|
|
'link': result.get('link', ''),
|
||
|
|
'snippet': result.get('snippet', ''),
|
||
|
|
'position': result.get('position', 0)
|
||
|
|
})
|
||
|
|
|
||
|
|
# 解析answer_box如果存在
|
||
|
|
if 'answer_box' in results:
|
||
|
|
answer = results['answer_box']
|
||
|
|
parsed.append({
|
||
|
|
'title': '直接回答:',
|
||
|
|
'link': answer.get('link', ''),
|
||
|
|
'snippet': answer.get('answer', ''),
|
||
|
|
'position': 0
|
||
|
|
})
|
||
|
|
|
||
|
|
return parsed
|
||
|
|
|
||
|
|
def parse_bing_results(self, results: Dict) -> List[Dict]:
|
||
|
|
"""解析Bing搜索结果"""
|
||
|
|
parsed = []
|
||
|
|
|
||
|
|
if 'organic_results' in results:
|
||
|
|
for result in results['organic_results']:
|
||
|
|
parsed.append({
|
||
|
|
'title': result.get('title', ''),
|
||
|
|
'link': result.get('link', ''),
|
||
|
|
'snippet': result.get('snippet', ''),
|
||
|
|
'position': result.get('position', 0)
|
||
|
|
})
|
||
|
|
|
||
|
|
return parsed
|
||
|
|
|
||
|
|
def parse_baidu_results(self, results: Dict) -> List[Dict]:
|
||
|
|
"""解析百度搜索结果"""
|
||
|
|
parsed = []
|
||
|
|
|
||
|
|
if 'organic_results' in results:
|
||
|
|
for result in results['organic_results']:
|
||
|
|
parsed.append({
|
||
|
|
'title': result.get('title', ''),
|
||
|
|
'link': result.get('link', ''),
|
||
|
|
'snippet': result.get('snippet', ''),
|
||
|
|
'position': result.get('position', 0)
|
||
|
|
})
|
||
|
|
|
||
|
|
# 解析百度特有的answer_box
|
||
|
|
if 'answer_box' in results:
|
||
|
|
for answer in results['answer_box']:
|
||
|
|
parsed.append({
|
||
|
|
'title': '直接回答:',
|
||
|
|
'link': answer.get('link', ''),
|
||
|
|
'snippet': f"{answer.get('answer', '')}\n来源: {answer.get('source', '')}",
|
||
|
|
'position': 0
|
||
|
|
})
|
||
|
|
|
||
|
|
return parsed
|
||
|
|
|
||
|
|
def format_results(self, results: List[Dict]) -> str:
|
||
|
|
"""将解析后的结果格式化为markdown文本"""
|
||
|
|
formatted = ""
|
||
|
|
for result in sorted(results, key=lambda x: x['position']):
|
||
|
|
formatted += f"### {result['title']}\n\n"
|
||
|
|
formatted += f"🔗 [{result['link']}]({result['link']})\n\n"
|
||
|
|
formatted += f"{result['snippet']}\n\n"
|
||
|
|
formatted += "---\n\n"
|
||
|
|
return formatted
|
||
|
|
|
||
|
|
def search_all_engines(self, query: str) -> tuple:
|
||
|
|
"""在所有搜索引擎中执行搜索并返回格式化结果"""
|
||
|
|
try:
|
||
|
|
results = {}
|
||
|
|
for engine in ["google", "bing", "baidu"]:
|
||
|
|
params = {
|
||
|
|
"api_key": self.api_key,
|
||
|
|
"engine": engine,
|
||
|
|
"q": query
|
||
|
|
}
|
||
|
|
# 为Google添加特殊参数
|
||
|
|
if engine == "google":
|
||
|
|
params.update({
|
||
|
|
"gl": "cn",
|
||
|
|
"hl": "zh-cn"
|
||
|
|
})
|
||
|
|
|
||
|
|
response = requests.get('https://serpapi.com/search', params=params)
|
||
|
|
if response.status_code == 200:
|
||
|
|
results[engine] = response.json()
|
||
|
|
else:
|
||
|
|
return f"{engine}搜索出错: HTTP {response.status_code}", "", ""
|
||
|
|
|
||
|
|
# 解析各引擎结果
|
||
|
|
google_parsed = self.parse_google_results(results['google'])
|
||
|
|
google_text = self.format_results(google_parsed)
|
||
|
|
|
||
|
|
bing_parsed = self.parse_bing_results(results['bing'])
|
||
|
|
bing_text = self.format_results(bing_parsed)
|
||
|
|
|
||
|
|
baidu_parsed = self.parse_baidu_results(results['baidu'])
|
||
|
|
baidu_text = self.format_results(baidu_parsed)
|
||
|
|
|
||
|
|
return google_text, bing_text, baidu_text
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
return f"搜索出错: {str(e)}", f"搜索出错: {str(e)}", f"搜索出错: {str(e)}"
|
||
|
|
|
||
|
|
def create_ui():
|
||
|
|
search_ui = SearchUI()
|
||
|
|
|
||
|
|
# 自定义CSS样式
|
||
|
|
custom_css = """
|
||
|
|
.container {padding: 20px;}
|
||
|
|
|
||
|
|
/* 搜索引擎区域样式 */
|
||
|
|
.google-container {background: rgba(66, 133, 244, 0.05); border-radius: 15px; padding: 15px;}
|
||
|
|
.bing-container {background: rgba(0, 169, 157, 0.05); border-radius: 15px; padding: 15px;}
|
||
|
|
.baidu-container {background: rgba(45, 97, 255, 0.05); border-radius: 15px; padding: 15px;}
|
||
|
|
|
||
|
|
/* 结果卡片样式 */
|
||
|
|
.result-card {
|
||
|
|
background: white;
|
||
|
|
border-radius: 10px;
|
||
|
|
padding: 15px;
|
||
|
|
margin: 10px 0;
|
||
|
|
box-shadow: 0 2px 4px rgba(0,0,0,0.1);
|
||
|
|
transition: all 0.3s ease;
|
||
|
|
}
|
||
|
|
|
||
|
|
.result-card:hover {
|
||
|
|
transform: translateY(-3px);
|
||
|
|
box-shadow: 0 4px 8px rgba(0,0,0,0.15);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* 链接样式 */
|
||
|
|
.result-card a {
|
||
|
|
color: #1a0dab;
|
||
|
|
text-decoration: none;
|
||
|
|
}
|
||
|
|
|
||
|
|
.result-card a:hover {
|
||
|
|
text-decoration: underline;
|
||
|
|
}
|
||
|
|
|
||
|
|
/* 分隔线样式 */
|
||
|
|
.google-container hr {border-color: rgba(66, 133, 244, 0.2);}
|
||
|
|
.bing-container hr {border-color: rgba(0, 169, 157, 0.2);}
|
||
|
|
.baidu-container hr {border-color: rgba(45, 97, 255, 0.2);}
|
||
|
|
"""
|
||
|
|
|
||
|
|
def format_with_css(results: str, engine: str) -> str:
|
||
|
|
"""为结果添加CSS类"""
|
||
|
|
# 将markdown结果分割成单独的卡片
|
||
|
|
cards = results.split("---")
|
||
|
|
formatted_cards = []
|
||
|
|
|
||
|
|
for card in cards:
|
||
|
|
if card.strip():
|
||
|
|
formatted_cards.append(f'<div class="result-card">{card}</div>')
|
||
|
|
|
||
|
|
return f'<div class="{engine}-container">{"".join(formatted_cards)}</div>'
|
||
|
|
|
||
|
|
with gr.Blocks(title="九鑫多搜索引擎聚合搜索", theme=gr.themes.Soft(), css=custom_css) as demo:
|
||
|
|
gr.Markdown("## 🔍 九鑫多搜索引擎聚合搜索")
|
||
|
|
|
||
|
|
with gr.Row():
|
||
|
|
query_input = gr.Textbox(
|
||
|
|
label="搜索关键词",
|
||
|
|
placeholder="请输入要搜索的内容...",
|
||
|
|
lines=1
|
||
|
|
)
|
||
|
|
search_button = gr.Button("搜索", variant="primary")
|
||
|
|
|
||
|
|
# 修改加载动画实现
|
||
|
|
loading_indicator = gr.HTML(
|
||
|
|
"""
|
||
|
|
<div style="text-align:center; margin:20px;">
|
||
|
|
<img src="https://media.giphy.com/media/3oEjI6SIIHBdRxXI40/giphy.gif" alt="loading..." />
|
||
|
|
<p>正在搜索中...</p>
|
||
|
|
</div>
|
||
|
|
""",
|
||
|
|
visible=False
|
||
|
|
)
|
||
|
|
|
||
|
|
with gr.Row():
|
||
|
|
with gr.Column():
|
||
|
|
gr.Markdown("### 🔵 Google 搜索结果")
|
||
|
|
google_output = gr.HTML()
|
||
|
|
|
||
|
|
with gr.Column():
|
||
|
|
gr.Markdown("### 🟢 Bing 搜索结果")
|
||
|
|
bing_output = gr.HTML()
|
||
|
|
|
||
|
|
with gr.Column():
|
||
|
|
gr.Markdown("### 🔷 百度搜索结果")
|
||
|
|
baidu_output = gr.HTML()
|
||
|
|
|
||
|
|
def on_search(query):
|
||
|
|
return [
|
||
|
|
True, # loading indicator visible
|
||
|
|
"", # clear google output
|
||
|
|
"", # clear bing output
|
||
|
|
"" # clear baidu output
|
||
|
|
]
|
||
|
|
|
||
|
|
def after_search(query):
|
||
|
|
google, bing, baidu = search_ui.search_all_engines(query)
|
||
|
|
return [
|
||
|
|
False, # hide loading indicator
|
||
|
|
format_with_css(google, "google"),
|
||
|
|
format_with_css(bing, "bing"),
|
||
|
|
format_with_css(baidu, "baidu")
|
||
|
|
]
|
||
|
|
|
||
|
|
search_button.click(
|
||
|
|
fn=on_search,
|
||
|
|
inputs=[query_input],
|
||
|
|
outputs=[loading_indicator, google_output, bing_output, baidu_output],
|
||
|
|
).then(
|
||
|
|
fn=after_search,
|
||
|
|
inputs=[query_input],
|
||
|
|
outputs=[loading_indicator, google_output, bing_output, baidu_output],
|
||
|
|
show_progress="minimal"
|
||
|
|
)
|
||
|
|
|
||
|
|
return demo
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
demo = create_ui()
|
||
|
|
demo.launch(share=True, server_name="0.0.0.0", server_port=10083)
|