change output

app.py

@@ -51,19 +51,35 @@ def predict(data):
         predictions = [x if x == max_count else 0 for x in predictions]
 
         #Result
-        result = (
-            f"bicep: {predictions[0]} | abs: {predictions[1]} | "
-            f"chess: {predictions[2]} | legs: {predictions[3]}"
-        )
-        return result
+        # result = (
+        #     f"bicep: {predictions[0]} | abs: {predictions[1]} | "
+        #     f"chess: {predictions[2]} | legs: {predictions[3]}"
+        # )
+        class_names = ["dumbbells", "situps", "pushups", "squats"]
+        if max_count == 0:
+            # 沒有任何類別達到閾值
+            result_json = {
+                "type": None,
+                "reps": 0
+            }
+        else:
+            # 找出哪個類別擁有 max_count
+            # 如果有多個類別並列最高，這裡僅選第一個
+            max_index = predictions.index(max_count)
+            result_json = {
+                "type": class_names[max_index],
+                "reps": max_count
+            }
+
+        return result_json
     except Exception as e:
-        return f"Error processing file: {str(e)}"
+        return {"error": str(e)}
 
 # 定义 Gradio 界面
 iface = gr.Interface(
     fn=predict,
     inputs=gr.JSON(label="json file"),
-    outputs=gr.Textbox(label="Prediction Result"),  
+    outputs=gr.JSON(label="Prediction Result"),  
     title="Burnsync classifier",
     description="Upload a JSON file with 6-channel IMU time series data to predict the action class."
 )