Create audio_transcriber.py

audio_transcriber.py

@@ -0,0 +1,28 @@
+import os
+import openai
+from smolagents import Tool
+
+# Load OpenAI API key from environment
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+class AudioTranscriptionTool(Tool):
+    name = "audio_transcriber"
+    description = "Transcribe a given audio file in mp3 or wav format to text using Whisper."
+    inputs = {
+        "file_path": {
+            "type": "string",
+            "description": "Path to the audio file (must be .mp3 or .wav)"
+        }
+    }
+    output_type = "string"
+
+    def forward(self, file_path: str) -> str:
+        try:
+            with open(file_path, "rb") as audio_file:
+                transcript = openai.audio.transcriptions.create(
+                    model="whisper-1",
+                    file=audio_file
+                )
+            return transcript.text
+        except Exception as e:
+            return f"Error transcribing audio: {e}"