Update app.py

app.py

@@ -2,7 +2,10 @@ import gradio as gr
 import duckdb
 import urllib.parse
 
-PARQUET_PATH = "/data/v3_repos.parquet"
+PARQUET_PATH = "data/v3_repos.parquet"
+OPT_OUT_REPO = "bigcode-project/opt-out-v2"
+NEW_ISSUE_URL = f"https://github.com/{OPT_OUT_REPO}/issues/new"
+TEMPLATE = "opt-out-request.yml"
 
 con = duckdb.connect()
 
@@ -19,39 +22,37 @@ We want to give developers agency over their source code \
 by letting them decide whether or not it should be used to develop and evaluate \
 machine learning models.
 
-Enter your GitHub username below to check if any of your repositories are in The Stack v3. \
-If your code is found, you will get a pre-filled opt-out link \
-that submits a properly formatted removal request on your behalf. \
-**Please use this tool to submit opt-out requests** rather than opening issues manually — \
-it ensures your request can be processed quickly and correctly.
+Enter your GitHub username (or an organization name) below to check if any of its \
+repositories are in The Stack v3. If your code is found, you'll get a pre-filled \
+opt-out link that submits a properly formatted removal request on your behalf. \
+**Please use this tool to submit opt-out requests** rather than opening issues \
+manually — it ensures your request can be processed quickly and correctly.
 """
 
 opt_out_text_template = """\
 ### Opt-out
 
-If you want your data to be removed from The Stack and model training, \
-open an issue with <a href="https://github.com/bigcode-project/opt-out-v2/issues/new?title={title}&body={body}" target="_blank">this link</a> \
-(if the link doesn't work, try right-clicking and opening it in a new tab) \
-or visit [https://github.com/bigcode-project/opt-out-v2/issues/new?&template=opt-out-request.md](https://github.com/bigcode-project/opt-out-v2/issues/new?&template=opt-out-request.md).\
+If you want your data removed from The Stack and model training, \
+<a href="{url}" target="_blank">open a pre-filled opt-out request</a> \
+(if the link doesn't work, try right-clicking and opening it in a new tab). \
+The form lets you remove your whole account or just specific repositories, and \
+add any other accounts or organizations you own — it walks you through it.\
 """
 
-opt_out_issue_title = "Opt-out request for {username}"
-opt_out_issue_body = """\
-I request that the following data is removed from The Stack:
 
-{repo_list}
+def issue_url(username: str) -> str:
+    """Link to the opt-out issue form, pre-seeded with the looked-up account.
 
-_Note_: If you don't want all repositories removed, edit the list above. \
-To exclude everything under your username, replace the list with a single "all" entry.
-"""
-
-
-def issue_url(username, repos):
-    title = urllib.parse.quote(opt_out_issue_title.format(username=username))
-    body = urllib.parse.quote(
-        opt_out_issue_body.format(repo_list=" - " + "\n - ".join(repos))
-    )
-    return opt_out_text_template.format(title=title, body=body)
+    Seeds the "remove entirely" list with the username; the form itself guides
+    the user through narrowing to specific repos or adding more accounts/orgs.
+    """
+    params = {
+        "template": TEMPLATE,
+        "title": "Opt-out request",
+        "remove_accounts": username,
+    }
+    query = urllib.parse.urlencode(params, quote_via=urllib.parse.quote)
+    return opt_out_text_template.format(url=f"{NEW_ISSUE_URL}?{query}")
 
 
 def check_username(username):
@@ -76,8 +77,7 @@ def check_username(username):
         lines.append(f"[{full}](https://github.com/{full})\n")
     output_md = "\n".join(lines).strip()
 
-    full_names = [f"{username}/{r}" for r in repos]
-    return output_md, issue_url(username, full_names)
+    return output_md, issue_url(username)
 
 
 with gr.Blocks() as demo:
@@ -86,7 +86,7 @@ with gr.Blocks() as demo:
         with col:
             gr.HTML(huggy_html)
             gr.Markdown(text)
-            username = gr.Text("", label="Your GitHub username:")
+            username = gr.Text("", label="Your GitHub username or organization:")
             check_button = gr.Button("Check!")
             repos_md = gr.Markdown()
             opt_out_md = gr.Markdown()