import gradio as gr
import requests
import json

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, 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": "1",
            "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 format_results(results):
    all_results = "## 모든 뉴스 결과\n\n"
    result_table = "## 뉴스 결과 표\n\n"
    result_table += "| 제목 | 링크 | 시간 | 출처 | 디버그 정보 |\n"
    result_table += "|------|------|------|------|-------------|\n"
    
    debug_info = "## 디버그 정보\n\n"
    debug_info += f"Raw API Response:\n```json\n{json.dumps(results, indent=2)}\n```\n\n"
    
    try:
        if not isinstance(results, dict):
            raise ValueError("결과가 사전 형식이 아닙니다.")
        
        if "results" not in results:
            raise ValueError("'results' 키가 응답에 없습니다.")
        
        news_results = results["results"].get("news", [])
        debug_info += f"뉴스 결과 수: {len(news_results)}\n\n"
        
        for idx, result in enumerate(news_results):
            title = result.get("title", "제목 없음")
            url = result.get("url", "#")
            snippet = result.get("snippet", "내용 없음")
            channel = result.get("channel", "알 수 없음")
            time_str = result.get("time", "알 수 없는 시간")
            
            single_debug_info = f"뉴스 {idx + 1} - 제목: {title}, 링크: {url}, 시간: {time_str}, 출처: {channel}"
            
            result_table += f"| {title[:30]}... | [{url[:30]}...]({url}) | {time_str} | {channel} | {single_debug_info[:50]}... |\n"
            
            article_info = f"""
### [{title}]({url})
{snippet}
**출처:** {channel} - {time_str}
---
"""
            all_results += article_info
        
    except Exception as e:
        error_message = f"결과 처리 중 오류 발생: {str(e)}"
        debug_info += error_message + "\n"
        all_results = error_message + "\n\n"
        
        result_table += f"| 오류 발생 | - | - | - | {error_message} |\n"
    
    return all_results, result_table + "\n\n" + debug_info


def serphouse_search(query, country, page, num_result):
    results = search_serphouse(query, country, page, num_result)
    all_results, result_table_and_debug = format_results(results)
    return all_results, result_table_and_debug

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

iface = gr.Interface(
    fn=serphouse_search,
    inputs=[
        gr.Textbox(label="검색어"),
        gr.Dropdown(MAJOR_COUNTRIES, label="국가"),
        gr.Slider(1, 10, 1, label="페이지"),
        gr.Slider(1, 100, 10, label="결과 수")
    ],
    outputs=[
        gr.Markdown(label="모든 결과"),
        gr.Markdown(label="결과 표 및 디버그 정보")
    ],
    title="SERPHouse 뉴스 검색 인터페이스",
    description="검색어를 입력하고 국가를 선택하여 SERPHouse API에서 뉴스 결과를 가져옵니다.",
    theme="Nymbo/Nymbo_Theme",
    css=css
)

iface.launch()