Update app.py

app.py

@@ -2,10 +2,14 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 import json
 import gzip
-
+import urllib
 
 usernames = {}
 
+filepath = hf_hub_download(repo_id="bigcode/the-stack-username-to-repo", filename="username_to_repo.json.gz", repo_type="dataset", revision="v1.2")
+with gzip.open(filepath, 'r') as f:
+    usernames["v1.2"] = json.loads(f.read().decode('utf-8'))
+
 filepath = hf_hub_download(repo_id="bigcode/the-stack-username-to-repo", filename="username_to_repo.json.gz", repo_type="dataset", revision="v1.1")
 with gzip.open(filepath, 'r') as f:
     usernames["v1.1"] = json.loads(f.read().decode('utf-8'))
@@ -15,17 +19,44 @@ with gzip.open(filepath, 'r') as f:
     usernames["v1.0"] = json.loads(f.read().decode('utf-8'))
 
 text = """\
-![](https://huggingface.co/spaces/bigcode/in-the-stack/resolve/main/banner.png)
+![](https://huggingface.co/spaces/lvwerra/in-the-stack-gr/resolve/main/banner.png)
 **_The Stack is an open governance interface between the AI community and the open source community._**
 
 # Am I in The Stack?
 
-As part of the BigCode project, we released and maintain [The Stack](https://huggingface.co/datasets/bigcode/the-stack), a 3.1 TB dataset of permissively licensed source code in 30 programming languages. One of our goals in this project is to give people agency over their source code by letting them decide whether or not it should be used to develop and evaluate machine learning models, as we acknowledge that not all developers may wish to have their data used for that purpose.
+As part of the BigCode project, we released and maintain [The Stack](https://huggingface.co/datasets/bigcode/the-stack), a 6 TB dataset of permissively licensed source code over 300 programming languages. One of our goals in this project is to give people agency over their source code by letting them decide whether or not it should be used to develop and evaluate machine learning models, as we acknowledge that not all developers may wish to have their data used for that purpose.
 """ + """\
 
 This tool lets you check if a repository under a given username is part of The Stack dataset. Would you like to have your data removed from future versions of The Stack? You can opt-out following the instructions [here](https://www.bigcode-project.org/docs/about/the-stack/#how-can-i-request-that-my-data-be-removed-from-the-stack).
 """
 
+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 a right click and open 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) .\
+"""
+
+opt_out_issue_title = """Opt-out request for {username}"""
+opt_out_issue_body = """\
+I request that the following data is removed from The Stack:
+
+ - Commits
+ - GitHub issue
+{repo_list}
+
+_Note_: If you don't want all resources to be included just remove the elements from the list above. If you would like to exclude all repositories and resources just add a single element "all" to the list.
+"""
+
+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)))
+    
+    opt_out_text = opt_out_text_template.format(title=title, body=body)
+    
+    return opt_out_text
+
 def check_username(username, version):
     output_md = ""
     if username in usernames[version] and len(usernames[version][username])>0:
@@ -34,22 +65,26 @@ def check_username(username, version):
         output_md += f"**Yes**, there is code from **{len(repos)} {repo_word}** in The Stack:\n\n"
         for repo in repos:
             output_md += f"_{repo}_\n\n"
+
+        return output_md.strip(), issue_url(username, repos)
     else:
         output_md += "**No**, your code is not in The Stack."
-    return output_md.strip()
+        return output_md.strip(), ""
 
 with gr.Blocks() as demo:
     with gr.Row():
         _, colum_2, _ = gr.Column(scale=1), gr.Column(scale=6), gr.Column(scale=1)
         with colum_2:
             gr.Markdown(text)
-            version = gr.Dropdown(["v1.1", "v1.0"], label="The Stack version:", value="v1.1")
+            version = gr.Dropdown(["v1.2", "v1.1", "v1.0"], label="The Stack version:", value="v1.2")
             username = gr.Text("", label="Your GitHub username:")
             check_button = gr.Button("Check!")
             
             repos = gr.Markdown()
+            opt_out = gr.Markdown()
+
             
-            check_button.click(check_username, [username, version], repos)
+            check_button.click(check_username, [username, version], [repos, opt_out])
 
 
 demo.launch()