Update main.py

main.py

@@ -2,7 +2,12 @@ from fastapi import FastAPI
 import os
 import json
 import google.generativeai as genai
+from pydantic import BaseModel, validator
 
+
+class Item(BaseModel):
+    text: str = "sddddddddddd"
+    
 app = FastAPI()
 
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
@@ -28,7 +33,7 @@ task_description = " You need to classify each message you receive among the fol
 def classify_msg(message):
     prompt_parts = [
         task_description,
-        f"Message: {message}",
+        f"Message: {message['text']}",
         "Category: ",
     ]
 
@@ -46,5 +51,5 @@ async def root():
  return {"Text Emotion Classification":"Version 1.5 'Text'"}
 
 @app.post("/classify/")
-def read_user(text: str):
-    return classify_msg(text)
+def read_user(js: Item):
+    return classify_msg(js)