| import os |
| from fastapi import FastAPI, Header, HTTPException |
| from transformers import pipeline |
| from pydantic import BaseModel |
|
|
| import os |
|
|
| |
| API_KEY = os.getenv("API_KEY") |
|
|
| |
| sentiment_model = pipeline( |
| "sentiment-analysis", |
| model="cardiffnlp/twitter-roberta-base-sentiment-latest" |
| ) |
|
|
| |
| app = FastAPI(title="Sentiment Analysis API") |
|
|
| |
| class TextRequest(BaseModel): |
| text: str |
|
|
| |
| @app.post("/analyze") |
| def analyze_sentiment(request: TextRequest, x_api_key: str = Header(...)): |
| if x_api_key != API_KEY: |
| raise HTTPException(status_code=401, detail="Invalid API Key") |
| result = sentiment_model(request.text) |
| return {"result": result} |
|
|
| @app.get("/hello") |
| def hello_world(): |
| return {"message": "Hello World"} |
