feat: change api response format

app.py

@@ -66,7 +66,7 @@ def predict_hate_speech(audio_path):
         api_name="/predict"
     )
 
-    profanityFound = any(word in stt_text.split() for word in profanity_hn)
+    profanityFound = [word for word in stt_text.split() if word in profanity_hn]
     threshold = 0.6
     isHate = (engResult[0] != "NEITHER" and engResult[1] > threshold) or (
         hingResult[0] != "NAG" and hingResult[1] > threshold) or (
@@ -77,19 +77,38 @@ def predict_hate_speech(audio_path):
     audConf = audResult['Confidence'] if audResult['Classification'] == 'Hate Speech\n' else (1 - audResult['Confidence'])
 
     confidence = (engConf + hingConf + audConf) / 3
-    # print(profanityFound, engResult, hingResult, audResult)
-    if profanityFound:
-        return ["hate", f"Result: Profanity Found", f"Text: {stt_text}"]
-    elif isHate:
-        return ["hate", f"Confidence: {confidence}", f"Text: {stt_text}"]
 
-    return ["not_hate", "No hate found, yay!"]
+    if len(profanityFound) > 0:
+        return {
+            'prediciton' : 'hate',
+            'language' : 'Hindi',
+            'label' : 'Profanity found',
+            'confidence' : None, 
+            'hate_text' : profanityFound
+        }
+        
+    if isHate:
+        return {
+            'prediction' : 'hate',
+            'language' : 'English' if engConf > hingConf else 'Hinglish',
+            'label' : None,
+            'confidence' : confidence,
+            'hate_text' : stt_text
+        }
+    
+    return {
+        'prediction' : 'not_hate',
+        'language' : None,
+        'label' : None,
+        'confidence' : None,
+        'hate_text' : None
+    }
 
 
 iface = gr.Interface(
     fn=predict_hate_speech,
     inputs=gr.Audio(type="filepath", label="Upload Audio"),
-    outputs=gr.Textbox(label="Hate Speech Analysis"),
+    outputs=gr.JSON(),
     title="Hate Speech Audio Pipeline",
     description="Upload an audio file to detect potential hate speech content.",
     examples=[