Update app.py

app.py

@@ -3,7 +3,7 @@ import base64
 import json
 import os
 import pathlib
-from typing import AsyncGenerator, Literal, Optional, List
+from typing import AsyncGenerator, Literal, Optional, List, Union
 
 import gradio as gr
 import numpy as np
@@ -32,26 +32,10 @@ load_dotenv()
 
 # Define preset system prompts with more detailed and actionable instructions
 SYSTEM_PROMPTS = {
-    "default": [
-        "You are a helpful, harmless, and honest AI assistant.",
-        "Provide clear, concise, and accurate information.",
-        "Adapt your communication style to be most helpful to the user."
-    ],
-    "Behavior Analyst": [
-        "You are an expert in behavior analysis, Positive Behavior Interventions and Supports, and Restorative approaches and practices.",
-        "Your primary role is to help educators understand and address student behavior issues.",
-        "Provide evidence-based strategies, empathetic insights, and practical solutions.",
-        "Focus on understanding the underlying causes of behavior and developing supportive interventions."
-    ],
-    "UDL Expert": [
-        "You are an expert in Universal Design for Learning (UDL).",
-        "Your role is to help educators design units, lessons, and activities to be universally accessible.",
-        "Provide detailed guidance on addressing student variability and potential learning barriers.",
-        "Offer concrete strategies for multiple means of engagement, representation, and expression."
-    ],
-    "prompt3": [
-        "PLACEHOLDER FOR PROMPT 3 WITH MULTIPLE INSTRUCTION PARTS"
-    ]
+    "default": "You are a helpful, harmless, and honest AI assistant. Provide clear, concise, and accurate information.",
+    "Behavior Analyst": "You are an expert in behavior analysis, Positive Behavior Interventions and Supports, and Restorative approaches and practices. Your primary role is to help educators understand and address student behavior issues. Provide evidence-based strategies, empathetic insights, and practical solutions.",
+    "UDL Expert": "You are an expert in Universal Design for Learning (UDL). Your role is to help educators design units, lessons, and activities to be universally accessible. Provide detailed guidance on addressing student variability and potential learning barriers.",
+    "prompt3": "PLACEHOLDER FOR PROMPT 3"
 }
 
 
@@ -90,7 +74,7 @@ class GeminiHandler(AsyncStreamHandler):
         self, 
         prompt_key: Optional[str] = None, 
         custom_prompt: Optional[str] = None
-    ) -> Optional[List[str]]:
+    ) -> Optional[str]:
         """
         Prepare system instruction based on preset or custom prompt.
         
@@ -99,12 +83,11 @@ class GeminiHandler(AsyncStreamHandler):
             custom_prompt: Custom user-defined system prompt
         
         Returns:
-            List of system instruction parts or None
+            System instruction as a string or None
         """
         # Check for custom prompt first
         if custom_prompt:
-            # If custom prompt is provided, split it into paragraphs
-            return [p.strip() for p in custom_prompt.split('\n') if p.strip()]
+            return custom_prompt
         
         # Then check for preset prompt
         if prompt_key and prompt_key in SYSTEM_PROMPTS:
@@ -128,19 +111,26 @@ class GeminiHandler(AsyncStreamHandler):
             http_options={"api_version": "v1alpha"},
         )
 
-        # Create config with system instructions if available
-        config = LiveConnectConfig(
-            response_modalities=["AUDIO"],  # type: ignore
-            speech_config=SpeechConfig(
+        # Create config for the connection
+        config_kwargs = {
+            "response_modalities": ["AUDIO"],
+            "speech_config": SpeechConfig(
                 voice_config=VoiceConfig(
                     prebuilt_voice_config=PrebuiltVoiceConfig(
                         voice_name=voice_name,
                     )
                 )
-            ),
-            # Add system instruction if available
-            **({"system_instruction": system_instruction} if system_instruction else {})
-        )
+            )
+        }
+
+        # Add system instruction if available
+        if system_instruction:
+            config_kwargs["system_instruction"] = {
+                "parts": [{"text": system_instruction}]
+            }
+        
+        # Create the configuration
+        config = LiveConnectConfig(**config_kwargs)
         
         try:
             async with client.aio.live.connect(
@@ -175,7 +165,6 @@ class GeminiHandler(AsyncStreamHandler):
     def shutdown(self) -> None:
         self.quit.set()
 
-
 stream = Stream(
     modality="audio",
     mode="send-receive",