Create app.py

app.py

@@ -0,0 +1,79 @@
+import time, requests, threading
+from flask import Flask, render_template, request, session
+from flask_socketio import SocketIO, emit
+
+app = Flask(__name__)
+app.config['SECRET_KEY'] = 'suno_studio_secret_2026'
+socketio = SocketIO(app, cors_allowed_origins="*")
+
+# API Base URL
+BASE_URL = "https://api.sunoapi.org/api/v1"
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@socketio.on('save_token')
+def handle_token(data):
+    token = data.get('token')
+    if token:
+        session['api_token'] = token
+        emit('status', {'msg': '🔑 Token saved. Checking credits...'})
+        check_credits(token)
+
+def check_credits(token):
+    try:
+        res = requests.get(f"{BASE_URL}/generate/credit", headers={"Authorization": f"Bearer {token}"})
+        emit('credits', {'val': res.json().get('data', 0)})
+    except:
+        emit('credits', {'val': 'Error'})
+
+@socketio.on('start_gen')
+def handle_gen(payload):
+    token = session.get('api_token')
+    if not token:
+        emit('status', {'msg': '❌ Error: No API Token provided.'})
+        return
+
+    # Prepare Headers & SID
+    headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"}
+    sid = request.sid
+
+    # The payload from the frontend already includes model, style, prompt (lyrics), etc.
+    try:
+        res = requests.post(f"{BASE_URL}/generate/upload-cover", json=payload, headers=headers)
+        res.raise_for_status()
+        task_id = res.json().get('data', {}).get('task_id')
+        
+        if task_id:
+            emit('status', {'msg': f'🚀 Task {task_id} is live! Composing...'})
+            threading.Thread(target=poll_status, args=(task_id, sid, headers)).start()
+        else:
+            emit('status', {'msg': '❌ API Error: Task ID not found in response.'})
+    except Exception as e:
+        emit('status', {'msg': f'❌ Request Failed: {str(e)}'})
+
+def poll_status(task_id, sid, headers):
+    """Background polling for results."""
+    for _ in range(60): # Poll for 10 minutes max
+        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': '✨ Cover Generation Complete!', 
+                    'url': data.get('audio_url'), 
+                    'done': True
+                }, room=sid)
+                return
+            elif status == 'failed':
+                socketio.emit('status', {'msg': '❌ Suno server reported a failure.'}, room=sid)
+                return
+        except: break
+    socketio.emit('status', {'msg': '⏱️ Polling timed out.'}, room=sid)
+
+if __name__ == '__main__':
+    socketio.run(app, debug=True)