add results table

app.py

@@ -501,18 +501,25 @@ def inference(in_video, use_60fps,
             else: # end of event
                 pass 
     print(f"Recovery times: {recovery_times}")
-            
+    jumps_per_second = np.clip(1 / ((periodLength / fps) + 0.0001), 0, 10)
+    jumping_speed = np.copy(jumps_per_second)
 
-    if LOCAL:
-        if both_feet:
-            count_msg = f"## Count (both feet): {count_pred:.1f}, Marks: {marks_count_pred:.1f}, Phase: {phase_count:.1f}, Confidence: {total_confidence:.2f}, Time to Speed: {time_to_speed:.2f} seconds"
-        else:
-            count_msg = f"## Count (one foot): {count_pred:.1f}, Marks: {marks_count_pred:.1f}, Phase: {phase_count:.1f}, Confidence: {total_confidence:.2f}, Time to Speed: {time_to_speed:.2f} seconds"
-    else:
-        if both_feet:
-            count_msg = f"## Count (both feet): {count_pred:.1f}, Confidence: {total_confidence:.2f}"
-        else:
-            count_msg = f"## Count (one foot): {count_pred:.1f}, Confidence: {total_confidence:.2f}"
+    foot_label = "both feet" if both_feet else "one foot"
+    count_msg = f"## 🏅 Results ({foot_label}: {count_pred:.1f})\n\n"
+    count_msg += f"| Metric | Value |\n|---|---|\n"
+    count_msg += f"| **Count** | {count_pred:.1f} |\n"
+    count_msg += f"| **Confidence** | {total_confidence:.2f} |\n"
+    count_msg += f"| **Fastest Speed** | {fastest_jumps_per_second:.2f} jumps/sec |\n"
+    count_msg += f"| **Average Speed** | {np.mean(jumps_per_second[periodicity_mask]):.2f} jumps/sec |\n"
+    count_msg += f"| **Slowest Speed** | {lowest_jps:.2f} jumps/sec |\n"
+    count_msg += f"| **Slowdown** | {abs(slowdown):.2f} jumps/sec ({abs(slowdown_percent):.1f}%) |\n"
+    if num_misses > 0:
+        count_msg += f"| **Misses** | {num_misses} |\n"
+        if recovery_times:
+            avg_recovery = sum(recovery_times) / len(recovery_times)
+            count_msg += f"| **Avg Recovery Time** | {avg_recovery:.2f}s |\n"
+    if beep_detection_on and time_to_speed > 0:
+        count_msg += f"| **Time to Speed** | {time_to_speed:.2f}s |\n"
 
     if api_call:
         if CACHE_API_CALLS:
@@ -599,8 +606,7 @@ def inference(in_video, use_60fps,
     # plt.close()
     
 
-    jumps_per_second = np.clip(1 / ((periodLength / fps) + 0.0001), 0, 10)
-    jumping_speed = np.copy(jumps_per_second)
+    
     misses = periodicity < miss_threshold
     jumps_per_second[misses] = 0
     frame_type = np.array(['miss' if miss else 'frame' for miss in misses])
@@ -624,13 +630,11 @@ def inference(in_video, use_60fps,
     fig = px.scatter(data_frame=df,
                     x='seconds', 
                     y='jumps per second',
-                    #symbol='frame_type',
-                    #symbol_map={'frame': 'circle', 'miss': 'circle-open', 'jump': 'triangle-down'},
-                    color='event_type',
+                    color='jumping speed',
                     size='jumps_size',
                     size_max=8,
-                    color_continuous_scale=[(t, c) for t, c in zip(event_type_tick_vals, event_type_colors)],
-                    range_color=(0,1),
+                    color_continuous_scale='Turbo',
+                    range_color=(0, 10),
                     title="Jumping speed (jumps-per-second)",
                     trendline='rolling',
                     trendline_options=dict(window=16),
@@ -662,9 +666,7 @@ def inference(in_video, use_60fps,
     # remove white outline from marks
     fig.update_traces(marker_line_width = 0)
     fig.update_layout(coloraxis_colorbar=dict(
-        tickvals=event_type_tick_vals,
-        ticktext=['single<br>rope', 'double<br>dutch', 'double<br>unders', 'single<br>bounces', 'double<br>bounces', 'triple<br>unders', 'other'],
-        title='event type'
+        title='jumps/sec'
     ))
 
 
@@ -812,7 +814,7 @@ with gr.Blocks() as demo:
     api_dummy_button.click(api_inference, [in_video, use_60fps, beep_detection_on, event_length, count_only, api_token], 
                            outputs=[period_length], api_name='inference')
     examples = [
-        ['files/wc2023.mp4', True, True, 30, 0.5, 0.5],
+        ['files/wc2023.mp4', False, True, 30, 0.5, 0.5],
     ]
     gr.Examples(examples, 
                 inputs=[in_video, use_60fps, beep_detection_on, event_length, miss_threshold, marks_threshold],