Update app.py

app.py

@@ -6,49 +6,38 @@ app = Flask(__name__)
 app.config['SECRET_KEY'] = 'server_direct_save_2026'
 socketio = SocketIO(app, cors_allowed_origins="*")
 
-# Create a permanent folder on the server for the music
 MUSIC_FOLDER = 'saved_music'
 if not os.path.exists(MUSIC_FOLDER):
     os.makedirs(MUSIC_FOLDER)
 
-# Map to link taskIds to User Session IDs (for the callback)
 task_to_sid = {}
 
 @app.route('/')
 def index():
     return render_template('index.html')
 
-# 1. THE CALLBACK (Saves the file to server storage)
 @app.route('/callback', methods=['POST'])
 def callback():
     data = request.json
+    if not data: return jsonify({"status": "no data"}), 400
+    
+    # Extract taskId and audio_url carefully
     task_id = data.get('taskId') or data.get('task_id')
     audio_url = data.get('audio_url')
 
     if audio_url and task_id:
-        # Download the file to the server immediately
         filename = f"{task_id}.mp3"
         filepath = os.path.join(MUSIC_FOLDER, filename)
-        
         try:
             r = requests.get(audio_url)
             with open(filepath, 'wb') as f:
                 f.write(r.content)
-            
-            # Find the user who owns this task and tell them
             if task_id in task_to_sid:
-                user_sid = task_to_sid[task_id]
-                socketio.emit('status', {
-                    'msg': '✅ Saved to Server!',
-                    'file_name': filename,
-                    'done': True
-                }, room=user_sid)
+                socketio.emit('status', {'msg': '✅ Saved to Server!', 'file_name': filename, 'done': True}, room=task_to_sid[task_id])
         except Exception as e:
-            print(f"Callback save error: {e}")
-
+            print(f"Callback error: {e}")
     return jsonify({"status": "ok"}), 200
 
-# 2. THE DIRECT DOWNLOAD (Serves the file from your server)
 @app.route('/get_music/<filename>')
 def get_music(filename):
     return send_from_directory(MUSIC_FOLDER, filename, as_attachment=True)
@@ -61,21 +50,30 @@ def save_token(data):
 @socketio.on('start_gen')
 def handle_gen(payload):
     token = session.get('api_token')
+    if not token:
+        emit('status', {'msg': '❌ Error: Set Token first.'})
+        return
+
     headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
-    
-    # Set the callback URL dynamically
     payload['callBackUrl'] = request.host_url + "callback"
     
     try:
         res = requests.post("https://api.sunoapi.org/api/v1/generate/upload-cover", json=payload, headers=headers)
-        res_data = res.json()
-        task_id = res_data.get('data', {}).get('taskId')
+        res_json = res.json()
         
-        if task_id:
-            task_to_sid[task_id] = request.sid # Map user to task
-            emit('status', {'msg': f'🚀 Processing Task: {task_id}...'})
+        # SAFETY CHECK: Verify 'data' exists before calling .get()
+        api_data = res_json.get('data')
+        if api_data:
+            task_id = api_data.get('taskId')
+            task_to_sid[task_id] = request.sid
+            emit('status', {'msg': f'🚀 Task {task_id} started...'})
+        else:
+            # If 'data' is missing, show the actual error from Suno
+            error_msg = res_json.get('msg') or "Unknown API Error"
+            emit('status', {'msg': f'❌ Suno API says: {error_msg}'})
+            
     except Exception as e:
-        emit('status', {'msg': f'❌ Error: {str(e)}'})
+        emit('status', {'msg': f'❌ Script Error: {str(e)}'})
 
 if __name__ == '__main__':
-    socketio.run(app, host="0.0.0.0")
+    socketio.run(app, host="0.0.0.0", port=5000)