Update main.py

main.py

@@ -1,5 +1,6 @@
 from fastapi import Depends, FastAPI, HTTPException, status
 from sqlalchemy.orm import Session
+from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 import schemas, models
 from transformers import pipeline
@@ -24,20 +25,27 @@ def get_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")
-
+@app.post("/analyze_sentiment", response_model=schemas.SentimentResult, status_code=status.HTTP_201_CREATED)
+def create_sentiment_result(
+    sentiment_result: schemas.SentimentResultCreate,
+    text_input: str,
+    db: Session = Depends(get_db)
+):
     try:
+        # 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"
+            )
+
         text_content = text_input
         sentiment_analysis_result = pipe(text_content)
 
         # Create a new SentimentResult instance
         new_sentiment_result = models.SentimentResult(
             score=sentiment_analysis_result[0]['score'],
-            label = sentiment_analysis_result[0]['label'],
+            label=sentiment_analysis_result[0]['label'],
             text_input=text_content
         )
 
@@ -46,9 +54,14 @@ def create_sentiment_result(sentiment_result: schemas.SentimentResultCreate, tex
         db.commit()
         db.refresh(new_sentiment_result)
 
-        return new_sentiment_result
+        # Return JSON response
+        return JSONResponse(content=new_sentiment_result.dict(), status_code=status.HTTP_201_CREATED)
+    
     except Exception as e:
-        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=f"An error occurred: {str(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)):