app.py

import gradio as gr
import requests

def brave_search(query, api_key):
    url = "https://api.search.brave.com/res/v1/web/search"
    headers = {
        "Accept": "application/json",
        "X-Subscription-Token": api_key
    }
    params = {
        "q": query,
        "count": 5
    }
    response = requests.get(url, headers=headers, params=params)
    if response.status_code == 200:
        results = response.json().get("web", {}).get("results", [])
        if not results:
            return "検索結果が見つかりませんでした。"
        output = ""
        for r in results:
            output += f"【{r.get('title','No Title')}】\n{r.get('url','')}\n{r.get('description','')}\n\n"
        return output
    else:
        return f"エラーが発生しました: {response.status_code}"

with gr.Blocks() as demo:
    gr.Markdown("# Brave API 検索アプリ")
    api_key = gr.Textbox(label="Brave APIキー", type="password")
    query = gr.Textbox(label="検索クエリ")
    output = gr.Textbox(label="検索結果", lines=10)
    search_btn = gr.Button("検索")
    search_btn.click(fn=brave_search, inputs=[query, api_key], outputs=output)

if __name__ == "__main__":
    demo.launch(mcp_server=True)