Update app.py

app.py

@@ -3,7 +3,6 @@ import requests
 from pydub import AudioSegment
 import os
 import tempfile
-import json
 
 app = Flask(__name__)
 
@@ -27,12 +26,18 @@ def home():
         <h3>Usage</h3>
         <p>Send a POST request to <code>/get_audio_duration</code> with JSON like:</p>
         <pre>
-[{
-  "data": "{\"audio\":{\"url\":\"https://example.com/audio.mp3\",\"content_type\":\"audio/mpeg\",\"file_name\":\"audio.mp3\",\"file_size\":12345},\"seed\":123}"
-}]
+{
+  "audio": {
+    "url": "https://example.com/audio.mp3",
+    "content_type": "audio/mpeg",
+    "file_name": "audio.mp3",
+    "file_size": 12345
+  },
+  "seed": 123
+}
         </pre>
         <p>Example response: <code>{"duration_seconds": 10.5}</code></p>
-        <p>Test it with curl: <code>curl -X POST https://eldhos98-audio-duration-n8n.hf.space/get_audio_duration -H "Content-Type: application/json" -d '[{\"data\":\"{\\\"audio\\\":{\\\"url\\\":\\\"https://v3.fal.media/files/kangaroo/dPK1FYLC1TNxY5W4QO0AO_output.mp3\\\",\\\"content_type\\\":\\\"audio/mpeg\\\",\\\"file_name\\\":\\\"output.mp3\\\",\\\"file_size\\\":426781},\\\"seed\\\":1322097652}\"}]'</code></p>
+        <p>Test it with curl: <code>curl -X POST https://eldhos98-audio-duration-n8n.hf.space/get_audio_duration -H "Content-Type: application/json" -d '{"audio":{"url":"https://v3.fal.media/files/kangaroo/dPK1FYLC1TNxY5W4QO0AO_output.mp3","content_type":"audio/mpeg","file_name":"output.mp3","file_size":426781},"seed":1322097652}'</code></p>
     </body>
     </html>
     """
@@ -42,24 +47,10 @@ def home():
 @app.route('/get_audio_duration', methods=['POST'])
 def get_audio_duration():
     try:
-        # Get JSON input (expecting an array with a single object)
+        # Get JSON input
         data = request.get_json()
-        if not isinstance(data, list) or len(data) == 0:
-            return jsonify({"error": "Input must be a non-empty array"}), 400
+        audio_url = data.get('audio', {}).get('url')
         
-        # Extract the 'data' field from the first object
-        data_str = data[0].get('data')
-        if not data_str:
-            return jsonify({"error": "No 'data' field provided in the input"}), 400
-
-        # Parse the stringified JSON in 'data'
-        try:
-            inner_data = json.loads(data_str)
-        except json.JSONDecodeError:
-            return jsonify({"error": "Invalid JSON in 'data' field"}), 400
-
-        # Extract the audio URL
-        audio_url = inner_data.get('audio', {}).get('url')
         if not audio_url:
             return jsonify({"error": "No audio URL provided"}), 400
 
@@ -68,7 +59,7 @@ def get_audio_duration():
             # Download the audio file
             response = requests.get(audio_url)
             if response.status_code != 200:
-                return jsonify({"error": "Failed to download audio file"}), 400
+                return jupytext({"error": "Failed to download audio file"}), 400
             
             # Write audio content to temporary file
             temp_file.write(response.content)