Update app.py

app.py

@@ -5,7 +5,7 @@ from pathlib import Path
 from huggingface_hub import HfApi, Repository
 
 # Matplotlib style
-plt.rcParams.update({"font.family":"sans-serif", "font.size":10})
+plt.rcParams.update({"font.family":"sans-serif","font.size":10})
 
 # Global DataFrame
 df = pd.DataFrame()
@@ -13,21 +13,19 @@ df = pd.DataFrame()
 def upload_csv(file):
     global df
     df = pd.read_csv(file.name)
-    if "text" not in df.columns or "label" not in df.columns:
+    if not {"text","label"}.issubset(df.columns):
         return (
-            None,  # hide table
+            None, 
             "❌ CSV must contain 'text' and 'label' columns.",
-            gr.update(visible=False),  # hide actions
-            gr.update(visible=False),  # hide outputs
-            gr.update(visible=False),  # hide push accordion
+            gr.update(visible=False), # hide action buttons
+            gr.update(visible=False), # hide push
         )
     df["label"] = df["label"].fillna("")
     return (
-        df[["text","label"]],                 # show table
-        "✅ Uploaded! Edit below or use the buttons.",
-        gr.update(visible=True),              # show action row
-        gr.update(visible=True),              # show output row
-        gr.update(visible=True),              # show push accordion
+        df[["text","label"]],
+        "✅ Uploaded! You can now annotate and use the buttons below.",
+        gr.update(visible=True),  # show action buttons
+        gr.update(visible=True),  # show push accordion
     )
 
 def save_changes(table):
@@ -42,30 +40,32 @@ def download_csv():
     return path
 
 def make_figure():
-    global df
     counts = df["label"].value_counts().sort_values(ascending=False)
-    labels, values = counts.index.tolist(), counts.values.tolist()
-    fig, (ax_table, ax_bar) = plt.subplots(
+    labels, values = list(counts.index), list(counts.values)
+    fig, (ax1, ax2) = plt.subplots(
         ncols=2,
         gridspec_kw={"width_ratios":[1,2]},
-        figsize=(8, max(2, len(labels)*0.4)),
+        figsize=(8, max(2,len(labels)*0.4)),
         tight_layout=True
     )
-    # Table
-    ax_table.axis("off")
-    data = [[l,v] for l,v in zip(labels, values)]
-    tbl = ax_table.table(cellText=data, colLabels=["Label","Count"], loc="center")
+    # table
+    ax1.axis("off")
+    tbl = ax1.table(
+        cellText=[[l,v] for l,v in zip(labels,values)],
+        colLabels=["Label","Count"],
+        loc="center"
+    )
     tbl.auto_set_font_size(False); tbl.set_fontsize(10); tbl.scale(1,1.2)
-    # Bar chart
-    ax_bar.barh(labels, values, color="#222")
-    ax_bar.invert_yaxis(); ax_bar.set_xlabel("Count")
+    # bar chart
+    ax2.barh(labels, values, color="#222222")
+    ax2.invert_yaxis(); ax2.set_xlabel("Count")
     return fig
 
 def visualize_and_download():
     fig = make_figure()
-    png_path = "label_distribution.png"
-    fig.savefig(png_path, dpi=150, bbox_inches="tight")
-    return fig, png_path
+    out_png = "label_distribution.png"
+    fig.savefig(out_png, dpi=150, bbox_inches="tight")
+    return fig, out_png
 
 def push_to_hub(repo_name, hf_token):
     global df
@@ -84,7 +84,7 @@ def push_to_hub(repo_name, hf_token):
             use_auth_token=hf_token
         )
         df.to_csv(local_dir/"data.csv", index=False)
-        repo.push_to_hub(commit_message="📑 Updated data")
+        repo.push_to_hub(commit_message="📑 Updated annotated data")
         return f"🚀 Pushed to datasets/{repo_name}"
     except Exception as e:
         return f"❌ Push failed: {e}"
@@ -93,58 +93,52 @@ with gr.Blocks() as app:
     gr.Markdown("## 🏷️ Label It! Text Annotation Tool")
     gr.Markdown("Upload a `.csv` (columns: **text**, **label**), then annotate, export, visualize, or push.")
 
-    # Step 1: Upload
+    # STEP 1: Upload
     with gr.Row():
-        csv_input = gr.File(label="📁 Upload CSV", file_types=[".csv"])
+        csv_in = gr.File(label="📁 Upload CSV", file_types=[".csv"])
         upload_btn = gr.Button("Upload")
 
-    # Table + Status
-    table = gr.Dataframe(headers=["text","label"], interactive=True, visible=False)
-    status = gr.Textbox(label="Status", interactive=False)
-
-    # Step 2: Actions (hidden initially)
-    with gr.Row(visible=False) as action_row:
-        save_btn      = gr.Button("💾 Save")
-        download_btn  = gr.Button("⬇️ Download CSV")
-        visualize_btn = gr.Button("📊 Visualize")
-
-    # Outputs row (hidden initially)
-    with gr.Row(visible=False) as output_row:
-        download_csv_out   = gr.File(label="📥 CSV File")
-        chart_plot         = gr.Plot(label="Label Distribution")
-        download_chart_out = gr.File(label="📥 Chart PNG")
-
-    # Push section
-    push_accordion = gr.Accordion("📦 Push to Hugging Face Hub", open=False, visible=False)
-    with push_accordion:
-        repo_in     = gr.Textbox(label="Repo (username/dataset)")
-        token_in    = gr.Textbox(label="🔑 HF Token", type="password")
-        push_btn    = gr.Button("🚀 Push")
-        push_status = gr.Textbox(label="Push Status", interactive=False)
-
-    # Bind events
+    # Editable table + status
+    table     = gr.Dataframe(headers=["text","label"], interactive=True, visible=False)
+    status    = gr.Textbox(label="Status", interactive=False)
+
+    # STEP 2: Action buttons (hidden initially)
+    with gr.Row(visible=False) as actions:
+        save_btn         = gr.Button("💾 Save")
+        download_csv_btn = gr.DownloadButton(fn=download_csv, label="⬇️ Download CSV", file_name="annotated_data.csv")
+        download_png_btn = gr.DownloadButton(fn=visualize_and_download, label="⬇️ Download Chart", file_name="label_distribution.png")
+        visualize_btn    = gr.Button("📊 Visualize")
+
+    chart_plot = gr.Plot(visible=False)
+
+    # Push accordion
+    push_acc   = gr.Accordion("📦 Push to Hugging Face Hub", open=False, visible=False)
+    with push_acc:
+        repo_in   = gr.Textbox(label="Repo (username/dataset)")
+        token_in  = gr.Textbox(label="🔑 HF Token", type="password")
+        push_btn  = gr.Button("🚀 Push")
+        push_out  = gr.Textbox(label="Push Status", interactive=False)
+
+    # Event wiring
     upload_btn.click(
         upload_csv,
-        inputs=csv_input,
-        outputs=[table, status, action_row, output_row, push_accordion]
+        inputs=csv_in,
+        outputs=[table, status, actions, push_acc]
     )
     save_btn.click(
         save_changes,
         inputs=table,
         outputs=status
     )
-    download_btn.click(
-        download_csv,
-        outputs=download_csv_out
-    )
     visualize_btn.click(
         visualize_and_download,
-        outputs=[chart_plot, download_chart_out]
+        inputs=None,
+        outputs=[chart_plot, download_png_btn]
     )
     push_btn.click(
         push_to_hub,
         inputs=[repo_in, token_in],
-        outputs=push_status
+        outputs=push_out
     )
 
 app.launch()