Initial upload app.py

app.py

@@ -0,0 +1,72 @@
+import gradio as gr
+import pandas as pd
+import json
+
+def process_df(df):
+    # Set the current date and time
+    df['date'] = pd.Timestamp.now().strftime('%Y-%m-%d %H:%M')
+    
+    # Create clickable links in the "Model" column
+    df['Model'] = df.apply(lambda x: f'<a href="{x["Link"]}" target="_blank">{x["Model"]}</a>', axis=1)
+    
+    # Drop the "Link" column
+    df = df.drop(['Link'], axis=1)
+    
+    # Convert columns to float
+    cols = df.columns.drop(['Model', 'date'])
+    df[cols] = df[cols].astype(float)
+    
+    # Highlight the minimum values in the columns
+    styled_df = df.style.highlight_min(color='lightgreen', axis=0, subset=cols)
+    
+    return styled_df
+
+def get_eval(jsonfile):
+    global leaderboard_df
+    
+    # Load JSON data
+    data = json.load(open(jsonfile))
+    
+    # Convert JSON data to DataFrame
+    df = pd.DataFrame.from_dict(data, orient='columns')
+    
+    # Concatenate with the existing leaderboard
+    leaderboard_df = pd.concat([leaderboard_df, df], ignore_index=True)
+    
+    # Process the dataframe
+    df = process_df(leaderboard_df)
+    
+    return df
+
+def refresh_leaderboard():
+    global leaderboard_df
+    if len(leaderboard_df) == 0:
+        return leaderboard_df
+    else:
+        return process_df(leaderboard_df)
+
+# Initialize an empty dataframe for the leaderboard
+columns = ['Model', 'date', 'ucfd_eng', 'ucfd_lug',
+           'sema_eng', 'sema_lug', 'trac_fm_lug']
+leaderboard_df = pd.DataFrame(columns=columns)
+
+# Create Gradio interface
+block = gr.Blocks()
+
+with block:
+    gr.Markdown("## Leaderboard for Sunbird AI ASR")
+    
+    with gr.Row():
+        input_json = gr.UploadButton("Upload a JSON file", file_count="single", scale=0)
+    
+    with gr.Tabs():
+        with gr.TabItem("ASR Leaderboard"):
+            with gr.Row():
+                data = gr.DataFrame(datatype="markdown")
+                
+    input_json.upload(get_eval, inputs=input_json, outputs=data)
+    
+    # Running the function on page load in addition to when the button is clicked
+    block.load(refresh_leaderboard, inputs=None, outputs=data)
+
+block.launch(share=True)