Spaces:
Running
Running
| import os | |
| import json | |
| import requests | |
| import time | |
| from github import Github, GithubIntegration | |
| import gradio as gr | |
| app_id = '307916' | |
| owner='duerrsimon' | |
| repo_name='bioicons' | |
| app_key = os.environ.get('GITHUB_TOKEN') | |
| licenses = ['mit', 'cc-by-3.0', 'cc-0', 'cc-by-4.0', 'bsd', 'cc-by-sa-3.0', 'cc-by-sa-4.0'] | |
| categories = ["Machine_Learning", "Genetics", "Chemistry", "Scientific_graphs", "Cell_membrane", "Microbiology", "Oncology", "Molecular_modelling", "Intracellular_components", "Cell_types", "Cell_lines", "Plants_Algae", "Chemo-_and_Bioinformatics", "Cell_culture", "Tissues", "Computer_hardware", "Nucleic_acids", "General_items", "Safety_symbols", "Receptors_channels", "Human_physiology", "Animals", "Lab_apparatus", "Blood_Immunology", "Viruses", "Parasites"] | |
| def create_branch(site, branch_name): | |
| return site.create_git_ref( | |
| 'refs/heads/{branch_name}'.format(**locals()), | |
| site.get_branch('main').commit.sha | |
| ) | |
| def create_pull_request(site,branch_name, github_user): | |
| if github_user!="": | |
| message = f"submitted by: @{github_user}." | |
| else: | |
| message ="No github account provided." | |
| return site.create_pull( | |
| title="New icons submitted via web", | |
| body= message, | |
| base="main", | |
| head=branch_name | |
| ) | |
| def remove_special_chars(author): | |
| author = author.replace(" ", "-") | |
| author = author.replace(".", "-") | |
| author = author.replace("(", "-") | |
| author = author.replace(")", "-") | |
| author = author.replace(":", "-") | |
| author = author.replace(";", "-") | |
| author = author.replace(",", "-") | |
| author = author.replace("!", "-") | |
| author = author.replace("?", "-") | |
| author = author.replace("=", "-") | |
| author = author.replace("+", "-") | |
| author = author.replace("/", "-") | |
| return author | |
| def process_and_create_pull(icon_files,category, new_cateogry_check, new_category, license, creator_name, creator_url, copyright, github_user): | |
| git_integration = GithubIntegration( | |
| app_id, | |
| app_key, | |
| ) | |
| api = Github( | |
| login_or_token=git_integration.get_access_token( | |
| git_integration.get_repo_installation(owner, repo_name).id | |
| ).token | |
| ) | |
| site = api.get_repo('duerrsimon/bioicons') | |
| if copyright: | |
| if new_cateogry_check: | |
| category = remove_special_chars(new_category) | |
| hash = '{0:010x}'.format(int(time.time() * 256))[:8] | |
| branch_name = f"new-icons-{hash}" | |
| create_branch(site,branch_name) | |
| authors = requests.get("https://bioicons.com/icons/authors.json").json() | |
| #remove whitespace and other special characters | |
| author = remove_special_chars(creator_name) | |
| if author not in authors.keys(): | |
| index_file = site.get_contents('static/icons/authors.json') | |
| authors[author]=creator_url | |
| updated_content = json.dumps(authors, indent=4) | |
| site.update_file( | |
| path='static/icons/authors.json', | |
| message='Insert author', | |
| content=updated_content, | |
| sha=index_file.sha, | |
| branch=branch_name | |
| ) | |
| #all_files = [] | |
| #contents = site.get_contents("static/icons/{license}/{category}/{author}/") | |
| #print(contents) | |
| #while contents: | |
| # file_content = contents.pop(0) | |
| # if file_content.type == "dir": | |
| # contents.extend(site.get_contents(file_content.path)) | |
| # else: | |
| # file = file_content | |
| # all_files.append(str(file).replace('ContentFile(path="','').replace('")','')) | |
| #print("files in directory:", all_files) | |
| for file in icon_files: | |
| #get file content from filesystem | |
| with open(file.name, 'r') as fp: | |
| file_content = fp.read() | |
| filename = os.path.basename(file.name).replace(" ", "_") | |
| git_file = f'static/icons/{license}/{category}/{author}/{filename}' | |
| try: | |
| site.create_file( | |
| path=git_file, | |
| message=f'create {filename}', | |
| content=file_content, | |
| branch=branch_name | |
| ) | |
| except Exception as e: | |
| print(e, git_file, "already may exist, trying updating") | |
| contents = site.get_contents(git_file) | |
| site.update_file( | |
| path=git_file, | |
| message=f'update {filename}', | |
| content=file_content, | |
| sha=contents.sha, | |
| branch=branch_name | |
| ) | |
| pull = create_pull_request(site,branch_name, github_user) | |
| return f"""<link href="https://cdnjs.cloudflare.com/ajax/libs/flowbite/1.6.4/flowbite.min.css" rel="stylesheet" /> | |
| <div class="flex p-4 mb-4 text-sm text-green-800 rounded-lg bg-green-200 " role="alert"> | |
| <span class="sr-only">Success</span> | |
| <div> | |
| <span class="font-medium">Thank you!</span> Your pull request for the submitted illustrations has been submitted on GitHub. You can track your submission on Github at <br> <a class="text-green-400 text-lg hover:underline" target="_blank" href="{pull.html_url}">{pull.html_url}</a>. | |
| </div> | |
| </div> | |
| """ | |
| else: | |
| return """<link href="https://cdnjs.cloudflare.com/ajax/libs/flowbite/1.6.4/flowbite.min.css" rel="stylesheet" /> | |
| <div class="flex p-4 mb-4 text-sm text-red-800 rounded-lg bg-red-200 dark:bg-gray-800 dark:text-red-400" role="alert"> | |
| <svg aria-hidden="true" class="flex-shrink-0 inline w-5 h-5 mr-3" fill="currentColor" viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg"> | |
| <path stroke-linecap="round" stroke-linejoin="round" d="M12 9v3.75m-9.303 3.376c-.866 1.5.217 3.374 1.948 3.374h14.71c1.73 0 2.813-1.874 1.948-3.374L13.949 3.378c-.866-1.5-3.032-1.5-3.898 0L2.697 16.126zM12 15.75h.007v.008H12v-.008z" /> | |
| </svg> | |
| <span class="sr-only">Info</span> | |
| <div> | |
| <span class="font-medium">Copyright error.</span> You need to certify that the illustrations have a suitable license and that copyright allows to share it on Bioicons. | |
| </div> | |
| </div> | |
| """ | |
| def shownewcategory(new_category, check): | |
| if check: | |
| return gr.Textbox(visible=True) | |
| return gr.Textbox(visible=False) | |
| with gr.Blocks() as demo: | |
| with gr.Row(): | |
| creator_name = gr.Textbox(label="Creator") | |
| creator_url = gr.Textbox(label="Creator URL", placeholder="Link to personal website,Twitter,Mastodon, GitHub") | |
| github_user = gr.Textbox(label="Your GitHub Username", placeholder="GitHub username is used to communicate with you") | |
| icon_files = gr.File(label="Icon files", file_types=["svg"], file_count="multiple") | |
| with gr.Row(): | |
| category = gr.Dropdown(label="Category", choices=categories) | |
| new_category_check = gr.Checkbox(info="New category?", label="Desired category not listed?") | |
| new_category = gr.Textbox(label="New category", visible=False) | |
| new_category_check.change(fn=shownewcategory, inputs=[new_category,new_category_check], outputs=new_category) | |
| license = gr.Dropdown(label="License", choices=licenses) | |
| copyright = gr.Checkbox(info="Copyright", label="I certify that I own the copyright of the submitted illustrations or have appropriately credited the original creator as required by the license of the illustration.") | |
| btn = gr.Button(value="Submit illustrations") | |
| output = gr.HTML(label="Output", value="") | |
| btn.click(process_and_create_pull, inputs=[icon_files,category, new_category_check, new_category, license, creator_name, creator_url, copyright, github_user], outputs=output, api_name="bioicons-submit") | |
| demo.launch() | |