Update app.py

app.py

@@ -5,9 +5,80 @@ import os
 import random
 from werkzeug.utils import secure_filename
 
+os.environ['MODEL_PATH'] = '/tmp/pretrained_models'
+
 app = Flask(__name__)
 app.config['MAX_CONTENT_LENGTH'] = 100 * 1024 * 1024
 
+@app.route('/upload', methods=['GET'])
+def upload_and_split():
+    separator = None
+    input_file = None
+    base_dir = None
+    
+    try:
+        audio_url = request.args.get('url')
+        if not audio_url:
+            return jsonify({'error': 'URL parameter required'}), 400
+
+        random_id = random.randint(1000, 9999)
+        input_file = f'/tmp/input_{random_id}.mp3'
+        output_dir = '/tmp'
+
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+        }
+        
+        response = requests.get(audio_url, timeout=60, headers=headers, allow_redirects=True, stream=True)
+        response.raise_for_status()
+        
+        with open(input_file, 'wb') as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+
+        separator = Separator('spleeter:2stems')
+        separator.separate_to_file(input_file, output_dir)
+
+        base_name = f'input_{random_id}'
+        base_dir = f'/tmp/{base_name}'
+        vocal_src = f'{base_dir}/vocals.wav'
+        instrumental_src = f'{base_dir}/accompaniment.wav'
+        
+        vocal_dest = f'/tmp/vocal_{random_id}.mp3'
+        instrumental_dest = f'/tmp/instrumental_{random_id}.mp3'
+
+        os.system(f'ffmpeg -i "{vocal_src}" -b:a 192k "{vocal_dest}" -y -loglevel quiet')
+        os.system(f'ffmpeg -i "{instrumental_src}" -b:a 192k "{instrumental_dest}" -y -loglevel quiet')
+
+        return jsonify({
+            'success': True,
+            'message': 'Audio processed successfully',
+            'vocal': vocal_dest,
+            'instrumental': instrumental_dest,
+            'id': random_id,
+            'download_vocal': f'/download/vocal_{random_id}.mp3',
+            'download_instrumental': f'/download/instrumental_{random_id}.mp3'
+        })
+
+    except requests.exceptions.RequestException as e:
+        return jsonify({'error': f'Failed to download audio: {str(e)}'}), 400
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+    
+    finally:
+        if separator:
+            del separator
+        if input_file and os.path.exists(input_file):
+            try:
+                os.remove(input_file)
+            except:
+                pass
+        if base_dir and os.path.exists(base_dir):
+            try:
+                os.system(f'rm -rf {base_dir}')
+            except:
+                pass
+
 @app.route('/vocal', methods=['GET', 'POST'])
 def split_vocal():
     separator = None
@@ -102,14 +173,16 @@ def index():
     return jsonify({
         'message': 'Audio Splitter API',
         'endpoints': {
-            'GET /vocal?url=<audio_url>': 'Split audio from URL',
+            'GET /upload?url=<audio_url>': 'Download from URL, split, then delete source',
+            'GET /vocal?url=<audio_url>': 'Split audio from URL (legacy)',
             'POST /vocal': 'Upload and split audio file (form-data with key "file")',
             'GET /download/<filename>': 'Download separated files',
             'GET /health': 'Health check'
         },
-        'example_curl': {
-            'url': 'curl "https://herzaj-spleeter.hf.space/vocal?url=YOUR_AUDIO_URL"',
-            'upload': 'curl -X POST -F "file=@audio.mp3" https://herzaj-spleeter.hf.space/vocal'
+        'example': {
+            'upload': 'curl "https://herzaj-spleeter.hf.space/upload?url=YOUR_AUDIO_URL"',
+            'vocal_url': 'curl "https://herzaj-spleeter.hf.space/vocal?url=YOUR_AUDIO_URL"',
+            'vocal_file': 'curl -X POST -F "file=@audio.mp3" https://herzaj-spleeter.hf.space/vocal'
         }
     })