Update main.py

main.py

@@ -129,12 +129,26 @@ async def classify_text(text: str = Form(...)):
     except Exception as e:
         return JSONResponse(content=f"Error classifying text: {str(e)}", status_code=500)
 
-@app.post("/test_classify/", description="Classify the provided text with positive, neutral, or negative sentiment.")
+@@app.post("/test_classify/", description="Classify the provided text with positive, neutral, or negative sentiment.")
 async def test_classify_text(text: str = Form(...)):
     try:
         # Perform text classification using the updated model that returns positive, neutral, or negative
         result = nlp_classification_v2(text)
-        return result
+
+        # Get the label (positive, neutral, negative) and format it as required
+        label = result[0]['label']
+
+        # Map the model labels to human-readable format
+        label_map = {
+            "LABEL_0": "Negative",  # Cardiff NLP's model may use these labels
+            "LABEL_1": "Neutral",   # Remap LABEL_1 to Neutral
+            "LABEL_2": "Positive"   # LABEL_2 corresponds to Positive
+        }
+
+        # Get the readable label from the map
+        formatted_label = label_map.get(label, "Unknown")
+
+        return {"label": formatted_label, "score": result[0]['score']}
     except Exception as e:
         return JSONResponse(content=f"Error classifying text: {str(e)}", status_code=500)