Update audio_transcriber.py

audio_transcriber.py

@@ -1,28 +1,35 @@
 import os
+import requests
 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": {
+        "url": {
             "type": "string",
-            "description": "Path to the audio file (must be .mp3 or .wav)"
+            "description": "URL to the audio file (.mp3 or .wav)"
         }
     }
     output_type = "string"
 
-    def forward(self, file_path: str) -> str:
+    def forward(self, url: str) -> str:
         try:
-            with open(file_path, "rb") as audio_file:
+            # Download audio
+            filename = "/tmp/audio_input.mp3"
+            response = requests.get(url)
+            with open(filename, "wb") as f:
+                f.write(response.content)
+
+            # Transcribe with Whisper
+            with open(filename, "rb") as audio_file:
                 transcript = openai.audio.transcriptions.create(
                     model="whisper-1",
                     file=audio_file
                 )
-            return transcript.text
+            return transcript.text.strip()
         except Exception as e:
             return f"Error transcribing audio: {e}"