Update app.py

app.py

@@ -8,6 +8,7 @@ import gradio as gr
 import os
 import torch.nn.functional as F
 from scipy.special import expit
+import json
 
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -74,12 +75,19 @@ def process_audio(input_data, segment_duration=10):
         else:
             confidence_percentage = 1 - expit(decision_score).item()
         segment_predictions.append(pred)
-        line = f"Segment {idx + 1}: {'Real' if pred == 1 else 'Fake'} (Confidence: {np.round(confidence_percentage*100, 2)}%)"
-        output_lines.append(line)
-    overall_prediction = 1 if sum(segment_predictions) > (len(segment_predictions) / 2) else 0
-    overall_line = f"Overall Prediction: {'Real' if overall_prediction == 1 else 'Fake'}"
-    output_str = overall_line + "\n" + "\n".join(output_lines)
-    return output_str
+        output_dict = {
+    "classification": "real" if sum(segment_predictions) > (len(segment_predictions) / 2) else "fake",
+    "segments": [
+        {
+            "segment": idx + 1,
+            "prediction": "real" if pred == 1 else "fake",
+            "confidence": round(conf * 100, 2)
+        }
+        for idx, (pred, conf) in enumerate(zip(segment_predictions, confidence_scores))
+    ]
+}
+    json_output = json.dumps(output_dict, indent=4)
+    return json_output
 
 def gradio_interface(audio):
     if audio: