FEAT: Added Tracing for user selections

app.py

@@ -16,6 +16,7 @@ from utils import (
 )
 
 from dotenv import load_dotenv
+from langsmith import traceable
 
 load_dotenv()
 
@@ -153,6 +154,7 @@ def generate_section_variants_handler(
         return f"Error generating variants: {str(e)}", None, None, None, None, None, None
 
 
+@traceable(run_type="chain", name="groq_call")
 def select_variant(section_type, variant_num, append):
     """Select a variant for a specific section, with an option to append."""
     global ALL_VARIANTS, SELECTED_VARIANTS
@@ -166,17 +168,23 @@ def select_variant(section_type, variant_num, append):
             return f"Variant {variant_num} not found for {section_type}", "None"
 
         # Store the selection
-        SELECTED_VARIANTS[section_type] = ALL_VARIANTS[section_type][variant_key]["config"]
+        variant_info = ALL_VARIANTS[section_type][variant_key]
+        SELECTED_VARIANTS[section_type] = variant_info["config"]
         
         # Create a display string for the selected variant
-        variant_info = ALL_VARIANTS[section_type][variant_key]
         display_text = f"Selected: Variant {variant_num}"
         if "description" in variant_info:
             display_text += f" - {variant_info['description']}"
 
+        # Log selection details for tracing
+        logger.info(f"Selected variant for {section_type}: {variant_key}")
+        logger.info(f"Variant description: {variant_info.get('description', 'No description')}")
+        logger.info(f"Stems used: {variant_info.get('config', {}).get('stems', [])}")
+
         return f"Selected variant {variant_num} for {section_type}", display_text
 
     except Exception as e:
+        logger.error(f"Error selecting variant: {str(e)}")
         return f"Error selecting variant: {str(e)}", "None"
 
 
@@ -447,6 +455,7 @@ with gr.Blocks(title="Interactive Music Track Generator") as demo:
         
         edm_arrangement_tab()
 
+
     with gr.Tab("2. Generate Section Variants"):
         gr.Markdown("### Generate and Select Variants for Each Section")
         gr.Markdown("Generate variants for each section and select which one to use in the final track")

requirements.txt

@@ -8,3 +8,5 @@ python-dotenv
 plotly-express
 pandas
 pydub
+langsmith
+langchain

utils.py

@@ -8,6 +8,8 @@ from pydub.effects import low_pass_filter, high_pass_filter
 from tqdm import tqdm
 from groq import Groq
 import logging
+from langsmith import traceable
+from langchain.callbacks import LangChainTracer
 
 from dotenv import load_dotenv
 import random
@@ -20,6 +22,8 @@ logger.setLevel(logging.INFO)
 logger.addHandler(logging.StreamHandler())
 logger.addHandler(logging.FileHandler("log.txt"))
 
+
+@traceable(run_type="llm", name="groq_call")
 def make_groq_call(stems, song_name, p, section_type=None, bpm=120, bars=16):
     """
     Make a call to the Groq API to get music production instructions.