Spaces:
Runtime error
Runtime error
File size: 2,107 Bytes
062c36e 25562f7 062c36e 25562f7 062c36e 25562f7 062c36e 25562f7 062c36e 25562f7 c19a330 e8e558d 25562f7 062c36e 25562f7 062c36e 25562f7 062c36e 25562f7 062c36e 25562f7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
from fastapi import Depends, FastAPI, HTTPException, status
from sqlalchemy.orm import Session
import schemas, models
from transformers import pipeline
import database
app = FastAPI()
models.database.Base.metadata.create_all(bind=database.engine)
classifier = pipeline("sentiment-analysis")
# Database session dependency
def get_db():
db = database.SessionLocal()
try:
yield db
finally:
db.close()
@app.post("/analyze_sentiment", status_code=status.HTTP_201_CREATED)
def create_sentiment_result(sentiment_result: schemas.SentimentResultCreate, text_input: str, db: Session = Depends(get_db)):
# Perform input validation
if not isinstance(text_input, str) or not text_input.strip():
raise HTTPException(status_code=status.HTTP_422_UNPROCESSABLE_ENTITY, detail="Invalid text input")
try:
text_content = text_input
sentiment_analysis_result = classifier(text_content)
# Create a new SentimentResult instance
new_sentiment_result = models.SentimentResult(
score=sentiment_analysis_result[0]['score'],
label = sentiment_analysis_result[0]['label'],
text_input=text_content
)
# Add the new SentimentResult to the database
db.add(new_sentiment_result)
db.commit()
db.refresh(new_sentiment_result)
return new_sentiment_result
except Exception as e:
raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=f"An error occurred: {str(e)}")
@app.delete("/sentiment/{id}", status_code=status.HTTP_204_NO_CONTENT)
def delete_sentiment_result(id: int, db: Session = Depends(get_db)):
# Retrieve the sentiment result and check for existence
sentiment_result_to_delete = db.query(models.SentimentResult).filter(models.SentimentResult.id == id).first()
if sentiment_result_to_delete is None:
raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail=f"SentimentResult with ID {id} not found")
# Delete the sentiment result
db.delete(sentiment_result_to_delete)
db.commit()
|