Update crew.py

crew.py

@@ -9,6 +9,7 @@ from crewai_tools import (
     WebsiteSearchTool,
     YoutubeVideoSearchTool
 )
+from openai import OpenAI
 from openinference.instrumentation.crewai import CrewAIInstrumentor
 from phoenix.otel import register
 from util import get_final_answer
@@ -32,11 +33,12 @@ def run_crew(question, file_name):
     # Tools
 
     @tool("Audio Analysis Tool")
-    def audio_analysis_tool(file_name: str) -> str:
+    def audio_analysis_tool(question: str, file_name: str) -> str:
         """Transcribe audio file using OpenAI's Whisper model.
     
            Args:
-               file_path (str): Path to the audio file to transcribe
+               question (str): Question to answer
+               file_name (str): Name of the audio file to transcribe
                 
            Returns:
                str: Transcribed text from the audio file
@@ -50,13 +52,13 @@ def run_crew(question, file_name):
             
         try:
             # Load Whisper model (using base model by default)
-            model = whisper.load_model("base")
-            
-            # Transcribe the audio file
-            result = model.transcribe(file_name)
-            
-            # Return transcribed text
-            return result["text"].strip()
+            client = OpenAI()
+            client.audio.transcriptions.create(
+                file=open(audio_filepath, "rb"),
+                model="whisper-1",
+                prompt=prompt,
+            )
+            return transcript.text
         except Exception as e:
             raise RuntimeError(f"Failed to transcribe audio: {str(e)}")
         ###