Update app.py

app.py

@@ -3,25 +3,27 @@ from flask import Flask, render_template, request, session, send_from_directory,
 from flask_socketio import SocketIO, emit
 
 app = Flask(__name__)
-app.config['SECRET_KEY'] = 'server_direct_save_2026'
+app.config['SECRET_KEY'] = 'suno_server_direct_2026'
 socketio = SocketIO(app, cors_allowed_origins="*")
 
+# Create folder for persistent storage
 MUSIC_FOLDER = 'saved_music'
 if not os.path.exists(MUSIC_FOLDER):
     os.makedirs(MUSIC_FOLDER)
 
+# Map taskId to the specific user's SID
 task_to_sid = {}
 
 @app.route('/')
 def index():
     return render_template('index.html')
 
+# --- WEBHOOK CALLBACK ---
 @app.route('/callback', methods=['POST'])
 def callback():
     data = request.json
-    if not data: return jsonify({"status": "no data"}), 400
+    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')
 
@@ -29,15 +31,24 @@ def callback():
         filename = f"{task_id}.mp3"
         filepath = os.path.join(MUSIC_FOLDER, filename)
         try:
+            # Download file from Suno to our server
             r = requests.get(audio_url)
             with open(filepath, 'wb') as f:
                 f.write(r.content)
+            
+            # Notify the specific browser tab via SocketIO
             if task_id in task_to_sid:
-                socketio.emit('status', {'msg': '✅ Saved to Server!', 'file_name': filename, 'done': True}, room=task_to_sid[task_id])
+                socketio.emit('status', {
+                    'msg': '✅ Success! Saved to server storage.',
+                    'file_name': filename,
+                    'done': True
+                }, room=task_to_sid[task_id])
         except Exception as e:
             print(f"Callback error: {e}")
-    return jsonify({"status": "ok"}), 200
 
+    return jsonify({"status": "received"}), 200
+
+# --- SERVE SAVED FILES ---
 @app.route('/get_music/<filename>')
 def get_music(filename):
     return send_from_directory(MUSIC_FOLDER, filename, as_attachment=True)
@@ -51,29 +62,35 @@ def save_token(data):
 def handle_gen(payload):
     token = session.get('api_token')
     if not token:
-        emit('status', {'msg': '❌ Error: Set Token first.'})
+        emit('status', {'msg': '❌ Error: Please set API Token.'})
         return
 
     headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
+    
+    # SAFETY FIX 1: Ensure instrumental is never null
+    if 'instrumental' not in payload or payload['instrumental'] is None:
+        payload['instrumental'] = False
+
+    # Set dynamic callback
     payload['callBackUrl'] = request.host_url + "callback"
     
     try:
         res = requests.post("https://api.sunoapi.org/api/v1/generate/upload-cover", json=payload, headers=headers)
         res_json = res.json()
         
-        # SAFETY CHECK: Verify 'data' exists before calling .get()
+        # SAFETY FIX 2: Check 'data' before getting taskId
         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...'})
+            emit('status', {'msg': f'🚀 Task {task_id} started. Waiting for server download...'})
         else:
-            # If 'data' is missing, show the actual error from Suno
-            error_msg = res_json.get('msg') or "Unknown API Error"
+            error_msg = res_json.get('msg') or "API rejected request"
             emit('status', {'msg': f'❌ Suno API says: {error_msg}'})
             
     except Exception as e:
-        emit('status', {'msg': f'❌ Script Error: {str(e)}'})
+        emit('status', {'msg': f'❌ System Error: {str(e)}'})
 
 if __name__ == '__main__':
-    socketio.run(app, host="0.0.0.0", port=5000)
+    # Use port 7860 for Hugging Face or 5000 for local/server
+    socketio.run(app, host="0.0.0.0", port=int(os.environ.get("PORT", 5000)))