import gradio as gr
from openbb_platform.openbb import sdk
import os

API_KEY = os.environ.get("OPENBB_API_KEY", "your_secret_key")  # 优先读取环境变量

def query(symbol: str, request: gr.Request):
    # 校验密钥
    if request.headers.get("x-api-key") != API_KEY:
        return "Unauthorized: Invalid API Key"
    try:
        result = sdk.stocks.price(symbol)
        return str(result)
    except Exception as e:
        return f"Error: {e}"

iface = gr.Interface(
    fn=query,
    inputs="text",
    outputs="text",
    title="OpenBB Query"
)
iface.launch()