add login requirement

app.py

@@ -1,7 +1,7 @@
 import csv
 import os
 from datetime import datetime
-from typing import Optional
+from typing import Optional, Union
 
 import gradio as gr
 from huggingface_hub import HfApi, Repository
@@ -24,15 +24,17 @@ repo: Optional[Repository] = None
 #     repo = Repository(local_dir=DATADIR, clone_from=DATASET_REPO_URL, token=HF_TOKEN)
 
 
-def neuron_export(model_id: str, task: str) -> str:
+def neuron_export(model_id: str, task: str, oauth_token: gr.OAuthToken) -> str:
+    if oauth_token.token is None:
+        return "You must be logged in to use this space"
+
     if not model_id:
         return f"### Invalid input 🐞 Please specify a model name, got {model_id}"
 
     try:
-        api = HfApi(token=HF_TOKEN)  # Use HF_TOKEN if available, else anonymous
-        token = HF_TOKEN  # Pass token to convert only if available
+        api = HfApi(token=oauth_token.token)
 
-        error, commit_info = convert(api=api, model_id=model_id, task=task, token=token)
+        error, commit_info = convert(api=api, model_id=model_id, task=task, token=oauth_token.token)
         if error != "0":
             return error
 
@@ -56,7 +58,7 @@ def neuron_export(model_id: str, task: str) -> str:
             print("[dataset]", commit_url)
 
         pr_revision = commit_info.pr_revision.replace("/", "%2F")
-        return f"#### Success 🔥 This model was successfully exported and a PR was opened: [{commit_info.pr_url}]({commit_info.pr_url}). To use the model before the PR is approved, go to https://huggingface.co/{model_id}/tree/{pr_revision}"
+        return f"#### Success 🔥 Yay! This model was successfully exported and a PR was opened using your token: [{commit_info.pr_url}]({commit_info.pr_url}). If you would like to use the exported model without waiting for the PR to be approved, head to https://huggingface.co/{model_id}/tree/{pr_revision}"
 
     except Exception as e:
         return f"#### Error: {e}"
@@ -77,15 +79,16 @@ TITLE = """
 """
 
 DESCRIPTION = """
-Export 🤗 Transformers models hosted on the Hugging Face Hub to AWS Neuron-optimized format for Inferentia/Trainium acceleration.
+This Space allows you to automatically export 🤗 transformers models hosted on the Hugging Face Hub to AWS Neuron-optimized format for Inferentia/Trainium acceleration. It opens a PR on the target model, and it is up to the owner of the original model to merge the PR to allow people to leverage Neuron optimization!
 
 **Features:**
 - Automatically opens PR with Neuron-optimized model
 - Preserves original model weights
 - Adds proper tags to model card
 
-**Note:**
-- PR creation requires the Space owner to have a valid write token set via HF_WRITE_TOKEN
+**Requirements:**
+- Model must be compatible with [Optimum Neuron](https://huggingface.co/docs/optimum-neuron)
+- User must be logged in with write token
 """
 
 # Custom CSS to fix dark mode compatibility and transparency issues
@@ -145,6 +148,10 @@ CUSTOM_CSS = """
 """
 
 with gr.Blocks(css=CUSTOM_CSS) as demo:
+    # Login requirement notice and button
+    gr.Markdown("**You must be logged in to use this space**")
+    gr.LoginButton(min_width=250)
+    
     # Centered title and image
     gr.HTML(TITLE_IMAGE)
     gr.HTML(TITLE)