File size: 6,458 Bytes
5f685fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
"""
import re
import gradio as gr
from gui.asset_components import (background_music_checkbox,
background_video_checkbox,
getBackgroundMusicChoices,
getBackgroundVideoChoices)
from gui.ui_abstract_component import AbstractComponentUI
from shortGPT.config.asset_db import AssetDatabase
class AssetLibrary(AbstractComponentUI):
def __init__(self):
pass
def create_ui(self):
'''Create the asset library UI'''
with gr.Tab("Asset library") as asset_library_ui:
with gr.Column():
with gr.Accordion("Add your own video / audio / image", open=False) as accordion:
with gr.Column(visible=True):
asset_name = gr.Textbox(label="Name (required)")
asset_type = gr.Radio(["background video", "background music"], value="background video", label="Type")
youtube_url = gr.Textbox(label="URL (https://youtube.com/xyz)")
add_youtube_link = gr.Button("ADD")
with gr.Row():
with gr.Column(scale=3):
asset_dataframe_ui = gr.Dataframe(self.__fulfill_df, interactive=False)
with gr.Column(scale=2):
gr.Markdown("Preview")
asset_preview_ui = gr.HTML(self.__get_first_preview)
delete_button = gr.Button("🗑️ Delete", scale=0, variant="primary")
delete_button.click(self.__delete_clicked, [delete_button], [asset_dataframe_ui, asset_preview_ui, delete_button, background_video_checkbox, background_music_checkbox])
asset_dataframe_ui.select(self.__preview_asset, [asset_dataframe_ui], [asset_preview_ui, delete_button])
add_youtube_link.click(
self.__verify_youtube_asset_inputs, [asset_name, youtube_url, asset_type], []).success(self.__add_youtube_asset, [asset_name, youtube_url, asset_type], [asset_dataframe_ui, asset_preview_ui, delete_button, accordion, background_video_checkbox, background_music_checkbox])
return asset_library_ui
def __fulfill_df(self):
'''Get the asset dataframe'''
return AssetDatabase.get_df()
def __verify_youtube_asset_inputs(self, asset_name, yt_url, type):
if not asset_name or not re.match("^[A-Za-z0-9 _-]*$", asset_name):
raise gr.Error('Invalid asset name. Please provide a valid name that you will recognize (Only use letters and numbers)')
if not yt_url.startswith("https://youtube.com/") and not yt_url.startswith("https://www.youtube.com/"):
raise gr.Error('Invalid YouTube URL. Please provide a valid URL.')
if AssetDatabase.asset_exists(asset_name):
raise gr.Error('An asset already exists with this name, please choose a different name.')
def __add_youtube_asset(self, asset_name, yt_url, type):
'''Add a youtube asset'''
AssetDatabase.add_remote_asset(asset_name, type, yt_url)
latest_df = AssetDatabase.get_df()
return gr.DataFrame.update(value=latest_df), gr.HTML.update(value=self.__get_asset_embed(latest_df, 0)),\
gr.update(value=f"🗑️ Delete {latest_df.iloc[0]['name']}"),\
gr.Accordion.update(open=False),\
gr.CheckboxGroup.update(choices=getBackgroundVideoChoices(), interactive=True),\
gr.CheckboxGroup.update(choices=getBackgroundMusicChoices(), interactive=True)
def __get_first_preview(self):
'''Get the first preview'''
return self.__get_asset_embed(AssetDatabase.get_df(), 0)
def __delete_clicked(self, button_name):
'''Delete an asset'''
asset_name = button_name.split("🗑️ Delete ")[-1]
AssetDatabase.remove_asset(asset_name)
data = AssetDatabase.get_df()
if len(data) > 0:
return gr.update(value=data),\
gr.HTML.update(value=self.__get_asset_embed(data, 0)),\
gr.update(value=f"🗑️ Delete {data.iloc[0]['name']}"),\
gr.CheckboxGroup.update(choices=getBackgroundVideoChoices(), interactive=True),\
gr.CheckboxGroup.update(choices=getBackgroundMusicChoices(), interactive=True)
return gr.Dataframe.update(value=data),\
gr.HTML.update(visible=True),\
gr.Button.update(value="🗑️ Delete"),\
gr.CheckboxGroup.update(choices=getBackgroundVideoChoices(), interactive=True),\
gr.CheckboxGroup.update(choices=getBackgroundMusicChoices(), interactive=True)
def __preview_asset(self, data, evt: gr.SelectData):
'''Preview an asset'''
html_embed = self.__get_asset_embed(data, evt.index[0])
return gr.HTML.update(value=html_embed), gr.update(value=f"🗑️ Delete {data.iloc[evt.index[0]]['name']}")
def __get_asset_embed(self, data, row):
'''Get the asset embed'''
embed_height = 300
embed_width = 300
asset_link = data.iloc[row]['link']
if 'youtube.com' in asset_link:
asset_link = f"https://youtube.com/embed/{asset_link.split('?v=')[-1]}"
embed_html = f'<iframe width="{embed_width}" height="{embed_height}" src="{asset_link}"></iframe>'
elif 'public/' in asset_link:
asset_link = f"http://localhost:31415/file={asset_link}"
file_ext = asset_link.split('.')[-1]
if file_ext in ['mp3', 'wav', 'ogg']:
audio_type = 'audio/mpeg' if file_ext == 'mp3' else f'audio/{file_ext}'
embed_html = f'<audio controls><source src="{asset_link}" type="{audio_type}">Your browser does not support the audio tag.</audio>'
elif file_ext in ['mp4', 'webm', 'ogg', 'mov']:
video_type = 'video/mp4' if file_ext == 'mp4' else f'video/{file_ext}'
embed_html = f'<video width="{embed_width}" height="{embed_height}" style="max-height: 100%;" controls><source src="{asset_link}" type="{video_type}">Your browser does not support the video tag.</video>'
elif file_ext in ['jpg', 'jpeg', 'png', 'gif']:
embed_html = f'<img src="{asset_link}" width="{embed_width}" height="{embed_height}">'
else:
embed_html = 'Unsupported file type'
return embed_html
"""
|