Add note -- only show for non-paying and non-admin users

GooeyAI · Oct 12, 2023 · cb71576 · cb71576
1 parent 6eb75c1
commit cb71576
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/daras_ai_v2/text_to_speech_settings_widgets.py b/daras_ai_v2/text_to_speech_settings_widgets.py
@@ -134,7 +134,7 @@ class TextToSpeechProviders(Enum):
 }
 
 
-def text_to_speech_settings():
+def text_to_speech_settings(page=None):
     st.write(
         """
         ##### 🗣️ Voice Settings
@@ -226,6 +226,12 @@ def text_to_speech_settings():
 
         case TextToSpeechProviders.ELEVEN_LABS.name:
             with col2:
+                if not (
+                    page
+                    and (page.is_current_user_paying() or page.is_current_user_admin())
+                ):
+                    st.caption("Note: Eleven Labs is only available for paying users.")
+
                 st.selectbox(
                     """
                     ###### Voice name (ElevenLabs)

diff --git a/recipes/TextToSpeech.py b/recipes/TextToSpeech.py
@@ -104,7 +104,7 @@ def validate_form_v2(self):
         assert st.session_state["text_prompt"], "Text input cannot be empty"
 
     def render_settings(self):
-        text_to_speech_settings()
+        text_to_speech_settings(page=self)
 
     def get_raw_price(self, state: dict):
         tts_provider = self._get_tts_provider(state)

diff --git a/recipes/VideoBots.py b/recipes/VideoBots.py
@@ -375,7 +375,7 @@ def render_settings(self):
             st.write("---")
             st.session_state["tts_provider"] = None
         else:
-            text_to_speech_settings()
+            text_to_speech_settings(page=self)
 
         st.write("---")
         if not "__enable_video" in st.session_state: