PRANJAL KAR commited on
Commit ·
3c41c21
1
Parent(s): 8478f4b
Add logging to utils.py for error handling in generate_section_variants function
Browse files
utils.py
CHANGED
|
@@ -7,6 +7,8 @@ from pydub import AudioSegment, silence
|
|
| 7 |
from pydub.effects import low_pass_filter, high_pass_filter
|
| 8 |
from tqdm import tqdm
|
| 9 |
from groq import Groq
|
|
|
|
|
|
|
| 10 |
from dotenv import load_dotenv
|
| 11 |
import random
|
| 12 |
from temp_choose import (
|
|
@@ -22,6 +24,8 @@ from temp_choose import (
|
|
| 22 |
|
| 23 |
load_dotenv()
|
| 24 |
|
|
|
|
|
|
|
| 25 |
|
| 26 |
def make_groq_call(stems, song_name, p, section_type=None, bpm=120, bars=16):
|
| 27 |
"""
|
|
@@ -280,14 +284,18 @@ def generate_section_variants(
|
|
| 280 |
dict: Dictionary of variant audio segments and their descriptions
|
| 281 |
"""
|
| 282 |
stems = get_stems(stems_folder)
|
| 283 |
-
|
| 284 |
-
|
| 285 |
-
|
|
|
|
| 286 |
p,
|
| 287 |
section_type=section_type,
|
| 288 |
-
|
| 289 |
-
|
| 290 |
-
|
|
|
|
|
|
|
|
|
|
| 291 |
|
| 292 |
# Load audio files
|
| 293 |
if not audio_stems:
|
|
|
|
| 7 |
from pydub.effects import low_pass_filter, high_pass_filter
|
| 8 |
from tqdm import tqdm
|
| 9 |
from groq import Groq
|
| 10 |
+
import logging
|
| 11 |
+
|
| 12 |
from dotenv import load_dotenv
|
| 13 |
import random
|
| 14 |
from temp_choose import (
|
|
|
|
| 24 |
|
| 25 |
load_dotenv()
|
| 26 |
|
| 27 |
+
logger = logging.getLogger(__name__)
|
| 28 |
+
logger.setLevel(logging.INFO)
|
| 29 |
|
| 30 |
def make_groq_call(stems, song_name, p, section_type=None, bpm=120, bars=16):
|
| 31 |
"""
|
|
|
|
| 284 |
dict: Dictionary of variant audio segments and their descriptions
|
| 285 |
"""
|
| 286 |
stems = get_stems(stems_folder)
|
| 287 |
+
try:
|
| 288 |
+
llm_response = make_groq_call(
|
| 289 |
+
stems,
|
| 290 |
+
f"{section_type} section",
|
| 291 |
p,
|
| 292 |
section_type=section_type,
|
| 293 |
+
bpm=bpm,
|
| 294 |
+
bars=bars,
|
| 295 |
+
)
|
| 296 |
+
except Exception as e:
|
| 297 |
+
logger.error(f"Error generating variants for {section_type} section: {e}")
|
| 298 |
+
return {}
|
| 299 |
|
| 300 |
# Load audio files
|
| 301 |
if not audio_stems:
|