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 is_within_24_hours(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 == 1:
                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 { border: 1px solid #ddd; padding: 15px; margin-bottom: 20px; border-radius: 5px; }
        .news-title { font-size: 18px; color: #1a0dab; text-decoration: none; }
        .news-snippet { color: #545454; margin: 10px 0; }
        .news-meta { font-size: 12px; color: #006621; }
    </style>
    <div class="news-container">
        <h2>Search Results (Last 24 Hours)</h2>
    """
    
    if isinstance(results, dict) and "results" in results:
        news_results = results["results"].get("news", [])
        filtered_results = [result for result in news_results if is_within_24_hours(result.get("time", "").strip())]
        
        if not filtered_results:
            html_output += "<p>No news results found within the last 24 hours.</p>"
        else:
            for result in filtered_results:
                html_output += f"""
                <div class="news-item">
                    <a href="{result.get('url', '#')}" class="news-title">{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 += "<p>Unexpected response format or no results found.</p>"
    
    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)

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()