Update app.py

app.py

@@ -4,80 +4,76 @@ import io
 import os
 from huggingface_hub import HfApi, Repository
 
-# Global variable to store the DataFrame
+# Global 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 contain 'text' and 'label' columns."
+        return gr.update(visible=False), "❌ CSV must have 'text' and 'label' columns."
     df["label"] = df["label"].fillna("")
-    return df[["text", "label"]], "✅ File uploaded. You can now annotate."
+    # Show the dataframe and status
+    return (
+        gr.update(value=df[["text","label"]], visible=True),
+        "✅ File uploaded — you can now edit the labels below."
+    )
 
-# Save changes from the editable table
-def save_changes(edited_data):
+def save_changes(edited_table):
     global df
-    df = pd.DataFrame(edited_data, columns=["text", "label"])
-    return "✅ Changes saved."
+    df = pd.DataFrame(edited_table, 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}"
     local_path = f"./{repo_name.replace('/', '_')}"
-    
     if os.path.exists(local_path):
         os.system(f"rm -rf {local_path}")
-    
     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"🚀 Data pushed to {repo_url}"
 
-# Gradio app
-with gr.Blocks(title="CSV Labeling App") as app:
-    gr.Markdown("# 🏷️ Label it! Text Annotation Tool")
-    gr.Markdown("Upload a `.csv` with `text` and `label` columns. You can annotate, save, and publish your data.")
+with gr.Blocks(theme=gr.themes.Default()) as app:
+    gr.Markdown("## 🏷️ Label It! Text Annotation Tool")
+    gr.Markdown("Upload a `.csv` with `text` + `label` columns, annotate in-place, then export or publish.")
 
     with gr.Row():
         file_input = gr.File(label="📁 Upload CSV", file_types=[".csv"])
-        upload_button = gr.Button("Upload")
+        upload_btn = gr.Button("Upload")
 
-    dataframe = gr.Dataframe(
-        headers=["text", "label"],
-        label="📝 Annotate Labels Below",
+    df_table = gr.Dataframe(
+        headers=["text","label"],
         interactive=True,
         visible=False,
-        row_count=10,
-        col_count=(2, "fixed")
+        label="📝 Editable Table"
     )
-
     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="📥 Download", interactive=False)
+        save_btn    = gr.Button("💾 Save")
+        download_btn= gr.Button("⬇️ Download CSV")
+        download_out= gr.File(label="Download")
 
-    with gr.Row():
-        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")
+    with gr.Accordion("📦 Push to Hugging Face Hub", open=False):
+        repo_input  = gr.Textbox(label="Repo (username/dataset-name)")
+        token_input = gr.Textbox(label="HF Token", type="password")
+        push_btn    = gr.Button("🚀 Push")
         push_status = gr.Textbox(label="Push Status", interactive=False)
 
-    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)
+    # ------------------------
+    upload_btn.click(upload_csv,     inputs=file_input,        outputs=[df_table, status])
+    save_btn.click(  save_changes,   inputs=df_table,          outputs=status)
+    download_btn.click(download_csv, outputs=download_out)
+    push_btn.click(  push_to_hub,    inputs=[repo_input,token_input], outputs=push_status)
+    # ------------------------
 
-app.launch()
+    # Launch the app
+    app.launch()