Update app.py

app.py

@@ -3,6 +3,7 @@ import torch
 import shutil
 import logging
 import gradio as gr
+import pandas as pd
 
 from PolUVR.separator import Separator
 
@@ -122,6 +123,46 @@ DEMUCS_MODELS = [
     'htdemucs_ft.yaml',
 ]
 
+def load_data(file_path):
+    df = pd.read_csv(file_path)
+    return df
+
+def sort_data(architecture, purpose, sdr_metric):
+    df = load_data("models.csv")
+
+    if architecture != "All":
+        df = df[df['Architecture'] == architecture]
+    if purpose != "All":
+        df = df[df['Purpose'] == purpose]
+    if sdr_metric != "All":
+        df = df.sort_values(by=sdr_metric, ascending=False)
+    
+    return df
+
+def create_leaderboard_tab():
+    architectures = ["All", "Roformer", "MDXC", "MDX", "VR-ARCH", "DEMUCS", "Other"]
+    purposes = ["All", "VocInst", "DeEcho", "DeReverb", "DeNoise", "Karaoke", "Crowd", "Other"]
+    sdr_metrics = ["All", "Vocal SDR", "Instrumental SDR", "Drums SDR", "Bass SDR", "Guitar SDR", "Piano SDR", "Other SDR"]
+    
+    with gr.Row():
+        architecture_dropdown = gr.Dropdown(choices=architectures, label="Architecture")
+        purpose_dropdown = gr.Dropdown(choices=purposes, label="Purpose")
+        sdr_metric_dropdown = gr.Dropdown(choices=sdr_metrics, label="SDR Metric")
+    
+    with gr.Row():
+        sort_button = gr.Button("Sort Models", variant="primary")
+    
+    with gr.Row():
+        leaderboard_output = gr.Dataframe(label="Leaderboard", interactive=False)
+    
+    sort_button.click(
+        sort_data,
+        inputs=[architecture_dropdown, purpose_dropdown, sdr_metric_dropdown],
+        outputs=leaderboard_output
+    )
+
+    return leaderboard_output
+
 def print_message(input_file, model_name):
     """Prints information about the audio separation process."""
     base_name = os.path.splitext(os.path.basename(input_file))[0]
@@ -486,7 +527,12 @@ with gr.Blocks(
                 guitar_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Guitar Stem", info="Output example: Music_(Guitar)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
                 piano_stem = gr.Textbox(value="NAME_(STEM)_MODEL", label="Piano Stem", info="Output example: Music_(Piano)_BS-Roformer-Viperx-1297", placeholder="NAME_(STEM)_MODEL")
 
-    with gr.TabItem("Credits"):
+    with gr.Tab("Leaderboard"):
+        gr.Markdown("## Model Leaderboard")
+        gr.Markdown("Sort models by architecture, purpose, and SDR metrics.")
+        create_leaderboard_tab()
+    
+    with gr.Tab("Credits"):
         gr.Markdown(
             """
             PolUVR created by **[Politrees](https://github.com/Bebra777228)**.