Update crew.py

crew.py

@@ -1,6 +1,7 @@
 import os
 
 from crewai import Agent, Crew, Process, Task
+from crewai.tools import tool
 from crewai_tools import (
     CodeInterpreterTool,
     SerperDevTool,
@@ -17,6 +18,7 @@ from util import get_final_answer
 MANAGER_MODEL      = "gpt-4.1"
 AGENT_MODEL        = "gpt-4.1"
 FINAL_ANSWER_MODEL = "gpt-4.5-preview"
+STT_MODEL          = "whisper-1"
 
 # LLM evaluation
 
@@ -35,7 +37,33 @@ CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
 def run_crew(question, file_path):
     # Custom tools
 
-    audio_analysis_tool = get_audio_analysis_tool()
+    @tool("Audio Analysis Tool")
+    def audio_analysis_tool(question: str, file_path: str) -> str:
+        """Transcribe audio file using OpenAI's Whisper model.
+    
+           Args:
+               question (str): Question to answer
+               file_path (str): Path of the audio file to transcribe
+                
+           Returns:
+               str: Transcribed text from the audio file
+                
+           Raises:
+               FileNotFoundError: If audio file does not exist
+               RuntimeError: If transcription fails"""
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"Audio file not found: {file_path}")
+            
+        try:
+            client = OpenAI()
+            transcript = client.audio.transcriptions.create(
+                file=open(file_path, "rb"),
+                model=STT_MODEL,
+                prompt=question,
+            )
+            return transcript.text
+        except Exception as e:
+            raise RuntimeError(f"Failed to transcribe audio: {str(e)}")
 
     # Built-in tools