Update app.py

app.py

@@ -4,27 +4,31 @@ import io
 import os
 from huggingface_hub import HfApi, Repository
 
-# Global state
+# Global variable to store the DataFrame
 df = pd.DataFrame()
 
+# Upload CSV
 def upload_csv(file):
     global df
     df = pd.read_csv(file.name)
     if "text" not in df.columns or "label" not in df.columns:
-        return None, "❌ CSV must have 'text' and 'label' columns."
+        return None, "❌ CSV must contain 'text' and 'label' columns."
     df["label"] = df["label"].fillna("")
-    return df[["text", "label"]], "✅ File uploaded. You can now edit labels."
+    return df[["text", "label"]], "✅ File uploaded. You can now annotate."
 
-def save_changes(updated_table):
+# Save changes from the editable table
+def save_changes(edited_data):
     global df
-    df = pd.DataFrame(updated_table, columns=["text", "label"])
+    df = pd.DataFrame(edited_data, columns=["text", "label"])
     return "✅ Changes saved."
 
+# Download the updated CSV
 def download_csv():
     global df
     csv_bytes = df.to_csv(index=False).encode("utf-8")
     return gr.File.update(value=io.BytesIO(csv_bytes), filename="annotated_data.csv")
 
+# Push to Hugging Face Hub
 def push_to_hub(repo_name, hf_token):
     global df
     repo_url = f"https://huggingface.co/datasets/{repo_name}"
@@ -33,23 +37,22 @@ def push_to_hub(repo_name, hf_token):
     if os.path.exists(local_path):
         os.system(f"rm -rf {local_path}")
     
-    api = HfApi()
-    api.create_repo(repo_id=repo_name, token=hf_token, repo_type="dataset", exist_ok=True)
+    HfApi().create_repo(repo_id=repo_name, token=hf_token, repo_type="dataset", exist_ok=True)
     repo = Repository(local_dir=local_path, clone_from=repo_url, token=hf_token)
-    
     df.to_csv(f"{local_path}/data.csv", index=False)
     repo.push_to_hub()
     
-    return f"✅ Successfully pushed to: {repo_url}"
+    return f"🚀 Successfully pushed to: {repo_url}"
 
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as app:
+# Gradio app
+with gr.Blocks(title="CSV Labeling App") as app:
     gr.Markdown("# 🏷️ Label it! Text Annotation Tool")
-    gr.Markdown("Upload your CSV with `text` and `label` columns. Edit labels, save your work, and export or publish.")
+    gr.Markdown("Upload a `.csv` with `text` and `label` columns. You can annotate, save, and publish your data.")
 
     with gr.Row():
         file_input = gr.File(label="📁 Upload CSV", file_types=[".csv"])
-        upload_btn = gr.Button("Upload")
-    
+        upload_button = gr.Button("Upload")
+
     dataframe = gr.Dataframe(
         headers=["text", "label"],
         label="📝 Annotate Labels Below",
@@ -58,20 +61,23 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as app:
         row_count=10,
         col_count=(2, "fixed")
     )
-    
-    status = gr.Textbox(visible=True, label="Status", interactive=False)
+
+    status = gr.Textbox(label="Status", interactive=False)
 
     with gr.Row():
         save_btn = gr.Button("💾 Save")
         download_btn = gr.Button("⬇️ Download CSV")
-        download_file = gr.File(label="📥 Downloaded File")
+        download_file = gr.File(label="📥 Download", interactive=False)
 
     with gr.Row():
-        repo_input = gr.Textbox(label="📦 Hugging Face Dataset Repo (e.g. username/my-dataset)")
-        token_input = gr.Textbox(label="🔑 Hugging Face Token", type="password")
-        push_btn = gr.Button("🚀 Push to HF Hub")
+        repo_input = gr.Textbox(label="📦 HF Dataset Repo (e.g. username/my-dataset)")
+        token_input = gr.Textbox(label="🔑 HF Token", type="password")
+        push_btn = gr.Button("🚀 Push to Hub")
         push_status = gr.Textbox(label="Push Status", interactive=False)
 
-    upload_btn.click(fn=upload_csv, inputs=file_input, outputs=[dataframe, status])
+    upload_button.click(fn=upload_csv, inputs=file_input, outputs=[dataframe, status])
     save_btn.click(fn=save_changes, inputs=dataframe, outputs=status)
     download_btn.click(fn=download_csv, outputs=download_file)
+    push_btn.click(fn=push_to_hub, inputs=[repo_input, token_input], outputs=push_status)
+
+app.launch()