reduce loop length in generate_ai_midi function to enhance performance

index.html

@@ -88,132 +88,130 @@
         <div id="status">Chưa kết nối... Mở nRF Connect trên điện thoại để chuẩn bị!</div>
     </div>
 
-    <script>
+   <script>
         let currentBpm = 0;
-        let currentAudio = null; // Biến toàn cục để quản lý file âm thanh
+        let detectedCharUUID = null;
+        let bpmHistory = [];
 
         const bpmDisplay = document.getElementById('bpm-display');
         const statusText = document.getElementById('status');
         const btnConnect = document.getElementById('btn-connect');
-        const btnGenerate = document.getElementById('btn-generate');
-        const btnStop = document.getElementById('btn-stop');
 
-        // ==========================================
-        // 1. BLUETOOTH LOGIC
-        // ==========================================
+        // ================================
+        // HELPER: SMOOTH BPM
+        // ================================
+        function smoothBpm(newVal) {
+            bpmHistory.push(newVal);
+            if (bpmHistory.length > 5) bpmHistory.shift();
+
+            return Math.round(
+                bpmHistory.reduce((a, b) => a + b, 0) / bpmHistory.length
+            );
+        }
+
+        // ================================
+        // HELPER: PARSE DATA
+        // ================================
+        function handleAutoData(event, uuid) {
+            const value = event.target.value;
+
+            let raw = [];
+            for (let i = 0; i < value.byteLength; i++) {
+                raw.push(value.getUint8(i));
+            }
+
+            console.log("📡 UUID:", uuid, "DATA:", raw);
+
+            // ===== AUTO DETECT BPM =====
+            for (let i = 0; i < raw.length; i++) {
+                let v = raw[i];
+
+                // heuristic: BPM range
+                if (v >= 40 && v <= 180) {
+
+                    let newBpm = smoothBpm(v);
+
+                    // filter noise (tránh nhảy lung tung)
+                    if (currentBpm !== 0 && Math.abs(newBpm - currentBpm) > 40) {
+                        continue;
+                    }
+
+                    currentBpm = newBpm;
+
+                    bpmDisplay.innerText = `${currentBpm} BPM`;
+                    bpmDisplay.style.color = "#e94560";
+
+                    statusText.innerText = `Đã detect BPM từ UUID: ${uuid}`;
+                    statusText.style.color = "#4ecca3";
+
+                    detectedCharUUID = uuid;
+
+                    break;
+                }
+            }
+        }
+
+        // ================================
+        // MAIN: AUTO SCAN BLE
+        // ================================
         btnConnect.addEventListener('click', async () => {
             try {
-                statusText.innerText = "Đang mở máy quét Bluetooth...";
+                statusText.innerText = "Đang mở BLE scanner (auto)...";
+
                 const device = await navigator.bluetooth.requestDevice({
                     acceptAllDevices: true,
-                    optionalServices: ['heart_rate']
+                    optionalServices: [
+                        'battery_service',
+                        'device_information',
+                        'heart_rate'
+                    ]
                 });
 
-                statusText.innerText = `Đã chọn: ${device.name}. Đang kết nối...`;
+                statusText.innerText = `Đã chọn: ${device.name || "Unknown"} → đang connect...`;
+
                 const server = await device.gatt.connect();
-                const service = await server.getPrimaryService('heart_rate');
-                const characteristic = await service.getCharacteristic('heart_rate_measurement'); 
-
-                await characteristic.startNotifications();
-                characteristic.addEventListener('characteristicvaluechanged', handleHeartRateData);
-                
-                statusText.innerText = "Kết nối thành công! Đang đợi nhịp tim truyền lên...";
-                btnConnect.innerText = "✓ Đã kết nối Bluetooth";
-                btnConnect.style.backgroundColor = "#4ecca3";
-                btnConnect.style.color = "#1a1a2e";
-            } catch (error) {
-                console.error("Bluetooth Error:", error);
-                statusText.innerText = "Hủy kết nối hoặc lỗi thiết bị!";
-                statusText.style.color = "#ff4d4d";
-            }
-        });
 
-        function handleHeartRateData(event) {
-            const value = event.target.value;
-            const flags = value.getUint8(0);
-            
-            if ((flags & 0x01) === 0) {
-                currentBpm = value.getUint8(1);
-            } else {
-                currentBpm = value.getUint16(1, true);
-            }
-            
-            bpmDisplay.innerText = `${currentBpm} BPM`;
-            bpmDisplay.style.color = "#e94560"; 
-            statusText.innerText = "Đang đọc nhịp tim thành công. Sẵn sàng tạo nhạc!";
-            statusText.style.color = "#4ecca3";
-        }
+                const services = await server.getPrimaryServices();
 
