Test 3

app.py

@@ -71,53 +71,40 @@ def update_leader_info(leader_info, competition, best_model):
             leader_info[competition]["UID"] = best_model["ID"]
             leader_info[competition]["Days on Top"] = 1
     return leader_info[competition]
-@st.cache_data()
-def load_competition_data(last_update_time=None):
-    competition_summaries = {}
-    model_evaluations = {}
-    for competition in projects:
-        competition_summaries[competition] = fetch_competition_summary(api, projects, competition)
-        model_evaluations[competition] = fetch_models_evaluation(api, projects, competition)
-    
-    last_update_time = time.time()
-    return competition_summaries, model_evaluations, last_update_time
+@st.cache_data(ttl=60)
+def load_competition_data():
+    ranking_dfs = {}
+    for competition, details in projects.items():
+        df = fetch_runs_to_df(api, projects, competition)
+        if not df.empty:
+            ranking_df = fetch_models_to_df(api, projects, competition, df)
+            rank_by = ["Recall", "Accuracy", "ROC AUC"]
+            ascending = [False, False, False]
+            ranking_df = ranking_df.sort_values(by=rank_by, ascending=ascending)
+            ranking_dfs[competition] = ranking_df
+    return ranking_dfs
 
 # Streamlit app main function
 def main():
     st.title("LaUltimate Dashboard")
-    st.title("JEDZIESZMALINA")
     st.subheader("Welcome to the Competition Dashboard!")
-    st.write("Explore the various AI competitions and their respective rankings. Select a competition to view more details and rankings.")
 
-    update_interval = 20 * 60  # 20 minutes in seconds
+    # Load competition data
+    ranking_dfs = load_competition_data()
 
-    if 'last_update_time' not in st.session_state:
-        st.session_state.last_update_time = None
+    # Initialize or update leader_info
     if "leader_info" not in st.session_state:
         st.session_state.leader_info = {}
 
-    if st.session_state.last_update_time is None or (time.time() - st.session_state.last_update_time > update_interval):
-        competition_summaries, model_evaluations, st.session_state.last_update_time = load_competition_data(st.session_state.last_update_time)
-
-        for competition in projects:
-            if not model_evaluations[competition].empty:
-                best_model = model_evaluations[competition].iloc[0]
-                st.session_state.leader_info[competition] = update_leader_info(st.session_state.leader_info, competition, best_model)
-            else:
-                st.session_state.leader_info[competition] = {
-                    "Username": "N/A", "Model Name": "N/A", "Hotkey": "N/A",
-                    "Date": "N/A", "UID": "N/A", "Days on Top": "N/A"
-                }
-    else:
-        competition_summaries, model_evaluations, _ = load_competition_data(st.session_state.last_update_time)
-
-    st.write(f"Last updated: {time.ctime(st.session_state.last_update_time)}")
+    # Update leader info for each competition
+    for competition in ranking_dfs:
+        if not ranking_dfs[competition].empty:
+            best_model = ranking_dfs[competition].iloc[0]
+            st.session_state.leader_info[competition] = update_leader_info(st.session_state.leader_info, competition, best_model)
 
     st.markdown("### Competitions")
 
-    st.write("### Select Competition")
-
-    # Create a header for the table
+    # Create a header for the table with additional columns
     cols = st.columns([1, 3, 2, 2, 2, 2, 1, 2])
     cols[0].write("Index")
     cols[1].write("Competition Name")
@@ -128,13 +115,14 @@ def main():
     cols[6].write("Hotkey")
     cols[7].write("Days on Top")
 
+    # Display the project names in a table-like structure with buttons and placeholders for additional data
     for index, (competition, details) in enumerate(projects.items(), start=1):
-        leader_info = st.session_state.get("leader_info", {}).get(competition, {})
-
+        leader_info = st.session_state.leader_info.get(competition, {})
+        
         cols = st.columns([1, 3, 2, 2, 2, 2, 1, 2])
         cols[0].write(index)
-        if cols[1].button(competition):
-            st.query_params["comp"] = competition
+        if cols[1].button(competition, key=f"btn_{competition}"):
+            st.session_state.selected_competition = competition
         cols[2].write(leader_info.get("Username", "N/A"))
         cols[3].write(leader_info.get("Model Name", "N/A"))
         cols[4].write(leader_info.get("Date", "N/A"))
@@ -142,38 +130,46 @@ def main():
         cols[6].write(leader_info.get("Hotkey", "N/A"))
         cols[7].write(leader_info.get("Days on Top", "N/A"))
 
-    if hasattr(st, 'query_params'):
-        query_params = st.query_params
-        competition_name = query_params.get("comp")
-    else:
-    # Fallback for older Streamlit versions
-        competition_name = st.selectbox("Select a competition", list(projects.keys()))
-
-    if competition_name:
+    # Display selected competition details
+    if "selected_competition" in st.session_state:
+        competition_name = st.session_state.selected_competition
         st.write(f"Selected competition: {competition_name}")
-        competition_details = projects.get(competition_name, {})
-        description = competition_details.get("description", "No description available.")
         
-        st.header(f"Competition: {competition_name}")
-        st.write(description)
-
-        st.subheader("Competition Summary")
-        competition_summary_df = competition_summaries.get(competition_name, pd.DataFrame())
-        if not competition_summary_df.empty:
-            st.dataframe(competition_summary_df)
-        else:
-            st.warning("No competition summary data available.")
-
-        st.subheader("Models Evaluation")
-        models_evaluation_df = model_evaluations.get(competition_name, pd.DataFrame())
-        if not models_evaluation_df.empty:
-            st.dataframe(models_evaluation_df)
-        else:
-            st.warning("No models evaluation data available.")
+        try:
+            competition_details = projects.get(competition_name, {})
+            description = competition_details.get("description", "No description available.")
+            
+            st.header(f"Competition: {competition_name}")
+            st.write(description)
+
+            df = ranking_dfs.get(competition_name, pd.DataFrame())
+            if not df.empty:
+                st.subheader("Competition Summary")
+                try:
+                    competition_summary_df = fetch_competition_summary(api, projects, competition_name, df)
+                    if not competition_summary_df.empty:
+                        st.dataframe(competition_summary_df)
+                    else:
+                        st.warning("No competition summary data available.")
+                except Exception as e:
+                    st.error(f"Error fetching competition summary: {str(e)}")
+
+                st.subheader("Models Evaluation")
+                try:
+                    models_evaluation_df = fetch_models_to_df(api, projects, competition_name, df)
+                    if not models_evaluation_df.empty:
+                        st.dataframe(models_evaluation_df)
+                    else:
+                        st.warning("No models evaluation data available.")
+                except Exception as e:
+                    st.error(f"Error fetching models evaluation: {str(e)}")
+            else:
+                st.warning("No runs available for ranking.")
+        except Exception as e:
+            st.error(f"An error occurred while processing the competition data: {str(e)}")
     else:
         st.write("Please select a competition to view details.")
 
-
 # Run the app
 if __name__ == "__main__":
     main()