Add submission form to leaderboard

app.py

@@ -1,35 +1,57 @@
 import gradio as gr
 import pandas as pd
+import os
 
 CSV_PATH = "results.csv"
 
-# 기존 데이터 불러오기
-df = pd.read_csv(CSV_PATH)
-
-def submit(model, acc, f1, time, link):
-    new_row = {
-        "Model Name": model,
-        "Accuracy": float(acc),
-        "F1 Score": float(f1),
-        "Inference Time (s)": float(time),
-        "Hugging Face Link": link
-    }
-    # 기존 + 새로운 데이터
-    updated = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
-    updated.to_csv(CSV_PATH, index=False)
-    return updated
+# CSV가 없으면 빈 테이블 생성
+if not os.path.exists(CSV_PATH):
+    df_init = pd.DataFrame(columns=[
+        "Model Name", "Accuracy", "F1 Score", "Inference Time (s)", "Hugging Face Link"
+    ])
+    df_init.to_csv(CSV_PATH, index=False)
 
+# 제출 처리 함수
+def submit_result(model, acc, f1, time, link):
+    try:
+        df = pd.read_csv(CSV_PATH)
+        new_row = {
+            "Model Name": model.strip(),
+            "Accuracy": float(acc),
+            "F1 Score": float(f1),
+            "Inference Time (s)": float(time),
+            "Hugging Face Link": link.strip()
+        }
+        updated_df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
+        updated_df.to_csv(CSV_PATH, index=False)
+        return "✅ Submitted!", updated_df
+    except Exception as e:
+        return f"❌ Error: {str(e)}", pd.read_csv(CSV_PATH)
+
+# Gradio 앱
 with gr.Blocks() as demo:
-    gr.Markdown("# 📊 Model Leaderboard")
+    gr.Markdown("# 📊 Model Leaderboard + Submission")
+
     with gr.Row():
         model = gr.Textbox(label="Model Name")
-        acc = gr.Textbox(label="Accuracy")
-        f1 = gr.Textbox(label="F1 Score")
+        acc = gr.Textbox(label="Accuracy (e.g. 0.89)")
+        f1 = gr.Textbox(label="F1 Score (e.g. 0.87)")
         time = gr.Textbox(label="Inference Time (s)")
         link = gr.Textbox(label="Hugging Face Link")
+
     submit_btn = gr.Button("Submit Result")
-    output_df = gr.Dataframe(value=df, headers=list(df.columns), interactive=False)
+    status = gr.Textbox(label="Submission Status", interactive=False)
+    table = gr.Dataframe(
+        value=pd.read_csv(CSV_PATH),
+        headers=["Model Name", "Accuracy", "F1 Score", "Inference Time (s)", "Hugging Face Link"],
+        interactive=False,
+        label="Current Leaderboard"
+    )
 
-    submit_btn.click(fn=submit, inputs=[model, acc, f1, time, link], outputs=output_df)
+    submit_btn.click(
+        fn=submit_result,
+        inputs=[model, acc, f1, time, link],
+        outputs=[status, table]
+    )
 
 demo.launch()