-        // ==========================================
-        // 2. TẠO VÀ PHÁT NHẠC
-        // ==========================================
-        btnGenerate.addEventListener('click', async () => {
-            if (currentBpm === 0) {
-                alert("Bạn chưa kết nối Bluetooth hoặc chưa gửi số nhịp tim từ nRF Connect!");
-                return;
-            }
+                statusText.innerText = "Đang scan toàn bộ services... (check console)";
 
-            // TÍNH NĂNG MỚI: Tắt ngay bài nhạc cũ (nếu có) trước khi tạo bài mới
-            if (currentAudio) {
-                currentAudio.pause();
-                currentAudio.currentTime = 0; // Tua lại từ đầu
-                bpmDisplay.style.animation = "none";
-            }
+                for (const service of services) {
+                    console.log("🔵 Service:", service.uuid);
 
-            statusText.innerText = `Đang gửi ${currentBpm} BPM cho bộ não AI xử lý...`;
-            btnGenerate.innerText = "AI đang sáng tác...";
-            btnGenerate.disabled = true; 
-            btnStop.disabled = true;
+                    const characteristics = await service.getCharacteristics();
 
-            try {
-                const response = await fetch('/generate-vibe', {
-                    method: 'POST',
-                    headers: { 'Content-Type': 'application/json' },
-                    body: JSON.stringify({ bpm: currentBpm })
-                });
-                
-                const data = await response.json();
-
-                if (data.status === "success") {
-                    statusText.innerText = "Sáng tác xong! Đang phát nhạc 🎵";
-                    
-                    // Gắn file nhạc mới vào biến toàn cục
-                    currentAudio = new Audio(data.audio_url);
-                    currentAudio.play();
-
-                    bpmDisplay.style.animation = "pulse 1s infinite";
-                    btnStop.disabled = false; // Mở khóa nút Dừng
-                    
-                    // Xử lý khi nhạc tự chạy hết
-                    currentAudio.onended = () => {
-                        bpmDisplay.style.animation = "none";
-                        statusText.innerText = "Bản nhạc đã kết thúc. Sẵn sàng tạo trải nghiệm mới!";
-                        btnStop.disabled = true;
-                    };
-                } else {
-                    statusText.innerText = `Lỗi từ Server: ${data.message}`;
+                    for (const char of characteristics) {
+                        console.log("   ⚪ Char:", char.uuid, char.properties);
+
+                        // chỉ lấy char có notify / indicate
+                        if (char.properties.notify || char.properties.indicate) {
+                            try {
+                                await char.startNotifications();
+
+                                char.addEventListener(
+                                    'characteristicvaluechanged',
+                                    (event) => handleAutoData(event, char.uuid)
+                                );
+
+                                console.log("   ✅ Subscribed:", char.uuid);
+                            } catch (err) {
+                                console.log("   ❌ Cannot subscribe:", char.uuid);
+                            }
+                        }
+                    }
                 }
-            } catch (error) {
-                console.error("API Error:", error);
-                statusText.innerText = "Mất kết nối với máy chủ AI!";
-            } finally {
-                btnGenerate.innerText = "2. Sinh nhạc AI";
-                btnGenerate.disabled = false;
-            }
-        });
 
-        // ==========================================
-        // 3. DỪNG NHẠC THỦ CÔNG
-        // ==========================================
-        btnStop.addEventListener('click', () => {
-            if (currentAudio) {
-                currentAudio.pause(); // Cắt âm thanh
-                currentAudio.currentTime = 0; // Xóa cache phát
-                bpmDisplay.style.animation = "none"; // Tắt nhấp nháy
-                statusText.innerText = "Đã dừng phát nhạc.";
-                btnStop.disabled = true; // Khóa lại nút dừng
+                statusText.innerText = "Đã subscribe toàn bộ notify characteristics. Đang chờ BPM...";
+                btnConnect.innerText = "✓ BLE Connected (Auto Mode)";
+                btnConnect.style.backgroundColor = "#4ecca3";
+                btnConnect.style.color = "#1a1a2e";
+
+            } catch (error) {
+                console.error("❌ BLE ERROR:", error);
+                statusText.innerText = "Lỗi BLE hoặc user cancel!";
+                statusText.style.color = "#ff4d4d";
             }
         });
-    </script>
+</script>
 </body>
 </html>

main.py

@@ -231,7 +231,7 @@ def generate_ai_midi(bpm: int, output_file: str, num_notes: int = 200):
     inst.midiChannel = 0
     final_stream.insert(0, inst)
 
-    LOOP = 4  # 🔥 tăng độ dài nhạc
+    LOOP = 2  # 🔥 tăng độ dài nhạc
     total_length = offset # Lưu lại tổng chiều dài của 1 loop
     
     for i in range(LOOP):