Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 import pandas as pd
 import io
-from huggingface_hub import HfApi, HfFolder, Repository
 import os
+from huggingface_hub import HfApi, Repository
 
 # Global state
 df = pd.DataFrame()
@@ -10,35 +10,25 @@ 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:
-        return gr.update(visible=False), "CSV must contain 'text' and 'label' columns."
-    
-    # Fill label column if empty
+        return None, "❌ CSV must have 'text' and 'label' columns."
     df["label"] = df["label"].fillna("")
-    
-    # Return the editable table
-    return gr.Dataframe(
-        value=df,
-        headers=["text", "label"],
-        interactive=True,
-        label="Edit labels below"
-    ), "File uploaded successfully."
+    return df[["text", "label"]], "✅ File uploaded. You can now edit labels."
 
-def save_edits(updated_table):
+def save_changes(updated_table):
     global df
     df = pd.DataFrame(updated_table, columns=["text", "label"])
-    return "Changes saved."
+    return "✅ Changes saved."
 
 def download_csv():
-    # Create a downloadable CSV
-    csv_bytes = df.to_csv(index=False).encode()
+    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")
 
 def push_to_hub(repo_name, hf_token):
-    # Authenticate and push to Hugging Face Hub
+    global df
     repo_url = f"https://huggingface.co/datasets/{repo_name}"
-    local_path = f"./{repo_name}"
+    local_path = f"./{repo_name.replace('/', '_')}"
     
     if os.path.exists(local_path):
         os.system(f"rm -rf {local_path}")
@@ -46,35 +36,42 @@ def push_to_hub(repo_name, hf_token):
     api = HfApi()
     api.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"Pushed to Hugging Face: {repo_url}"
+    return f"✅ Successfully pushed to: {repo_url}"
+
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) 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.")
 
-with gr.Blocks() as demo:
-    gr.Markdown("## 🏷️ Label it! Text Labeling Tool")
-    
     with gr.Row():
-        csv_input = gr.File(label="Upload CSV", file_types=[".csv"])
+        file_input = gr.File(label="📁 Upload CSV", file_types=[".csv"])
         upload_btn = gr.Button("Upload")
     
-    df_output = gr.Dataframe(headers=["text", "label"], interactive=True, visible=False)
-    upload_status = gr.Textbox(visible=True, interactive=False)
+    dataframe = gr.Dataframe(
+        headers=["text", "label"],
+        label="📝 Annotate Labels Below",
+        interactive=True,
+        visible=False,
+        row_count=10,
+        col_count=(2, "fixed")
+    )
     
+    status = gr.Textbox(visible=True, label="Status", interactive=False)
+
     with gr.Row():
-        save_btn = gr.Button("Save Changes")
-        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_file = gr.File(label="📥 Downloaded File")
+
     with gr.Row():
-        hf_repo = gr.Textbox(label="HF Dataset Repo (e.g. your-username/my-dataset)")
-        hf_token = gr.Textbox(label="Hugging Face Token", type="password")
-        push_btn = gr.Button("Push to Hugging Face Hub")
-        push_status = gr.Textbox(interactive=False)
-    
-    upload_btn.click(fn=upload_csv, inputs=csv_input, outputs=[df_output, upload_status])
-    save_btn.click(fn=save_edits, inputs=df_output, outputs=upload_status)
-    download_btn.click(fn=download_csv, outputs=download_file)
-    push_btn.click(fn=push_to_hub, inputs=[hf_repo, hf_token], outputs=push_status)
+        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")
+        push_status = gr.Textbox(label="Push Status", interactive=False)
 
-demo.launch()
+    upload_btn.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)_