app.py

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, verbatim, 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": verbatim,
            "num": str(num_result)
        }
    }
    
    headers = {
        "accept": "application/json",
        "content-type": "application/json",
        "authorization": f"Bearer {API_KEY}"
    }
    
    response = requests.post(url, json=payload, headers=headers)
    
    if response.status_code == 200:
        return response.json()
    else:
        return f"Error: {response.status_code} - {response.text}"

def format_results(results):
    now = datetime.now()
    html_output = "<h2>Search Results</h2>"
    
    # 디버깅: API 응답 전체를 출력
    html_output += f"<pre>{json.dumps(results, indent=2)}</pre>"
    
    if isinstance(results, dict) and "results" in results:
        news_results = results["results"].get("news", [])
        if not news_results:
            html_output += "<p>No news results found in the API response.</p>"
        for result in news_results:
            time_str = result.get("time", "").strip()
            if time_str.endswith("ago"):
                time_parts = time_str.split()
                if len(time_parts) >= 2:
                    try:
                        time_value = int(time_parts[0])
                        time_unit = time_parts[1]
                        
                        if time_unit in ["minute", "minutes", "hour", "hours"] and time_value <= 24:
                            html_output += f"""
                            <div style='border: 1px solid #ddd; padding: 10px; margin-bottom: 10px;'>
                                <h3><a href='{result.get('url', '#')}'>{result.get('title', 'No Title')}</a></h3>
                                <p>{result.get('snippet', 'No Snippet')}</p>
                                <p><small>Source: {result.get('channel', 'Unknown')} - {result.get('time', 'Unknown time')}</small></p>
                            </div>
                            """
                    except ValueError:
                        html_output += f"<p>Error parsing time: {time_str}</p>"
                else:
                    html_output += f"<p>Invalid time format: {time_str}</p>"
            else:
                html_output += f"<p>Unexpected time format: {time_str}</p>"
    else:
        html_output += "<p>Unexpected response format or no results found.</p>"
    
    return html_output

def serphouse_search(query, country, verbatim, page, num_result):
    verbatim = "1" if verbatim else "0"
    
    results = search_serphouse(query, country, verbatim, 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.Slider(1, 10, 1, label="Page"),
        gr.Slider(1, 100, 10, label="Number of Results")
    ],
    outputs="html",
    title="SERPHouse News Search Interface",
    description="Enter your search query and select a country to get news results from the SERPHouse API. Only articles from the last 24 hours will be displayed."
)

iface.launch()