naprawianie all

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import os
 # import matplotlib.pyplot as plt
 
-from utils import fetch_runs_to_df, fetch_run
+from utils import fetch_runs_to_df, fetch_run, fetch_models_to_df
 
 # Access the API key from the environment variable
 wandb_api_key = os.getenv('WANDB_API_KEY')
@@ -84,27 +84,8 @@ if option == "Models":
 
     # Ensure the DataFrame is not empty
     if not df.empty:
-        # Fetch metrics for ranking (e.g., accuracy or loss)
-        ranking_data = []
-        for index, row in df.iterrows():
-            try:
-                # Fetch the run details
-                run = api.run(f"{projects[selected_project]['entity']}/{projects[selected_project]['project']}/{row['ID']}")
-                metrics = run.summary
-                model_name = run.config.get("model_name", "Unknown")  # Fetch model name from the config, defaulting to "Unknown"
-                
-                ranking_data.append({
-                    "Model Name": model_name,  # Add model name to the table
-                    "Run Name": row["Run Name"],
-                    "ID": row["ID"],
-                    "Accuracy": metrics.get("accuracy"),  # Example metric
-                    "Loss": metrics.get("loss")  # Example metric
-                })
-            except wandb.errors.CommError:
-                continue
-
         # Convert to DataFrame
-        ranking_df = pd.DataFrame(ranking_data)
+        ranking_df = fetch_models_to_df(api, projects, selected_project, df)
 
         # Rank by Accuracy (or another metric)
         ranking_df = ranking_df.sort_values(by="Accuracy", ascending=False).reset_index(drop=True)

utils.py

@@ -57,4 +57,43 @@ def fetch_run(api, projects, selected_project, selected_run_id):
         project = projects[selected_project]["project"]
         run = api.run(f"{entity}/{project}/{selected_run_id}")
 
-    return run
+    return run
+
+
+def fetch_models_to_df(api, projects, selected_project, df):
+    data = []
+    for index, row in df.iterrows():
+        try:
+            if selected_project == "All":
+                # Determine the project for the current run
+                for project_name, details in projects.items():
+                    entity = details["entity"]
+                    project = details["project"]
+                    try:
+                        run = api.run(f"{entity}/{project}/{row['ID']}")
+                        break
+                    except wandb.errors.CommError:
+                        continue
+                else:
+                    st.error(f"Run ID {row['ID']} not found in any project.")
+                    continue
+            else:
+                entity = projects[selected_project]["entity"]
+                project = projects[selected_project]["project"]
+                run = api.run(f"{entity}/{project}/{row['ID']}")
+
+            metrics = run.summary
+            model_name = run.config.get("model_name", "Unknown")  # Fetch model name from the config, defaulting to "Unknown"
+            
+            data.append({
+                "Model Name": model_name,  # Add model name to the table
+                "Run Name": row["Run Name"],
+                "ID": row["ID"],
+                "Accuracy": metrics.get("accuracy"),  # Example metric
+                "Loss": metrics.get("loss")  # Example metric
+            })
+        except wandb.errors.CommError:
+            continue
+
+    data_df = pd.DataFrame(data)
+    return data_df