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}"
    }
    
    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):
    time_parts = time_str.split()
    if len(time_parts) >= 2:
        try:
            value = int(time_parts[0])
            unit = time_parts[1].lower()
            if unit in ['minute', 'minutes', 'hour', 'hours']:
                return True
            elif unit in ['day', 'days'] and value <= 2:  # 최대 2일까지 허용
                return True
        except ValueError:
            pass
    return False

def format_results(results):
    html_output = """
    <style>
        .news-container { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; }
        .news-item { background-color: #f9f9f9; border: 1px solid #ddd; padding: 20px; margin-bottom: 20px; border-radius: 8px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); }
        .news-title { font-size: 20px; color: #1a0dab; text-decoration: none; font-weight: bold; }
        .news-title:hover { text-decoration: underline; }
        .news-snippet { color: #545454; margin: 15px 0; line-height: 1.4; }
        .news-meta { font-size: 14px; color: #006621; }
        .no-results { text-align: center; color: #666; font-style: italic; }
        .debug-info { background-color: #f0f0f0; padding: 10px; margin-top: 20px; border-radius: 5px; white-space: pre-wrap; }
    </style>
    <div class="news-container">
        <h2 style="text-align: center; color: #333;">Recent News Results</h2>
    """
    
    try:
        if isinstance(results, dict) and "results" in results:
            news_results = results["results"].get("news", [])
            filtered_results = [result for result in news_results if is_recent_news(result.get("time", "").strip())]
            
            if not filtered_results:
                html_output += '<p class="no-results">No recent news results found.</p>'
            else:
                for result in filtered_results:
                    html_output += f"""
                    <div class="news-item">
                        <a href="{result.get('url', '#')}" class="news-title" target="_blank">{result.get('title', 'No Title')}</a>
                        <p class="news-snippet">{result.get('snippet', 'No Snippet')}</p>
                        <p class="news-meta">Source: {result.get('channel', 'Unknown')} - {result.get('time', 'Unknown time')}</p>
                    </div>
                    """
        else:
            html_output += f'<p class="no-results">Unexpected response format: {type(results)}</p>'
    except Exception as e:
        html_output += f'<p class="no-results">Error processing results: {str(e)}</p>'
    
    # 디버그 정보 추가 (실제 사용 시 제거 가능)
    html_output += f'<div class="debug-info"><h3>Debug Info:</h3><pre>{json.dumps(results, indent=2)}</pre></div>'
    
    html_output += "</div>"
    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)

css = """
footer {
    visibility: hidden;
}
"""

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 recent news results from the SERPHouse API.",
    theme="Nymbo/Nymbo_Theme",
    css=css
)

iface.launch()