Spaces:
Runtime error
Runtime error
import os | |
import sys | |
import uuid | |
import gradio as gr | |
import requests | |
import json | |
from zipfile import ZipFile | |
import shutil | |
from pathlib import Path | |
from huggingface_hub import (create_repo,get_full_repo_name,upload_file,CommitOperationAdd,HfApi,snapshot_download) | |
from PIL import Image | |
uid = uuid.uuid4() | |
#token = os.environ['HF_TOKEN'] | |
#token_self = os.environ['HF_TOKEN'] | |
#o=os.environ['P'] | |
def show_s(name,token=""): | |
spaces=[] | |
spaces.append("") | |
api = HfApi(token=token) | |
author=name | |
s_ist = (api.list_spaces(author=author)) | |
for i,space in enumerate(s_ist): | |
space_ea = space.id.split("/",1)[1] | |
#s_info=api.space_info(f'{name}/{space}',files_metadata=True) | |
#print(s_info) | |
spaces.append(space_ea) | |
#print (space_ea) | |
return(gr.update(label="Spaces", choices=[s for s in spaces], interactive=True)) | |
def show_f(repo,name,token=""): | |
api = HfApi(token=token) | |
f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="space")) | |
print (f_ist) | |
file_list = [] | |
#file_out = [] | |
if not os.path.exists(name): | |
os.makedirs(name) | |
for d_app in f_ist: | |
if "/" in d_app: | |
dir_1=d_app.split("/",1)[0] | |
rem_1=d_app.split("/",1)[1] | |
if not os.path.exists(f'{name}/{dir_1}'): | |
os.makedirs(f'{name}/{dir_1}') | |
if "/" in rem_1: | |
dir_2=rem_1.split("/",1)[0] | |
rem_2=rem_1.split("/",1)[1] | |
if not os.path.exists(f'{name}/{dir_1}/{dir_2}'): | |
os.makedirs(f'{name}/{dir_1}/{dir_2}') | |
sf=rem_2.split(".",1)[1] | |
pf=rem_2.split(".",1)[0] | |
f_name=f'{dir_1}/{dir2}/{pf}.{sf}' | |
else: | |
sf=rem_1.split(".",1)[1] | |
pf=rem_1.split(".",1)[0] | |
f_name=f'{dir_1}/{pf}.{sf}' | |
print(f_name) | |
else: | |
sf=d_app.split(".",1)[1] | |
pf=d_app.split(".",1)[0] | |
f_name=f'{pf}.{sf}' | |
pass | |
r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}') | |
print(d_app) | |
#print (r.text) | |
uid = uuid.uuid4() | |
file = open(f'{name}/{f_name}','wb') | |
file.write(r.content) | |
file.close() | |
file_list.append(f'{name}/{f_name}') | |
with ZipFile(f"{name}.zip", "w") as zipObj: | |
for idx, file in enumerate(f_ist): | |
zipObj.write(f'{name}/{file}') | |
file_list.append(f'{name}.zip') | |
s_info=api.space_info(f'{repo}/{name}',files_metadata=True) | |
print(f's_info:: {s_info}') | |
''' | |
with open(f'{uid}-tmp.json','w') as f: | |
json.dump(s_info,f,indent=4) | |
f.close() | |
''' | |
info_json = api.get_space_runtime(f'{repo}/{name}') | |
print(info_json.raw) | |
return(gr.update(label="Files", choices=[f for f in f_ist],interactive=True), file_list,file_list, s_info) | |
def show_f2(repo,name,token=""): | |
api = HfApi(token=token) | |
#f_ist = snapshot_download(repo_id=f'{repo}/{name}', repo_type="space") | |
f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="space")) | |
print (f_ist) | |
file_list = [] | |
file_out = [] | |
if not os.path.exists(name): | |
os.makedirs(name) | |
for d_app in f_ist: | |
r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{d_app}') | |
#print (r.text) | |
uid = uuid.uuid4() | |
sf=d_app.split(".",1)[1] | |
pf=d_app.split(".",1)[0] | |
f_name=f'{pf}.{sf}' | |
file = open(f'{name}/{f_name}','w') | |
file.writelines(r.text) | |
file.close() | |
file_list.append(Path(f'{name}/{f_name}')) | |
file_out.append(d_app) | |
with ZipFile(f"{name}.zip", "w") as zipObj: | |
for idx, file in enumerate(f_ist): | |
zipObj.write(f'{name}/{file}') | |
file_list.append(f'{name}.zip') | |
return(gr.Dropdown.update(label="Files", choices=[f for f in f_ist],interactive=True), file_list) | |
def show_file_content(repo,name,file,token=""): | |
print(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}') | |
r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}') | |
text=r.text | |
html_text += f'<pre style="text-wrap:pretty;">{text}</pre>\n' | |
out_text = r.text | |
return(out_text) | |
def show_f_cont(repo,name,file,token=""): | |
html_text = '<html>\n<body>\n<div>\n' | |
images=[".png" , ".jpg" , ".gif" , ".webm" , ".mp4"] | |
is_im=False | |
for x in images: | |
if x in file: | |
html_text += f'<object data="https://huggingface.co/spaces/{repo}/{name}/resolve/main/{file}" style="width:100%;font-size:small;"></object>' | |
out_text = "Image File" | |
is_im=True | |
else: | |
pass | |
if is_im==False: | |
print(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}') | |
r = requests.get(f'https://huggingface.co/spaces/{repo}/{name}/raw/main/{file}') | |
text=r.text | |
html_text += f'<pre style="text-wrap:pretty;">{text}</pre>\n' | |
out_text = r.text | |
else: | |
pass | |
html_text += '</div>\n</body>\n</html>' | |
return(html_text) | |
def show_f_frame(repo,name,file,token=""): | |
file_url=f'https://huggingface.co/spaces/{repo}/{name}/blob/main/{file}' | |
out=f"""<div><iframe src='{file_url}' height='1200px' width='100%'></iframe>""" | |
return out | |
def show_f_frame2(repo,name,file,token=""): | |
file_url=f'https://huggingface.co/spaces/{repo}/{name}' | |
out=f"""<div><iframe src='{file_url}' height='1200px' width='100%'></iframe>""" | |
return out | |
def show_all(author,token=""): | |
spaces=[] | |
api = HfApi(token=token) | |
#author=name | |
s_ist = (api.list_spaces(author=author)) | |
file_list = [] | |
file_list_ea=[] | |
for i,space in enumerate(s_ist): | |
space_ea = space.id.split("/",1)[1] | |
spaces.append(space_ea) | |
#print (space_ea) | |
f_ist = (api.list_repo_files(repo_id=f'{author}/{space_ea}', repo_type="space")) | |
#print (f_ist) | |
if not os.path.exists(space_ea): | |
os.makedirs(space_ea) | |
for d_app in f_ist: | |
r = requests.get(f'https://huggingface.co/spaces/{author}/{space_ea}/raw/main/{d_app}') | |
#print (r.text) | |
uid = uuid.uuid4() | |
try: | |
sf=d_app.split(".",1)[1] | |
pf=d_app.split(".",1)[0] | |
f_name=f'{pf}.{sf}' | |
file = open(f'{space_ea}/{f_name}','w') | |
file.writelines(r.text) | |
file.close() | |
#file_list_ea.append(Path(f'{space}/{f_name}')) | |
except Exception: | |
pass | |
with ZipFile(f"{space_ea}.zip", "w") as zipObj: | |
for idx, file in enumerate(f_ist): | |
try: | |
zipObj.write(f'{space_ea}/{file}') | |
except Exception: | |
pass | |
file_list.append(f'{space_ea}.zip') | |
yield file_list | |
with ZipFile(f"{author}.zip", "w") as zipObj: | |
for idx, file in enumerate(file_list): | |
try: | |
zipObj.write(f'{file}') | |
except Exception: | |
pass | |
file_list.append(f'{author}.zip') | |
yield file_list | |
def show_all_z(author,token=""): | |
spaces=[] | |
api = HfApi(token=token) | |
#author=name | |
s_ist = (api.list_spaces(author=author)) | |
file_list = [] | |
file_list_ea=[] | |
for i,space in enumerate(s_ist): | |
space_ea = space.id.split("/",1)[1] | |
spaces.append(space_ea) | |
#print (space_ea) | |
f_ist = (api.list_repo_files(repo_id=f'{author}/{space_ea}', repo_type="space")) | |
#print (f_ist) | |
if not os.path.exists(space_ea): | |
os.makedirs(space_ea) | |
file= snapshot_download(repo_id=f'{author}/{space_ea}', repo_type="space") | |
shutil.make_archive(f"{space_ea}", 'zip', file) | |
file_list.append(f'{space_ea}.zip') | |
yield file_list | |
with ZipFile(f"{author}.zip", "w") as zipObj: | |
for idx, file in enumerate(file_list): | |
try: | |
zipObj.write(f'{file}') | |
except Exception: | |
pass | |
file_list.append(f'{author}.zip') | |
yield file_list | |
def dl_checked_fn(author,checklist,token=""): | |
spaces=[] | |
api = HfApi(token=token) | |
#author=name | |
s_ist = checklist | |
file_list = [] | |
file_list_ea=[] | |
for i,space_ea in enumerate(s_ist): | |
#spaces.append(space_ea) | |
#print (space_ea) | |
f_ist = (api.list_repo_files(repo_id=f'{author}/{space_ea}', repo_type="space")) | |
#print (f_ist) | |
if not os.path.exists(space_ea): | |
os.makedirs(space_ea) | |
file= snapshot_download(repo_id=f'{author}/{space_ea}', repo_type="space") | |
shutil.make_archive(f"{space_ea}", 'zip', file) | |
file_list.append(f'{space_ea}.zip') | |
yield file_list | |
with ZipFile(f"{author}.zip", "w") as zipObj: | |
for idx, file in enumerate(file_list): | |
try: | |
zipObj.write(f'{file}') | |
except Exception: | |
pass | |
file_list.append(f'{author}.zip') | |
yield file_list | |
def get_space_runtime(author,checklist,token=""): | |
api=HfApi(token=token) | |
space_info=[] | |
for space in checklist: | |
outp=api.get_space_runtime(f'{author}/{space}') | |
space_info.append(outp['stage']) | |
return space_info | |
def delete_checked(confirm_val,author,checklist,token=""): | |
if confirm_val=="CONFIRM": | |
api = HfApi(token=token) | |
s_ist = checklist | |
delete_list = [] | |
for i,space_ea in enumerate(s_ist): | |
try: | |
api.delete_repo(repo_id=f'{author}/{space_ea}',repo_type='space') | |
delete_list.append(f'Deleted:: {space_ea}\n') | |
yield delete_list,gr.update(visible=False) | |
except Exception as e: | |
yield e | |
else: | |
yield "Not Deleting", gr.update(visible=False) | |
def checkp(p): | |
if p == o: | |
return gr.update(visible=False), gr.update(visible=True) | |
elif p != o: | |
return None, None | |
def update_checked_message(inp): | |
html="" | |
for ea in inp: | |
html=f'{html}<br>{ea}' | |
return html | |
def ru_sure_fn(): | |
return gr.update(visible=True) |