import requests
from fastapi import APIRouter, Header
from .apiModel import Payload

router = APIRouter()

API_URL = "https://api-inference.huggingface.co/models/cardiffnlp/twitter-roberta-base-sentiment-latest"


@router.post("/api-inference/")
async def api_inference(
        authorization: str = Header(...),
        item: Payload = None):
    print("请求：", item)
    headers = {"Authorization": authorization}
    # Forward request to Hugging Face API
    response = requests.post(
        API_URL,
        headers=headers,
        json=item.dict()
    )
    return response.json()