Update crew.py

crew.py

@@ -16,6 +16,8 @@ from util import get_final_answer
 
 MANAGER_MODEL = "gpt-4.1"
 AGENT_MODEL = "gpt-4.1"
+STT_MODEL = "whisper-1"
+FINAL_ANSWER_MODEL = ""
 
 PHOENIX_API_KEY = os.environ["PHOENIX_API_KEY"]
 
@@ -29,7 +31,7 @@ tracer_provider = register(
 
 CrewAIInstrumentor().instrument(tracer_provider=tracer_provider)
 
-def run_crew(question, file_name):
+def run_crew(question, file_path):
     # Tools
 
     @tool("Audio Analysis Tool")
@@ -38,7 +40,7 @@ def run_crew(question, file_name):
     
            Args:
                question (str): Question to answer
-               file_name (str): Name of the audio file to transcribe
+               file_path (str): Path of the audio file to transcribe
                 
            Returns:
                str: Transcribed text from the audio file
@@ -47,15 +49,15 @@ def run_crew(question, file_name):
                FileNotFoundError: If audio file does not exist
                RuntimeError: If transcription fails"""
         ###
-        if not os.path.exists(file_name):
-            raise FileNotFoundError(f"Audio file not found: {file_name}")
+        if not os.path.exists(file_path):
+            raise FileNotFoundError(f"Audio file not found: {file_path}")
             
         try:
             # Load Whisper model (using base model by default)
             client = OpenAI()
             client.audio.transcriptions.create(
-                file=open(audio_filepath, "rb"),
-                model="whisper-1",
+                file=open(file_path, "rb"),
+                model=STT_MODEL,
                 prompt=prompt,
             )
             return transcript.text
@@ -165,7 +167,7 @@ def run_crew(question, file_name):
                 question = f"{question} File data:\n{file.read()}"
     
     initial_answer = crew.kickoff(inputs={"question": question})
-    final_answer = get_final_answer(question, str(initial_answer))
+    final_answer = get_final_answer(FINAL_ANSWER_MODEL, question, str(initial_answer))
 
     print(f"Question: {question}")
     print(f"Initial answer: {initial_answer}")