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>"
    
    if isinstance(results, dict) and "results" in results:
        news_results = results["results"].get("news", [])
        for result in news_results:
            time_str = result.get("time", "").strip()
            if time_str.endswith("ago"):
                time_value = int(time_str.split()[0])
                time_unit = time_str.split()[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>
                    """
    else:
        html_output += "<p>No results found or unexpected response format.</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)

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

iface = gr.Interface(theme="Nymbo/Nymbo_Theme", css=css,
    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="Global News Search-AI",
)

iface.launch()