Update app.py

app.py

@@ -1,28 +1,30 @@
-import time, requests, threading, os, io
+import eventlet
+eventlet.monkey_patch()  # This MUST be the first line
+
+import os, time, requests, threading, io
 from flask import Flask, render_template, request, session, send_from_directory, jsonify
 from flask_socketio import SocketIO, emit
 
 app = Flask(__name__)
-app.config['SECRET_KEY'] = 'suno_server_direct_2026'
-socketio = SocketIO(app, cors_allowed_origins="*")
+app.config['SECRET_KEY'] = 'suno_final_2026'
+# 'eventlet' is the key to fixing the "Invalid Session" error
+socketio = SocketIO(app, cors_allowed_origins="*", async_mode='eventlet')
 
-# 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
+# This dictionary remembers which browser tab belongs to which song task
 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": "error"}), 400
     
     task_id = data.get('taskId') or data.get('task_id')
     audio_url = data.get('audio_url')
@@ -31,24 +33,23 @@ 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)
+            # Download from Suno to your server
+            r = requests.get(audio_url, timeout=30)
             with open(filepath, 'wb') as f:
                 f.write(r.content)
             
-            # Notify the specific browser tab via SocketIO
+            # Send the download link to the user
             if task_id in task_to_sid:
                 socketio.emit('status', {
-                    'msg': '✅ Success! Saved to server storage.',
+                    'msg': '✨ Success! Music saved to server.',
                     'file_name': filename,
                     'done': True
                 }, room=task_to_sid[task_id])
         except Exception as e:
-            print(f"Callback error: {e}")
+            print(f"Callback Download Error: {e}")
 
-    return jsonify({"status": "received"}), 200
+    return jsonify({"status": "ok"}), 200
 
-# --- SERVE SAVED FILES ---
 @app.route('/get_music/<filename>')
 def get_music(filename):
     return send_from_directory(MUSIC_FOLDER, filename, as_attachment=True)
@@ -56,41 +57,41 @@ def get_music(filename):
 @socketio.on('save_token')
 def save_token(data):
     session['api_token'] = data.get('token')
-    emit('status', {'msg': '🔑 Token Linked.'})
+    emit('status', {'msg': '🔑 Token Linked and Secured.'})
 
 @socketio.on('start_gen')
 def handle_gen(payload):
     token = session.get('api_token')
     if not token:
-        emit('status', {'msg': '❌ Error: Please set API Token.'})
+        emit('status', {'msg': '❌ Error: Please connect your token first.'})
         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:
+    # Fix the "Instrumental cannot be null" error
+    if payload.get('instrumental') is None:
         payload['instrumental'] = False
 
-    # Set dynamic callback
+    # Tell Suno where to send the finished song
     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 FIX 2: Check 'data' before getting taskId
+        # Safe check to prevent NoneType error
         api_data = res_json.get('data')
         if api_data:
             task_id = api_data.get('taskId')
+            # Link this song to the current browser session
             task_to_sid[task_id] = request.sid
-            emit('status', {'msg': f'🚀 Task {task_id} started. Waiting for server download...'})
+            emit('status', {'msg': f'🚀 Task {task_id} started. Don\'t close this tab!'})
         else:
-            error_msg = res_json.get('msg') or "API rejected request"
-            emit('status', {'msg': f'❌ Suno API says: {error_msg}'})
+            msg = res_json.get('msg') or "Unknown API error"
+            emit('status', {'msg': f'❌ Suno Error: {msg}'})
             
     except Exception as e:
-        emit('status', {'msg': f'❌ System Error: {str(e)}'})
+        emit('status', {'msg': f'❌ Connection Error: {str(e)}'})
 
 if __name__ == '__main__':
-    # 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)))
+    socketio.run(app, host="0.0.0.0")