ainews-db / app.py
seawolf2357's picture
Update app.py
d36b7a7 verified
raw
history blame
4.16 kB
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"
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)
}
}
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:
return f"Error: {str(e)}"
def is_recent_news(time_str):
if not time_str:
return False
time_parts = time_str.lower().split()
if len(time_parts) < 2:
return False
try:
value = int(time_parts[0])
unit = time_parts[1]
if unit in ['minute', 'minutes', 'hour', 'hours']:
return True
elif unit in ['day', 'days']:
return value <= 2
except ValueError:
return False
return False
def format_results(results):
markdown_output = "## Recent News Results\n\n"
try:
if isinstance(results, dict) and "results" in results and "news" in results["results"]:
news_results = results["results"]["news"]
filtered_results = [result for result in news_results if is_recent_news(result.get("time", ""))]
if not filtered_results:
markdown_output += "*No recent news results found.*\n\n"
else:
for result in filtered_results:
markdown_output += f"""
### [{result.get('title', 'No Title')}]({result.get('url', '#')})
{result.get('snippet', 'No Snippet')}
**Source:** {result.get('channel', 'Unknown')} - {result.get('time', 'Unknown time')}
---
"""
else:
markdown_output += "*No valid news results found in the API response.*\n\n"
except Exception as e:
markdown_output += f"*Error processing results: {str(e)}*\n\n"
return markdown_output
def serphouse_search(query, country, page, num_result):
results = search_serphouse(query, country, page, num_result)
print("Raw results:", results) # 콘솔에 원본 결과 출력
formatted_results = format_results(results)
print("Formatted results:", formatted_results) # 콘솔에 포맷된 결과 출력
return f"Raw Results:\n\n```json\n{json.dumps(results, indent=2)}\n```\n\nFormatted Results:\n\n{formatted_results}"
css = """
footer {
visibility: hidden;
}
"""
iface = gr.Interface(
fn=serphouse_search,
inputs=[
gr.Textbox(label="Search Query"),
gr.Dropdown(MAJOR_COUNTRIES, label="Country"),
gr.Slider(1, 10, 1, label="Page"),
gr.Slider(1, 100, 10, label="Number of Results")
],
outputs=gr.Textbox(label="Results", lines=20),
title="SERPHouse News Search Interface",
description="Enter your search query and select a country to get recent news results from the SERPHouse API.",
theme="Nymbo/Nymbo_Theme",
css=css
)
iface.launch()