Update main.py

main.py

@@ -109,6 +109,7 @@ async def analyze(data: ReviewInput, x_api_key: str = Header(None)):
             sentiment_pipeline = pipeline("sentiment-analysis", model=data.model)
             sentiment = sentiment_pipeline(data.text)[0]
             emotion = detect_emotion(data.text)
+            churn_risk = assess_churn_risk(sentiment["label"], emotion)
 
             industry = detect_industry(data.text) if not data.industry or "auto" in data.industry.lower() else data.industry
             product_category = detect_product_category(data.text) if not data.product_category or "auto" in data.product_category.lower() else data.product_category
@@ -119,7 +120,8 @@ async def analyze(data: ReviewInput, x_api_key: str = Header(None)):
                 "emotion": emotion,
                 "product_category": product_category,
                 "device": "Web",
-                "industry": industry
+                "industry": industry,
+                "churn_risk": churn_risk
             }
 
         if data.follow_up: