fix chord mapping file error add sys and shutils

main.py

@@ -17,7 +17,8 @@ import os
 import tempfile
 from huggingface_hub import login
 from pathlib import Path
-
+import sys
+import shutil
 
 title = """# 🙋🏻‍♂️Welcome to 🌟Tonic's 🎼Jasco🎶AudioCraft Demo"""
 description = """Facebook presents JASCO, a temporally controlled text-to-music generation model utilizing both symbolic and audio-based conditions. JASCO can generate high-quality music samples conditioned on global text descriptions along with fine-grained local controls. JASCO is based on the Flow Matching modeling paradigm together with a novel conditioning method, allowing for music generation controlled both locally (e.g., chords) and globally (text description). [run this demo locally](https://huggingface.co/spaces/Tonic/audiocraft?docker=true) or [embed this space](https://huggingface.co/spaces/Tonic/audiocraft?embed=true) or [duplicate this space](https://huggingface.co/spaces/Tonic/audiocraft?duplicate=true) to run it privately . you can also use this demo via API by clicking the link at the bottom of the page."""
@@ -190,6 +191,8 @@ Model: facebook/jasco-chords-drums-1B
 - Melody-enabled models may be slower
 """
 
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
 MODEL = None
 INTERRUPTING = False
 
@@ -213,12 +216,29 @@ for subdir in ['models', 'cache', 'huggingface', 'drum_cache', 'transformers']:
     os.makedirs(os.path.join(CACHE_DIR, subdir), exist_ok=True)
 
 def cleanup_cache():
-    """Clean up temporary cache files"""
+    """Clean up temporary cache files while preserving chord mapping"""
     try:
-        import shutil
+        # Save the chord mapping if it exists
+        mapping_file = os.path.join(CACHE_DIR, "chord_to_index_mapping.pkl")
+        temp_mapping = None
+        if os.path.exists(mapping_file):
+            with open(mapping_file, 'rb') as f:
+                temp_mapping = pickle.load(f)
+        
+        # Clean the cache directory
         if os.path.exists(CACHE_DIR):
             shutil.rmtree(CACHE_DIR)
+        
+        # Recreate necessary directories
         os.makedirs(CACHE_DIR, exist_ok=True)
+        for subdir in ['models', 'cache', 'huggingface', 'drum_cache', 'transformers']:
+            os.makedirs(os.path.join(CACHE_DIR, subdir), exist_ok=True)
+        
+        # Restore the chord mapping
+        if temp_mapping is not None:
+            with open(mapping_file, 'wb') as f:
+                pickle.dump(temp_mapping, f)
+                
     except Exception as e:
         print(f"Error cleaning cache: {e}")
 
@@ -358,6 +378,26 @@ def patch_jasco_cache():
 # Apply the patch
 patch_jasco_cache()
 
+def ensure_chord_mapping():
+    """Ensure chord mapping file exists and is properly located"""
+    # Create mapping in the cache directory
+    cache_mapping_path = os.path.join(CACHE_DIR, "chord_to_index_mapping.pkl")
+    
+    # Create mapping in the models directory
+    models_dir = os.path.join(os.path.dirname(__file__), "models")
+    os.makedirs(models_dir, exist_ok=True)
+    models_mapping_path = os.path.join(models_dir, "chord_to_index_mapping.pkl")
+    
+    # Create the mapping if it doesn't exist in either location
+    chord_mapping = create_default_chord_mapping()
+    
+    # Save to both locations
+    for mapping_path in [cache_mapping_path, models_mapping_path]:
+        with open(mapping_path, "wb") as f:
+            pickle.dump(chord_mapping, f)
+    
+    return models_mapping_path
+
 # Create necessary directories
 os.makedirs("models", exist_ok=True)
 
@@ -402,6 +442,12 @@ def load_model(version='facebook/jasco-chords-drums-melody-400M'):
             raise gr.Error("Failed to load model")
     return MODEL
 
+# Initialize chord mapping at startup
+mapping_file = ensure_chord_mapping()
+os.environ['AUDIOCRAFT_CHORD_MAPPING'] = mapping_file
+print(f"Initial chord mapping file location: {mapping_file}")
+print(f"Chord mapping exists: {os.path.exists(mapping_file)}")
+
 class ModelLoadingError(Exception):
     pass