Update app.py

app.py

@@ -1,21 +1,25 @@
-#Emotion Detection API using FastAPI
 import gradio as gr
-from model import EmotionModel
+from model import EmotionModel, SuicidalIntentModel
 from utils import calculate_distress
 
 emotion_model = EmotionModel()
+suicide_model = SuicidalIntentModel()
 
 def analyze(text):
     emotions = emotion_model.predict(text)
-    distress = calculate_distress(emotions)
-    return {"emotions": emotions, "distress_score": distress}
-
+    suicide_risk = suicide_model.predict(text)
+    distress = calculate_distress(emotions, suicide_risk)
+    return {
+        "emotions": emotions,
+        "suicidal_risk": round(suicide_risk, 2),
+        "distress_score": distress
+    }
 
 app = gr.Interface(
     fn=analyze,
-    inputs=gr.Textbox(),
-    outputs="json"
+    inputs=gr.Textbox(label="Enter journal text"),
+    outputs="json",
+    title="Emotion + Distress + Suicide Risk Analyzer"
 )
-# Launch the app
-app.launch(share=True)
 
+app.launch(share=True)