Update tools/transcribe_audio.py

tools/transcribe_audio.py

@@ -1,4 +1,8 @@
 from langchain_core.tools import tool
+from loguru import logger
+import tempfile
+import whisper
+from tools.load_file import load_file
 
 @tool("transcribe_audio", parse_docstring=True)
 def transcribe_audio(file_name: str) -> str:
@@ -26,10 +30,3 @@ def transcribe_audio(file_name: str) -> str:
     except Exception as e:
         print(f"Error transcribing audio: {str(e)}")
         return ""
-
-
-if __name__ == "__main__":
-    print(transcribe_audio.invoke("../data/99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3.mp3"))
-    print(transcribe_audio.invoke(
-        "https://agents-course-unit4-scoring.hf.space/files/99c9cc74-fdc8-46c6-8f8d-3ce2d3bfeea3"))
-