Update app.py

app.py

@@ -3,10 +3,16 @@ from huggingface_hub import hf_hub_download
 import json
 import gzip
 
+
+usernames = {}
+
 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'))
 
+filepath = hf_hub_download(repo_id="bigcode/the-stack-username-to-repo", filename="username_to_repo.json.gz", repo_type="dataset")
 with gzip.open(filepath, 'r') as f:
-    usernames = json.loads(f.read().decode('utf-8'))
+    usernames["v1.0"] = json.loads(f.read().decode('utf-8'))
 
 text = """\
 **_The Stack is an open governance interface between the AI community and the open source community._**
@@ -19,10 +25,10 @@ As part of the BigCode project, we released and maintain [The Stack](https://hug
 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).
 """
 
-def check_username(username):
+def check_username(username, version):
     output_md = ""
-    if username in usernames and len(usernames[username])>0:
-        repos = usernames[username]
+    if username in usernames[version] and len(usernames[version][username])>0:
+        repos = usernames[version][username]
         repo_word = "repository" if len(repos)==1 else "repositories"
         output_md += f"**Yes**, there is code from **{len(repos)} {repo_word}** in The Stack:\n\n"
         for repo in repos:
@@ -33,12 +39,13 @@ def check_username(username):
 
 with gr.Blocks() as demo:
     gr.Markdown(text)
+    version = gr.Dropdown(["v1.1", "v1.0"], label="The Stack version:", value="v1.1")
     username = gr.Text("", label="Your GitHub username:")
     check_button = gr.Button("Check!")
     
     repos = gr.Markdown()
     
-    check_button.click(check_username, [username], repos)
+    check_button.click(check_username, [username, version], repos)
 
 
 demo.launch()