Update AudioClassifier.py

AudioClassifier.py

@@ -18,6 +18,7 @@ class AudioClassifier:
 
     Attributes:
         vocab (list): Vocabulary of valid commands
+        commands (list): List of corresponding mouse actions
         pipe: The Hugging Face Transformers pipeline for audio classification.
     """
 
@@ -28,6 +29,9 @@ class AudioClassifier:
         self.vocab = ["left", "right", "up", "down", "go", "follow",
                       "on", "off", "one", "two", "three", "stop"]
 
+        self.commands = ["left click", "right click", "scroll up", "scroll down", "double click", "sustained click", "enable cursor movement",
+                         "disable cursor movement", "slow cursor speed", "medium cursor speed", "fast cursor speed", "finish the application"]
+
         # Load the audio classification pipeline
         self.pipe = pipeline("audio-classification", model="0xb1/wav2vec2-base-finetuned-speech_commands-v0.02")
 
@@ -44,9 +48,11 @@ class AudioClassifier:
         _, audio = wavfile.read(audio_path)
         audio = decimate(audio, 3)
         result = self.pipe(audio)[0]["label"]
-
+        
         if result not in self.vocab:
-            result = 'unknown'
+            result = 'unknown command'
+        else:
+            result = result + ' ---> ' + '(' + self.commands[self.vocab.index(result)] + ')'
 
         return result