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
Files changed (1) hide show
  1. utils.py +14 -6
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
- llm_response = make_groq_call(
284
- stems,
285
- f"{section_type} section",
 
286
  p,
287
  section_type=section_type,
288
- bpm=bpm,
289
- bars=bars,
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: