Update app.py

app.py

@@ -1,28 +1,21 @@
 #Emotion Detection API using FastAPI
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
-from model import predict_emotions, calculate_distress
+import gradio as gr
+from model import EmotionModel
+from utils import calculate_distress
 
-app = FastAPI()
+emotion_model = EmotionModel()
 
-class TextInput(BaseModel):
-    text: str
+def analyze(text):
+    emotions = emotion_model.predict(text)
+    distress = calculate_distress(emotions)
+    return {"emotions": emotions, "distress_score": distress}
 
-@app.post("/analyze")
-async def analyze(input: TextInput):
-    try:
-        emotions = predict_emotions(input.text)
-        distress_score = calculate_distress(emotions)
 
-        return {
-            "text": input.text,
-            "emotions": emotions,
-            "distress_score": distress_score,
-            "warning": distress_score > 1.5  # Trigger warning if score high
-        }
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+app = gr.Interface(
+    fn=analyze,
+    inputs=gr.Textbox(),
+    outputs="json"
+)
+# Launch the app
+app.launch()
 
-@app.get("/")
-def root():
-    return {"message": "Emotion detector is live"}