from fastapi import FastAPI
from pydantic import BaseModel
from transformers import pipeline

app = FastAPI()

nlp = pipeline("sentiment-analysis")

class PromptInput(BaseModel):
    prompt: str

@app.post("/analyze")
def analyze_sentiment(prompt_input: PromptInput):
    result = nlp(prompt_input.prompt)
    return {
        'input': prompt_input.prompt,
        'output': result
    }