:sparkles: :wrench: reconfigure to radial, interface changes

groq-voicechat-demo.py

@@ -8,29 +8,19 @@ groq_api_key = os.environ.get("GROQ_API_KEY")
 
 client = Groq(api_key = groq_api_key)
 
-model_list =[
-    'openai/gpt-oss-120b',
-    'moonshotai/kimi-k2-instruct',
-    'meta-llama/llama-4-scout-17b-16e-instruct',
-    'openai/gpt-oss-20b',
-    'qwen/qwen3-32b',
-    'llama-3.3-70b-versatile',
-    'moonshotai/kimi-k2-instruct-0905',
-    'allam-2-7b',
-    'meta-llama/llama-4-maverick-17b-128e-instruct',
-    'llama-3.1-8b-instant',
-]
-
-reasoning_models = [
-    'openai/gpt-oss-120b',
-    'openai/gpt-oss-20b',
-    'qwen/qwen3-32b',
-]
-
-default_sys_prompt = """You are a helpful chatbot. You help the end user to the best of your ability."""
+options = {
+    'always agree': [ #I hate that I have a space here but I'm also being lazy.
+        'meta-llama/llama-4-maverick-17b-128e-instruct',
+        'You are roleplaying as a naughty friend who affirms and encourages all ideas, even silly ideas. You talk in a way that a 7 year old can understand. Keep your responses short and expressive.'],
+    'honest': [
+        'moonshotai/kimi-k2-instruct-0905',
+        'You are roleplaying as a good friend who is kind, fun and supportive, but who gently pushes back on bad or dangerous ideas, and encourages honesty and integrity. You talk in a way that a 7 year old can understand. Keep your responses short and expressive.'
+    ]
+}
+
 chat_history = []
 
-def groq_voicechat(new_message: tuple, chat_history: list[dict], model: str, system_prompt: str, ):
+def groq_voicechat(new_message: tuple, chat_history: list[dict], chatbot_options: str):
     '''
     Groq chat API call wrapper.
 
@@ -44,9 +34,9 @@ def groq_voicechat(new_message: tuple, chat_history: list[dict], model: str, sys
         - "" - used to delete old input msg in chat textbox lol
         - nonsys_msg_hist [list[dict]]: updated chat history
     '''
-    if model not in model_list:
-        raise ValueError(f"model must be in model_list: {model_list}")
-        return
+
+    model = options[chatbot_options][0]
+    system_prompt = options[chatbot_options][1]
 
     #augment chat hist
     
@@ -76,19 +66,11 @@ def groq_voicechat(new_message: tuple, chat_history: list[dict], model: str, sys
                      ]
 
     input_msg_hist.extend(nonsys_msg_hist)
-    
-    if model in reasoning_models:
-        chat_completion = client.chat.completions.create(
-            messages = input_msg_hist,
-            model = model,
-            include_reasoning = False, #removes reasoning tokens from output because I'm lazy
-        )
-    else:
-        chat_completion = client.chat.completions.create(
-            messages = input_msg_hist,
-            model = model,
-            # include_reasoning = False, #removes reasoning tokens from output because I'm lazy
-        )
+
+    chat_completion = client.chat.completions.create(
+        messages = input_msg_hist,
+        model = model,
+    )
     output_msg = chat_completion.choices[0].message.content
 
     # add to chat hist
@@ -108,13 +90,14 @@ def process_audio(audio: tuple):
 def create_demo():
     with gr.Blocks() as demo:
         with gr.Row():
-            model = gr.Dropdown(model_list,
-                               )
-        with gr.Row():
-            system_prompt = gr.Textbox(
-                value=default_sys_prompt,
-                interactive=True
+
+            chatbot_options = gr.Radio(
+                choices = list(options.keys()),
+                value = list(options.keys())[0],
+                label = "Chatbot behaviours",
+                show_label=True
             )
+
         with gr.Row():
             chatbot = gr.Chatbot(
                 label="Conversation",
@@ -127,34 +110,40 @@ def create_demo():
                 type="numpy",
                 streaming=False,
             )
-        with gr.Row():
-            clear = gr.ClearButton([voiceinput, chatbot], variant = 'stop')
 
         with gr.Row():
             playback_button = gr.Button("playback last message")
-        with gr.Row():
+            
+        with gr.Accordion("Chatbot Voice Options", open = False):
             voice_opps = gr.Dropdown(
+                label = "Choose the voice of the chatbot (some are American and some are British!)",
                 choices = list(voice_choices.keys()),
                 value = list(voice_choices.keys())[0]
             )
-        with gr.Row():
             voice_speed = gr.Slider(
+                label = "Choose the speed at which the chatbot talks",
                 minimum = 0.5,
                 maximum = 2,
                 value = 1,
                 step = 0.1,
             )
+
+        with gr.Row():
+            clear = gr.ClearButton(components = [voiceinput, chatbot], value = "Clear chat history", variant = 'stop')
+            
+
         with gr.Row():
             audio_out = gr.Audio(
                 label = "Output Audio",
                 interactive = False,
                 autoplay = True,
-                streaming = True
+                streaming = True,
+                visible = "hidden",
             )
         
         output = voiceinput.stop_recording(
             groq_voicechat, 
-            [voiceinput, chatbot, model, system_prompt,  ],
+            [voiceinput, chatbot, chatbot_options],
             [chatbot]
         ) #WHAT AM I DOING LOL - COME BACK TO THIS
 
@@ -165,8 +154,6 @@ def create_demo():
                 for chunk in gen_object:
                     yield chunk
             return None
-            #     return generate_tts(last_message)
-            # return None
         
         playback_button.click(
             playback_last_message,
@@ -181,6 +168,7 @@ if __name__ == "__main__":
     demo.launch(
         auth=("DigitalChild", "IhateBroccoli123"),
         ssr_mode=False,
+        theme="citrus"
     )