Spaces:
Sleeping
Sleeping
import gradio as gr | |
import requests | |
import json | |
from datetime import datetime, timedelta | |
API_KEY = "V38CNn4HXpLtynJQyOeoUensTEYoFy8PBUxKpDqAW1pawT1vfJ2BWtPQ98h6" | |
MAJOR_COUNTRIES = [ | |
"United States", "United Kingdom", "Canada", "Australia", "Germany", | |
"France", "Japan", "South Korea", "China", "India", | |
"Brazil", "Mexico", "Russia", "Italy", "Spain", | |
"Netherlands", "Sweden", "Switzerland", "Norway", "Denmark", | |
"Finland", "Belgium", "Austria", "New Zealand", "Ireland", | |
"Singapore", "Hong Kong", "Israel", "United Arab Emirates", "Saudi Arabia", | |
"South Africa", "Turkey", "Egypt", "Poland", "Czech Republic", | |
"Hungary", "Greece", "Portugal", "Argentina", "Chile", | |
"Colombia", "Peru", "Venezuela", "Thailand", "Malaysia", | |
"Indonesia", "Philippines", "Vietnam", "Pakistan", "Bangladesh" | |
] | |
def search_serphouse(query, country, page, num_result): | |
url = "https://api.serphouse.com/serp/live" | |
now = datetime.utcnow() | |
yesterday = now - timedelta(days=1) | |
date_range = f"{yesterday.strftime('%Y-%m-%d')},{now.strftime('%Y-%m-%d')}" | |
payload = { | |
"data": { | |
"q": query, | |
"domain": "google.com", | |
"loc": country, | |
"lang": "en", | |
"device": "desktop", | |
"serp_type": "news", | |
"page": str(page), | |
"verbatim": "1", | |
"num": str(num_result), | |
"date_range": date_range | |
} | |
} | |
headers = { | |
"accept": "application/json", | |
"content-type": "application/json", | |
"authorization": f"Bearer {API_KEY}" | |
} | |
try: | |
response = requests.post(url, json=payload, headers=headers) | |
response.raise_for_status() | |
return response.json() | |
except requests.RequestException as e: | |
error_msg = f"Error: {str(e)}" | |
if response.text: | |
error_msg += f"\nResponse content: {response.text}" | |
return {"error": error_msg} | |
def format_results(results): | |
all_results = "<h2>λͺ¨λ λ΄μ€ κ²°κ³Ό (24μκ° μ΄λ΄)</h2>" | |
debug_info = "<h2>λλ²κ·Έ μ 보</h2>" | |
if isinstance(results, dict) and "error" in results: | |
all_results += f"<p>μ€λ₯ λ°μ: {results['error']}</p>" | |
debug_info += f"<pre>{results['error']}</pre>" | |
return all_results, debug_info | |
debug_info += f"<pre>{json.dumps(results, indent=2, ensure_ascii=False)}</pre>" | |
try: | |
if not isinstance(results, dict): | |
raise ValueError("κ²°κ³Όκ° μ¬μ νμμ΄ μλλλ€.") | |
if "results" not in results: | |
raise ValueError("'results' ν€κ° μλ΅μ μμ΅λλ€.") | |
news_results = results["results"].get("news", []) | |
debug_info += f"<p>λ΄μ€ κ²°κ³Ό μ: {len(news_results)}</p>" | |
if not news_results: | |
all_results += "<p>κ²μ κ²°κ³Όκ° μμ΅λλ€.</p>" | |
else: | |
all_results += "<ol>" | |
for result in news_results: | |
title = result.get("title", "μ λͺ© μμ") | |
url = result.get("url", "#") | |
snippet = result.get("snippet", "λ΄μ© μμ") | |
channel = result.get("channel", "μ μ μμ") | |
time_str = result.get("time", "μ μ μλ μκ°") | |
article_info = f""" | |
<li> | |
<h3><a href="{url}" target="_blank">{title}</a></h3> | |
<p>{snippet}</p> | |
<p><strong>μΆμ²:</strong> {channel} - {time_str}</p> | |
</li> | |
""" | |
all_results += article_info | |
all_results += "</ol>" | |
except Exception as e: | |
error_message = f"κ²°κ³Ό μ²λ¦¬ μ€ μ€λ₯ λ°μ: {str(e)}" | |
debug_info += f"<p>{error_message}</p>" | |
all_results += f"<p>{error_message}</p>" | |
return all_results, debug_info | |
def serphouse_search(query, country, page, num_result): | |
results = search_serphouse(query, country, page, num_result) | |
all_results, debug_info = format_results(results) | |
return all_results, debug_info | |
css = """ | |
footer { | |
visibility: hidden; | |
} | |
ol { | |
padding-left: 20px; | |
} | |
li { | |
margin-bottom: 20px; | |
} | |
""" | |
iface = gr.Interface( | |
fn=serphouse_search, | |
inputs=[ | |
gr.Textbox(label="κ²μμ΄"), | |
gr.Dropdown(MAJOR_COUNTRIES, label="κ΅κ°"), | |
gr.Slider(1, 10, 1, label="νμ΄μ§"), | |
gr.Slider(1, 100, 10, label="κ²°κ³Ό μ") | |
], | |
outputs=[ | |
gr.HTML(label="λͺ¨λ κ²°κ³Ό"), | |
gr.HTML(label="λλ²κ·Έ μ 보") | |
], | |
title="24μκ° μ΄λ΄ λ΄μ€ κ²μ μΈν°νμ΄μ€", | |
description="κ²μμ΄λ₯Ό μ λ ₯νκ³ κ΅κ°λ₯Ό μ ννμ¬ 24μκ° μ΄λ΄μ λ΄μ€ κ²°κ³Όλ₯Ό κ°μ Έμ΅λλ€.", | |
theme="Nymbo/Nymbo_Theme", | |
css=css | |
) | |
iface.launch(auth=("gini","pick")) |