From ebef29bf02647028bad930af9862b717e86af41e Mon Sep 17 00:00:00 2001
From: Saurav Panda <sgp65@cornell.edu>
Date: Thu, 27 Feb 2025 00:37:31 -0800
Subject: [PATCH] TTS isnt working in chrome extension so added a message

---
 extensions/chrome/src/helpers/executors.tsx   | 53 ++++++++++++++++++-
 extensions/chrome/src/popup/workflow-view.tsx | 38 ++++++++++++-
 2 files changed, 89 insertions(+), 2 deletions(-)

diff --git a/extensions/chrome/src/helpers/executors.tsx b/extensions/chrome/src/helpers/executors.tsx
index 4bc9cb5..10f21f1 100644
--- a/extensions/chrome/src/helpers/executors.tsx
+++ b/extensions/chrome/src/helpers/executors.tsx
@@ -155,7 +155,8 @@ const nodeExecutors = {
         truncatedPrompt,
         {
           temperature: node.nodeData?.temperature || 0.7,
-          max_tokens: node.nodeData?.maxTokens || 2048
+          max_tokens: node.nodeData?.maxTokens || 2048,
+          system_prompt: node.nodeData?.systemPrompt
         }
       );
 
@@ -215,6 +216,11 @@ const nodeExecutors = {
   'transcriptionAgent': async (node: WorkflowStep, input: any, params?: ExecuteWorkflowParams) => {
     try {
       console.debug("transcription-agent", node, input);
+      
+      // Throw a specific error for speech transcription in Chrome extension
+      throw new Error("Speech transcription models are not supported in the Chrome extension. Please use the web app version instead.");
+      
+      // The code below will not execute due to the error above
       const browserAI = new BrowserAI();
 
       await browserAI.loadModel(node.nodeData?.model || 'whisper-tiny-en', {
@@ -246,6 +252,51 @@ const nodeExecutors = {
       throw error;
     }
   },
+
+  'ttsAgent': async (node: WorkflowStep, input: any, params?: ExecuteWorkflowParams) => {
+    try {
+      console.debug("tts-agent", node, input);
+      
+      // Throw a specific error for TTS in Chrome extension
+      throw new Error("Text-to-speech models are not supported in the Chrome extension. Please use the web app version instead.");
+      
+      // The code below will not execute due to the error above
+      const browserAI = new BrowserAI();
+
+      await browserAI.loadModel(node.nodeData?.model || 'kokoro-tts', {
+        onProgress: (progress: any) => {
+          const progressPercent = progress.progress || 0;
+          const eta = progress.eta || 0;
+          params?.onModelLoadProgress?.(progressPercent * 100, eta);
+        }
+      });
+
+      // Extract text input
+      if (!input) {
+        throw new Error('No text input provided to TTS agent');
+      }
+
+      // Generate speech
+      const audioData = await browserAI.textToSpeech(input, {
+        voice: node.nodeData?.voice || 'af_bella'
+      });
+      
+      // Create blob with proper MIME type
+      const blob = new Blob([audioData], { type: 'audio/wav' });
+      
+      // Create and store blob URL
+      const audioUrl = URL.createObjectURL(blob);
+
+      return {
+        success: true,
+        output: audioUrl,
+        log: `Text-to-speech generated successfully using ${node.nodeData?.model || 'bark-small'}`
+      };
+    } catch (error) {
+      console.error('TTSAgent error:', error);
+      throw error;
+    }
+  },
 };
 
 export const executeWorkflow = async ({
diff --git a/extensions/chrome/src/popup/workflow-view.tsx b/extensions/chrome/src/popup/workflow-view.tsx
index ac9f06e..c24f7de 100644
--- a/extensions/chrome/src/popup/workflow-view.tsx
+++ b/extensions/chrome/src/popup/workflow-view.tsx
@@ -59,6 +59,27 @@ export function WorkflowView({ workflow, onBack }: WorkflowViewProps) {
   const [modelLoadProgress, setModelLoadProgress] = useState<number | null>(null);
   const [modelLoadEta, setModelLoadEta] = useState<number | null>(null);
 
+  // Add this function to check for unsupported nodes
+  const hasUnsupportedNodes = () => {
+    return nodes.some(node => 
+      node.nodeType?.toLowerCase() === 'ttsagent' || 
+      node.nodeType?.toLowerCase() === 'transcriptionagent'
+    );
+  };
+
+  // Get the unsupported node warning message
+  const getUnsupportedWarningMessage = () => {
+    const unsupportedTypes = nodes
+      .filter(node => 
+        node.nodeType?.toLowerCase() === 'ttsagent' || 
+        node.nodeType?.toLowerCase() === 'transcriptionagent'
+      )
+      .map(node => node.nodeType?.toLowerCase() === 'ttsagent' ? 'Text-to-Speech' : 'Speech Transcription');
+    
+    const uniqueTypes = [...new Set(unsupportedTypes)];
+    return `${uniqueTypes.join(' and ')} ${uniqueTypes.length > 1 ? 'are' : 'is'} not supported in the Chrome extension. Please use the web app version instead.`;
+  };
+
   useEffect(() => {
     console.log('Workflow data received:', workflow);
     console.log('Initial nodes:', nodes);
@@ -71,6 +92,15 @@ export function WorkflowView({ workflow, onBack }: WorkflowViewProps) {
         fullNodeData: JSON.stringify(node.nodeData, null, 2)
       });
     });
+    
+    // Show toast warning if workflow contains unsupported nodes
+    if (hasUnsupportedNodes()) {
+      toast({
+        variant: "destructive",
+        title: "Unsupported workflow",
+        description: getUnsupportedWarningMessage()
+      });
+    }
   }, []);
 
   // Update areAllInputsFilled function
@@ -273,7 +303,7 @@ export function WorkflowView({ workflow, onBack }: WorkflowViewProps) {
         <div className="flex items-center gap-8">
           <Button 
             onClick={handleExecute}
-            disabled={isExecuting || !areAllInputsFilled()}
+            disabled={isExecuting || !areAllInputsFilled() || hasUnsupportedNodes()}
             className="flex items-center gap-2"
           >
             <Play className="h-4 w-4" />
@@ -283,6 +313,12 @@ export function WorkflowView({ workflow, onBack }: WorkflowViewProps) {
       </div>
 
       <div className="flex-1 overflow-y-auto">
+        {hasUnsupportedNodes() && (
+          <div className="p-3 bg-destructive/10 text-destructive text-sm mb-2 mx-4 mt-4 rounded-md">
+            <strong>Warning:</strong> {getUnsupportedWarningMessage()}
+          </div>
+        )}
+
         {modelLoadProgress !== null && (
           <div className="p-2 bg-primary/10 text-primary text-sm sticky top-0 z-10">
             <div className="flex items-center gap-2">