add code (#137)

freddyaboulton · Mar 7, 2025 · 11dae29 · 11dae29
1 parent 7dfee78
commit 11dae29
Show file tree

Hide file tree

Showing 2 changed files with 60 additions and 1 deletion.
diff --git a/backend/fastrtc/__init__.py b/backend/fastrtc/__init__.py
@@ -6,7 +6,7 @@
 from .reply_on_pause import AlgoOptions, ReplyOnPause, SileroVadOptions
 from .reply_on_stopwords import ReplyOnStopWords
 from .speech_to_text import MoonshineSTT, get_stt_model
-from .stream import Stream
+from .stream import Stream, UIArgs
 from .text_to_speech import KokoroTTSOptions, get_tts_model
 from .tracks import (
     AsyncAudioVideoStreamHandler,
@@ -62,4 +62,5 @@
     "get_tts_model",
     "KokoroTTSOptions",
     "wait_for_item",
+    "UIArgs",
 ]
diff --git a/backend/fastrtc/stream.py b/backend/fastrtc/stream.py
@@ -36,6 +36,8 @@ class Body(BaseModel):
 class UIArgs(TypedDict):
     title: NotRequired[str]
     """Title of the demo"""
+    subtitle: NotRequired[str]
+    """Subtitle of the demo. Text will be centered and displayed below the title."""
     icon: NotRequired[str]
     """Icon to display on the button instead of the wave animation. The icon should be a path/url to a .svg/.png/.jpeg file."""
     icon_button_color: NotRequired[str]
@@ -192,6 +194,14 @@ def _generate_default_ui(
                 </h1>
                 """
                 )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
+                """
+                    )
                 with gr.Row():
                     with gr.Column():
                         if additional_input_components:
@@ -231,6 +241,14 @@ def _generate_default_ui(
                 </h1>
                 """
                 )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
+                """
+                    )
                 with gr.Row():
                     if additional_input_components:
                         with gr.Column():
@@ -271,6 +289,14 @@ def _generate_default_ui(
                 </h1>
                 """
                 )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
+                """
+                    )
                 with gr.Column(elem_classes=["my-column"]):
                     with gr.Group(elem_classes=["my-group"]):
                         image = WebRTC(
@@ -310,6 +336,14 @@ def _generate_default_ui(
                 </h1>
                 """
                 )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
+                """
+                    )
                 with gr.Row():
                     with gr.Column():
                         for component in additional_input_components:
@@ -354,6 +388,14 @@ def _generate_default_ui(
                 </h1>
                 """
                 )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
+                """
+                    )
                 with gr.Row():
                     with gr.Column():
                         with gr.Group():
@@ -397,6 +439,14 @@ def _generate_default_ui(
                 </h1>
                 """
                 )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                    {ui_args.get("subtitle")}
+                </div>
+                """
+                    )
                 with gr.Row():
                     with gr.Column():
                         with gr.Group():
@@ -441,6 +491,14 @@ def _generate_default_ui(
                 </h1>
                 """
                 )
+                if ui_args.get("subtitle"):
+                    gr.Markdown(
+                        f"""
+                <div style='text-align: center'>
+                {ui_args.get("subtitle")}
+                </div>
+                """
+                    )
                 with gr.Row():
                     with gr.Column():
                         with gr.Group():