Update app.py

app.py

@@ -4,10 +4,8 @@ import matplotlib.pyplot as plt
 from pathlib import Path
 from huggingface_hub import HfApi, Repository
 
-plt.rcParams.update({
-    "font.family": "sans-serif",
-    "font.size": 10,
-})
+# Matplotlib style
+plt.rcParams.update({"font.family":"sans-serif", "font.size":10})
 
 # Global DataFrame
 df = pd.DataFrame()
@@ -17,21 +15,19 @@ def upload_csv(file):
     df = pd.read_csv(file.name)
     if "text" not in df.columns or "label" not in df.columns:
         return (
-            None,  # table
+            None,  # hide table
             "❌ CSV must contain 'text' and 'label' columns.",
-            gr.update(visible=False),  # save
-            gr.update(visible=False),  # download CSV
-            gr.update(visible=False),  # visualize
-            gr.update(visible=False),  # push accordion
+            gr.update(visible=False),  # hide actions
+            gr.update(visible=False),  # hide outputs
+            gr.update(visible=False),  # hide push accordion
         )
     df["label"] = df["label"].fillna("")
     return (
-        df[["text","label"]],
-        "✅ File uploaded — you can now annotate and use the buttons below.",
-        gr.update(visible=True),
-        gr.update(visible=True),
-        gr.update(visible=True),
-        gr.update(visible=True),
+        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
     )
 
 def save_changes(table):
@@ -45,14 +41,13 @@ def download_csv():
     df.to_csv(path, index=False)
     return path
 
-def create_distribution_figure():
+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(
         ncols=2,
-        gridspec_kw={"width_ratios": [1,2]},
+        gridspec_kw={"width_ratios":[1,2]},
         figsize=(8, max(2, len(labels)*0.4)),
         tight_layout=True
     )
@@ -66,11 +61,11 @@ def create_distribution_figure():
     ax_bar.invert_yaxis(); ax_bar.set_xlabel("Count")
     return fig
 
-def visualize_and_download_chart():
-    fig = create_distribution_figure()
-    out_path = "label_distribution.png"
-    fig.savefig(out_path, dpi=150, bbox_inches="tight")
-    return fig, out_path
+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
 
 def push_to_hub(repo_name, hf_token):
     global df
@@ -89,53 +84,67 @@ 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="📑 Update annotated data")
-        return f"🚀 Pushed to https://huggingface.co/datasets/{repo_name}"
+        repo.push_to_hub(commit_message="📑 Updated data")
+        return f"🚀 Pushed to datasets/{repo_name}"
     except Exception as e:
         return f"❌ Push failed: {e}"
 
-with gr.Blocks(theme=gr.themes.Default()) as app:
-    gr.Markdown("## 🏷️ Label It! Text Annotation Tool\n"
-                "Upload a `.csv` (with **text** + **label** columns), "
-                "then annotate, export, visualize, or publish.")
-    
+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
     with gr.Row():
-        file_input = gr.File(label="📁 Upload CSV", file_types=[".csv"])
+        csv_input = gr.File(label="📁 Upload CSV", file_types=[".csv"])
         upload_btn = gr.Button("Upload")
-    
-    # Editable table
+
+    # Table + Status
     table = gr.Dataframe(headers=["text","label"], interactive=True, visible=False)
     status = gr.Textbox(label="Status", interactive=False)
-    
-    # Step 2 buttons (hidden initially)
+
+    # 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 Distribution")
-        download_csv_out   = gr.File(label="📥 Download CSV")
+        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="📥 Download Chart")
-    
-    # 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-name)")
-        token_in  = gr.Textbox(label="🔑 HF Token", type="password")
-        push_btn  = gr.Button("🚀 Push")
+        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)
-    
-    # Event bindings
+
+    # Bind events
     upload_btn.click(
         upload_csv,
-        inputs=file_input,
-        outputs=[table, status,
-                 save_btn, download_btn, visualize_btn, push_acc]
+        inputs=csv_input,
+        outputs=[table, status, action_row, output_row, push_accordion]
+    )
+    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]
+    )
+    push_btn.click(
+        push_to_hub,
+        inputs=[repo_in, token_in],
+        outputs=push_status
     )
-    save_btn.click(save_changes, inputs=table, outputs=status)
-    download_btn.click(download_csv, outputs=download_csv_out)
-    visualize_btn.click(visualize_and_download_chart,
-                        outputs=[chart_plot, download_chart_out])
-    push_btn.click(push_to_hub, inputs=[repo_in, token_in], outputs=push_status)
 
 app.launch()