enhance generate_ai_midi to increase loop length, improve deep copy handling, and refine audio filtering parameters

main.py

@@ -4,6 +4,7 @@ from fastapi.responses import FileResponse
 from numpy.ma import copy
 from pydantic import BaseModel
 from pydub import AudioSegment
+import copy
 import torch
 import torch.nn as nn
 import pickle
@@ -223,14 +224,25 @@ def generate_ai_midi(bpm: int, output_file: str, num_notes: int = 200):
     inst.midiChannel = 0
     final_stream.insert(0, inst)
 
+    # ===== LOOP để dài hơn =====
+    final_stream = stream.Stream()
+
+    inst = instrument.Piano()
+    inst.midiProgram = 0
+    inst.midiChannel = 0
+    final_stream.insert(0, inst)
+
     LOOP = 4  # 🔥 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):
         for el in output_stream:
             try:
-                new_el = copy.deepcopy(el)  # ✅ FIX
-                new_el.offset = el.offset + i * offset
+                new_el = copy.deepcopy(el)  # Bây giờ hàm này đã hoạt động
+                new_el.offset = el.offset + (i * total_length) # Dùng total_length để an toàn
                 final_stream.append(new_el)
-            except:
+            except Exception as e:
+                print(f"Lỗi khi copy phần tử: {e}")
                 continue
 
     final_stream.write('midi', fp=output_file)
@@ -242,12 +254,21 @@ def generate_ai_midi(bpm: int, output_file: str, num_notes: int = 200):
 def apply_audio_filters(wav_path):
     sound = AudioSegment.from_wav(wav_path)
     
-    # Cắt toàn bộ tần số trên 600Hz (Đây là mức "Dark/Warm" - cực kỳ ấm áp)
-    # 600Hz sẽ loại bỏ hoàn toàn tiếng chói, chỉ để lại tiếng piano mờ ảo.
+    # Cắt toàn bộ tần số trên 1500Hz để lấy độ mờ ảo
     sound = sound.low_pass_filter(1500) 
     
-    # Làm mượt đầu cuối
-    sound = sound.fade_in(2000).fade_out(3000)
+    # Tính toán thời gian fade linh hoạt (Max là 2000ms in / 3000ms out)
+    # Nếu file nhạc ngắn hơn 5 giây, chỉ fade 10% thời lượng bài nhạc
+    audio_duration = len(sound)
+    fade_in_time = min(2000, int(audio_duration * 0.1))
+    fade_out_time = min(3000, int(audio_duration * 0.1))
+    
+    sound = sound.fade_in(fade_in_time).fade_out(fade_out_time)
+    
+    # Kéo volume lên một chút (tăng 3dB) để đảm bảo nghe rõ hơn sau khi lọc
+    sound = sound + 3
+    
+    sound.export(wav_path, format="wav")
     
     sound.export(wav_path, format="wav")
 @app.post("/generate-vibe")