Update agent.py

agent.py

@@ -87,11 +87,19 @@ class SpeechToTextTool(Tool):
 
     def __init__(self):
         super().__init__()
-        self.model = whisper.load_model("base")
+        try:
+            print("Loading whisper model 'base'...")
+            self.model = whisper.load_model("base")
+            print("Whisper model loaded successfully.")
+        except Exception as e:
+            print(f"Error loading whisper model: {e}")
+            self.model = None
 
     def forward(self, audio_path: str) -> str:
         if not os.path.exists(audio_path):
             return f"Error: File not found at {audio_path}"
+        if self.model is None:
+            return "Error: Whisper model failed to load during initialization."
         try:
             print(f"Starting transcription for {audio_path}...")
             result = self.model.transcribe(audio_path)