release 0.3

app.py

@@ -8,28 +8,14 @@ import gradio as gr
 import yaml
 
 from src.elevenlabs import (Speaker, check_voice_exists, get_make_voice,
-                            play_history, save_history)
-from src.openailib import top_response, speech_to_text
+                            play_history, save_history, set_elevenlabs_key)
+from src.openailib import top_response, speech_to_text, set_openai_key
 from src.tube import extract_audio
 
 logging.basicConfig(level=logging.INFO)
 log = logging.getLogger(__name__)
 
 
-def set_openai_key(openai_api_key_textbox):
-    log.info(f"Setting OpenAI key.")
-    os.environ["OPENAI_API_KEY"] = openai_api_key_textbox
-    import openai
-    openai.api_key = os.getenv("OPENAI_API_KEY")
-
-
-def set_elevenlabs_key(elevenlabs_api_key_textbox):
-    log.info(f"Setting ElevenLabs key.")
-    os.environ["ELEVENLABS_API_KEY"] = elevenlabs_api_key_textbox
-    import elevenlabs
-    elevenlabs.api_key = os.getenv("ELEVENLABS_API_KEY")
-
-
 class ConversationState:
     COLORS: list = ['#FFA07A', '#F08080', '#AFEEEE', '#B0E0E6', '#DDA0DD',
                     '#FFFFE0', '#F0E68C', '#90EE90', '#87CEFA', '#FFB6C1']

src/elevenlabs.py

@@ -16,13 +16,21 @@ from .utils import timeit
 logging.basicConfig(level=logging.INFO)
 log = logging.getLogger(__name__)
 
-try:
-    USER = ElevenLabsUser(os.environ["ELEVENLABS_API_KEY"])
-except KeyError as e:
-    USER = None
-    log.warning("ELEVENLABS_API_KEY not found in environment variables.")
-    pass
-
+USER = None
+
+def set_elevenlabs_key(elevenlabs_api_key_textbox=None):
+    global USER
+    log.info(f"Setting ElevenLabs key.")
+    if elevenlabs_api_key_textbox is not None:
+        os.environ["ELEVENLABS_API_KEY"] = elevenlabs_api_key_textbox
+    try:
+        USER = ElevenLabsUser(os.environ["ELEVENLABS_API_KEY"])
+    except KeyError as e:
+        USER = None
+        log.warning("ELEVENLABS_API_KEY not found in environment variables.")
+        pass
+
+set_elevenlabs_key()
 
 @dataclass
 class Speaker:

src/openailib.py

@@ -8,12 +8,18 @@ import openai
 logging.basicConfig(level=logging.INFO)
 log = logging.getLogger(__name__)
 
-try:
-    openai.api_key = os.getenv("OPENAI_API_KEY")
-except KeyError as e:
-    log.warning("OPENAI_API_KEY not found in environment variables.")
-    pass
 
+def set_openai_key(openai_api_key_textbox = None):
+    log.info(f"Setting OpenAI key.")
+    if openai_api_key_textbox is not None:
+        os.environ["OPENAI_API_KEY"] = openai_api_key_textbox
+    try:
+        openai.api_key = os.getenv("OPENAI_API_KEY")
+    except KeyError as e:
+        log.warning("OPENAI_API_KEY not found in environment variables.")
+        pass
+
+set_openai_key()
 
 @timeit
 def speech_to_text(audio_path):

voices.yaml

@@ -27,4 +27,16 @@ LexFridman:
       duration_seconds: 30
     - url: "https://youtu.be/DxREm3s1scA"
       start_minute: 41.4
-      duration_seconds: 30
+      duration_seconds: 30
+# MrBeast:
+#   description: "Charismatic YouTuber, known for outlandish philanthropy."
+# TaylorSwift:
+#   description: "Talented singer-songwriter, often reflects on past relationships."
+# GretaThunberg:
+#   description: "Passionate environmental activist, speaks truth to power."
+# JoeRogan:
+#   description: "Unfiltered and curious podcaster, delves into various topics."
+# OprahWinfrey:
+#   description: "Empathetic and inspiring media mogul, known for generosity."
+# TrevorNoah:
+#   description: "Witty comedian and TV host, offers humorous social commentary."