Update app.py

app.py

@@ -1,9 +1,9 @@
-import time, requests, threading
-from flask import Flask, render_template, request, session
+import time, requests, threading, io
+from flask import Flask, render_template, request, session, send_file
 from flask_socketio import SocketIO, emit
 
 app = Flask(__name__)
-app.config['SECRET_KEY'] = 'suno_studio_2026_secret'
+app.config['SECRET_KEY'] = 'suno_studio_full_2026'
 socketio = SocketIO(app, cors_allowed_origins="*")
 
 BASE_URL = "https://api.sunoapi.org/api/v1"
@@ -12,69 +12,66 @@ BASE_URL = "https://api.sunoapi.org/api/v1"
 def index():
     return render_template('index.html')
 
+# --- NEW: CALLBACK ROUTE ---
+@app.route('/callback', methods=['POST'])
+def suno_callback():
+    """Suno calls this endpoint when the generation is finished."""
+    data = request.json
+    # The taskId and status are sent here
+    task_id = data.get('taskId')
+    status = data.get('status')
+    audio_url = data.get('audio_url')
+
+    print(f"Callback received for {task_id}: {status}")
+
+    # Alert the specific user via SocketIO
+    socketio.emit('status', {
+        'msg': '✨ Callback Received: Track Ready!',
+        'url': audio_url,
+        'done': (status == 'complete')
+    })
+    return {"status": "success"}, 200
+
+# --- PROXY DOWNLOAD (Fixes your download issue) ---
+@app.route('/download_proxy')
+def download_proxy():
+    url = request.args.get('url')
+    title = request.args.get('title', 'suno_track').replace(" ", "_")
+    try:
+        res = requests.get(url)
+        return send_file(io.BytesIO(res.content), mimetype="audio/mpeg", as_attachment=True, download_name=f"{title}.mp3")
+    except:
+        return "Download Failed", 500
+
 @socketio.on('save_token')
 def handle_token(data):
-    token = data.get('token')
-    if token:
-        session['api_token'] = token
-        emit('status', {'msg': '🔑 Token saved. Checking balance...'})
-        try:
-            res = requests.get(f"{BASE_URL}/generate/credit", headers={"Authorization": f"Bearer {token}"})
-            balance = res.json().get('data', 0)
-            emit('credits', {'val': balance})
-        except:
-            emit('credits', {'val': 'Error'})
+    session['api_token'] = data.get('token')
+    emit('status', {'msg': '🔑 Token connected.'})
 
 @socketio.on('start_gen')
 def handle_gen(payload):
     token = session.get('api_token')
     if not token:
-        emit('status', {'msg': '❌ Error: Connect your API Token first.'})
+        emit('status', {'msg': '❌ Error: No Token.'})
         return
 
-    # Enforce model-specific rules from docs
-    if payload.get('model') == 'V4_5ALL':
-        emit('status', {'msg': '⚠️ V4_5ALL detected: Audio must be < 1 min.'})
-
     headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
     
+    # Note: Replace this with your actual public URL (e.g., from Ngrok)
+    # If this is not a real public URL, Suno will fail to send the callback.
+    payload['callBackUrl'] = "https://zenjoul80-suno-api.hf.space/callback"
+
     try:
-        # Standard API call as per docs
         res = requests.post(f"{BASE_URL}/generate/upload-cover", json=payload, headers=headers)
         res_data = res.json()
         
-        if res.status_code == 200 and res_data.get('data'):
-            task_id = res_data['data'].get('taskId') # Fixed key from docs: 'taskId'
-            emit('status', {'msg': f'🚀 Task Started! ID: {task_id}'})
-            threading.Thread(target=poll_status, args=(task_id, request.sid, headers)).start()
+        if res.status_code == 200:
+            task_id = res_data.get('data', {}).get('taskId')
+            emit('status', {'msg': f'🚀 Task {task_id} sent. Waiting for Callback...'})
         else:
-            msg = res_data.get('msg', 'Unknown Error')
-            emit('status', {'msg': f'❌ API Error: {msg}'})
-            
+            emit('status', {'msg': f"❌ Error: {res_data.get('msg')}"})
     except Exception as e:
-        emit('status', {'msg': f'❌ Script Error: {str(e)}'})
-
-def poll_status(task_id, sid, headers):
-    # Poll for 10 mins max
-    for _ in range(60):
-        time.sleep(10)
-        try:
-            res = requests.get(f"{BASE_URL}/suno/cover/record-info?taskId={task_id}", headers=headers)
-            data = res.json().get('data', {})
-            status = data.get('status')
-            
-            if status == 'complete':
-                socketio.emit('status', {
-                    'msg': '✨ Success! Your cover is ready.',
-                    'url': data.get('audio_url'),
-                    'done': True
-                }, room=sid)
-                return
-            elif status == 'failed':
-                socketio.emit('status', {'msg': '❌ Generation failed on server.'}, room=sid)
-                return
-        except: break
-    socketio.emit('status', {'msg': '⏱️ Task timed out.'}, room=sid)
+        emit('status', {'msg': f"❌ Failed: {str(e)}"})
 
 if __name__ == '__main__':
     socketio.run(app, host="0.0.0.0", debug=True)