Update src/main.py

src/main.py

@@ -87,7 +87,7 @@ def get_leaderboard_data():
     WITH BestScores AS (
         SELECT 
             username, 
-            MAX(bbox_mAP) as max_bbox_mAP 
+            MAX(segm_mAP) as max_segm_mAP 
         FROM submissions 
         GROUP BY username
     )
@@ -99,13 +99,13 @@ def get_leaderboard_data():
         s.segm_AP50, 
         MAX(s.timestamp) as last_submission -- MAX(timestamp) to get the most recent best submission
     FROM submissions s
-    INNER JOIN BestScores b ON s.username = b.username AND s.bbox_mAP = b.max_bbox_mAP
+    INNER JOIN BestScores b ON s.username = b.username AND s.segm_mAP = b.max_segm_mAP
     GROUP BY s.username
-    ORDER BY s.bbox_mAP DESC, s.timestamp ASC
+    ORDER BY s.segm_mAP DESC, s.timestamp ASC
     """
     df = pd.read_sql_query(query, conn)
     # Rename the column for display clarity
-    df = df.rename(columns={'bbox_mAP': 'Best_bbox_mAP'})
+    df = df.rename(columns={'segm_mAP': 'Best_segm_mAP'})
     
     conn.close()
     return df
@@ -205,14 +205,14 @@ Copyright © 2015, VizWiz team. All rights reserved. Redistribution and use soft
                     with st.spinner('Calculating score against Ground Truth...'):
                         # Ensure the ground truth path is correct
                         bbox_mAP, bbox_AP50, segm_mAP, segm_AP50  = evaluate_submission("src/biv_query.json", save_path)
-                        st.success(f"bbox_mAP, bbox_AP50, segm_mAP, segm_AP50 : {bbox_mAP, bbox_AP50, segm_mAP, segm_AP50}")
+                        st.success(f"bbox_mAP, bbox_AP50, segm_mAP, segm_AP50 : {bbox_mAP:.2f, bbox_AP50:.2f, segm_mAP:.2f, segm_AP50:.2f}")
                         if bbox_mAP is not None and bbox_AP50 is not None and segm_mAP is not None and segm_AP50 is not None:
-                            save_submission(st.session_state['username'], bbox_mAP, bbox_AP50, segm_mAP, segm_AP50 )
+                            save_submission(st.session_state['username'], bbox_mAP:.4f, bbox_AP50:.4f, segm_mAP:.4f, segm_AP50:.4f )
                             st.balloons()
                             st.success("Submission Successful!")
                             
                             # Display Metrics
-                            st.metric(label="Result : ", value=f"{bbox_mAP, bbox_AP50, segm_mAP, segm_AP50}%")
+                            # st.metric(label="Result : ", value=f"{bbox_mAP:.4f, bbox_AP50:.4f, segm_mAP:.4f, segm_AP50:.4f}")
         
 
     # --- PAGE: LEADERBOARD (Existing Code) ---