File size: 5,253 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 |
import time
import gradio as gr
from gui.asset_components import AssetComponentsUtils
from gui.ui_abstract_component import AbstractComponentUI
from shortGPT.api_utils.eleven_api import ElevenLabsAPI
from shortGPT.config.api_db import ApiKeyManager
class ConfigUI(AbstractComponentUI):
def __init__(self):
self.api_key_manager = ApiKeyManager()
eleven_key = self.api_key_manager.get_api_key('ELEVEN LABS')
self.eleven_labs_api = ElevenLabsAPI(eleven_key) if eleven_key else None
def on_show(self, button_text, textbox, button):
'''Show or hide the API key'''
if button_text == "Show":
return gr.Textbox.update(type="text"), gr.Button.update(value="Hide")
return gr.Textbox.update(type="password"), gr.Button.update(value="Show")
def verify_eleven_key(self, eleven_key, remaining_chars):
'''Verify the ElevenLabs API key'''
if (eleven_key and self.api_key_manager.get_api_key('ELEVEN LABS') != eleven_key):
try:
self.eleven_labs_api = ElevenLabsAPI(eleven_key)
print(self.eleven_labs_api)
return self.eleven_labs_api.get_remaining_characters()
except Exception as e:
raise gr.Error(e.args[0])
return remaining_chars
def save_keys(self, openai_key, eleven_key, pexels_key):
'''Save the keys in the database'''
if (self.api_key_manager.get_api_key('OPENAI') != openai_key):
self.api_key_manager.set_api_key("OPENAI", openai_key)
if (self.api_key_manager.get_api_key('PEXELS') != pexels_key):
self.api_key_manager.set_api_key("PEXELS", pexels_key)
if (self.api_key_manager.get_api_key('ELEVEN LABS') != eleven_key):
self.api_key_manager.set_api_key("ELEVEN LABS", eleven_key)
new_eleven_voices = AssetComponentsUtils.getElevenlabsVoices()
return gr.Textbox.update(value=openai_key),\
gr.Textbox.update(value=eleven_key),\
gr.Textbox.update(value=pexels_key),\
gr.Radio.update(choices=new_eleven_voices),\
gr.Radio.update(choices=new_eleven_voices)
return gr.Textbox.update(value=openai_key),\
gr.Textbox.update(value=eleven_key),\
gr.Textbox.update(value=pexels_key),\
gr.Radio.update(visible=True),\
gr.Radio.update(visible=True)
def get_eleven_remaining(self,):
'''Get the remaining characters from ElevenLabs API'''
if (self.eleven_labs_api):
try:
return self.eleven_labs_api.get_remaining_characters()
except Exception as e:
return e.args[0]
return ""
def back_to_normal(self):
'''Back to normal after 3 seconds'''
time.sleep(3)
return gr.Button.update(value="save")
def create_ui(self):
'''Create the config UI'''
with gr.Tab("Config") as config_ui:
with gr.Row():
with gr.Column():
with gr.Row():
openai_textbox = gr.Textbox(value=self.api_key_manager.get_api_key("OPENAI"), label=f"OPENAI API KEY", show_label=True, interactive=True, show_copy_button=True, type="password", scale=40)
show_openai_key = gr.Button("Show", size="sm", scale=1)
show_openai_key.click(self.on_show, [show_openai_key], [openai_textbox, show_openai_key])
with gr.Row():
eleven_labs_textbox = gr.Textbox(value=self.api_key_manager.get_api_key("ELEVEN LABS"), label=f"ELEVEN LABS API KEY", show_label=True, interactive=True, show_copy_button=True, type="password", scale=40)
eleven_characters_remaining = gr.Textbox(value=self.get_eleven_remaining(), label=f"CHARACTERS REMAINING", show_label=True, interactive=False, type="text", scale=40)
show_eleven_key = gr.Button("Show", size="sm", scale=1)
show_eleven_key.click(self.on_show, [show_eleven_key], [eleven_labs_textbox, show_eleven_key])
with gr.Row():
pexels_textbox = gr.Textbox(value=self.api_key_manager.get_api_key("PEXELS"), label=f"PEXELS KEY", show_label=True, interactive=True, show_copy_button=True, type="password", scale=40)
show_pexels_key = gr.Button("Show", size="sm", scale=1)
show_pexels_key.click(self.on_show, [show_pexels_key], [pexels_textbox, show_pexels_key])
save_button = gr.Button("save", size="sm", scale=1)
save_button.click(self.verify_eleven_key, [eleven_labs_textbox, eleven_characters_remaining], [eleven_characters_remaining]).success(
self.save_keys, [openai_textbox, eleven_labs_textbox, pexels_textbox], [openai_textbox, eleven_labs_textbox, pexels_textbox, AssetComponentsUtils.voiceChoice(), AssetComponentsUtils.voiceChoiceTranslation()])
save_button.click(lambda _: gr.Button.update(value="Keys Saved !"), [], [save_button])
save_button.click(self.back_to_normal, [], [save_button])
return config_ui
|