Update app.py

app.py

@@ -47,37 +47,39 @@ line_token = "C9r65PpEvIvOJSK2xMhgl53WvmOhhnKEOuQq7DsiVJT"
 async def classify_text(input: TextInput):
     if reloaded_pipe is None:
         raise HTTPException(status_code=500, detail="Model not loaded")
-
     try:
         result = reloaded_pipe(input.text)
         sentiment = result[0]['label'].lower()  
         score = result[0]['score']
-
+        
         if sentiment == 'neg':
-            message = f"[แจ้งเตือน CSI]: ความพึงพอใจของผู้ป่วย \n ข้อความ: {input.text} \n csi score: {score:.2f}"
+            message = f"[แจ้งเตือน CSI]: ความพึงพอใจของผู้ป่วย \nข้อความ: {input.text} \ncsi score: {score:.2f}"
             message_parts = split_message(message)
-
+            notification_status = []
             for i, part in enumerate(message_parts):
                 # status = send_line_notification(part, line_token)
                 status = 200
                 if status == 200:
-                    print(f"ส่งการแจ้งเตือนส่วนที่ {i+1}/{len(message_parts)} ผ่าน LINE สำเร็จ")
+                    notification_status.append(f"ส่งการแจ้งเตือนส่วนที่ {i+1}/{len(message_parts)} ผ่าน LINE สำเร็จ")
                 else:
-                    print(f"การส่งการแจ้งเตือนส่วนที่ {i+1}/{len(message_parts)} ผ่าน LINE ล้มเหลว")
-
+                    notification_status.append(f"การส่งการแจ้งเตือนส่วนที่ {i+1}/{len(message_parts)} ผ่าน LINE ล้มเหลว")
             return {
                 "result": result, 
-                "message": f"Negative sentiment detected and notification sent to LINE. \n{message}",
-                "formatted_message": message
+                "sentiment": "negative",
+                "score": score,
+                "message": "Negative sentiment detected and notification sent to LINE.",
+                "formatted_message": message,
+                "notification_status": notification_status
             }
         else:
-            message = f"[Sentiment Info]: ข้อความ: {input.text} \n csi score: {score:.2f}"
+            message = f"[Sentiment Info]: ข้อความ: {input.text} \ncsi score: {score:.2f}"
             return {
                 "result": result, 
+                "sentiment": "positive" if sentiment == 'pos' else "neutral",
+                "score": score,
                 "message": "Sentiment is not negative. No notification sent.",
                 "formatted_message": message
             }
-
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))