ainews-db / app.py
seawolf2357's picture
Update app.py
5d3044b verified
raw
history blame
2.87 kB
import gradio as gr
import requests
API_KEY = "V38CNn4HXpLtynJQyOeoUensTEYoFy8PBUxKpDqAW1pawT1vfJ2BWtPQ98h6"
# List of 50 major countries
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, verbatim, gfilter, page, num_result):
url = "https://api.serphouse.com/serp/live"
params = {
"q": query,
"domain": "google.com",
"lang": "en",
"device": "desktop",
"serp_type": "news",
"loc": country,
"country": country,
"verbatim": verbatim,
"gfilter": gfilter,
"page": page,
"num_result": num_result
}
headers = {
"Authorization": f"Bearer {API_KEY}"
}
response = requests.get(url, params=params, headers=headers)
if response.status_code == 200:
return response.json()
else:
return f"Error: {response.status_code} - {response.text}"
def format_results(results):
if isinstance(results, dict) and "results" in results:
news_results = results["results"].get("news", [])
formatted_results = ""
for i, result in enumerate(news_results, 1):
formatted_results += f"{i}. {result['title']}\n"
formatted_results += f" URL: {result['link']}\n"
formatted_results += f" Snippet: {result['snippet']}\n\n"
return formatted_results
else:
return str(results)
def serphouse_search(query, country, verbatim, gfilter, page, num_result):
verbatim = "1" if verbatim else "0"
gfilter = "1" if gfilter else "0"
results = search_serphouse(query, country, verbatim, gfilter, page, num_result)
return format_results(results)
iface = gr.Interface(
fn=serphouse_search,
inputs=[
gr.Textbox(label="Search Query"),
gr.Dropdown(MAJOR_COUNTRIES, label="Country"),
gr.Checkbox(label="Verbatim"),
gr.Checkbox(label="Google Filter"),
gr.Slider(1, 10, 1, label="Page"),
gr.Slider(1, 100, 10, label="Number of Results")
],
outputs="text",
title="SERPHouse News Search Interface",
description="Enter your search query and select a country to get news results from the SERPHouse API."
)
iface.launch()