Update app.py

app.py

@@ -169,7 +169,7 @@ for model_key, model_id in TRANSLATION_MODELS.items():
 # Constants
 SAMPLE_RATE = 16000
 OUTPUT_DIR = "/tmp/audio_outputs"
-REPO_REFERENCE_DIR = "./reference_audio"
+REFERENCE_AUDIO_DIR = "./reference_audios" 
 
 try:
     os.makedirs(OUTPUT_DIR, exist_ok=True)
@@ -744,22 +744,25 @@ def upload_reference_audio():
 
 def init_reference_audio():
     try:
-        os.makedirs(REFERENCE_AUDIO_DIR, exist_ok=True)
-        logger.info(f"📁 Created reference audio directory: {REFERENCE_AUDIO_DIR}")
+        # Create the output directory first
+        os.makedirs(OUTPUT_DIR, exist_ok=True)
+        logger.info(f"📁 Created output directory: {OUTPUT_DIR}")
         
-        # Copy reference files from repository to runtime directory
-        if os.path.exists(REPO_REFERENCE_DIR):
-            for pattern_dir in os.listdir(REPO_REFERENCE_DIR):
-                src_dir = os.path.join(REPO_REFERENCE_DIR, pattern_dir)
-                if os.path.isdir(src_dir):
-                    dst_dir = os.path.join(REFERENCE_AUDIO_DIR, pattern_dir)
-                    os.makedirs(dst_dir, exist_ok=True)
-                    
-                    # Copy all wav files
-                    for wav_file in glob.glob(os.path.join(src_dir, "*.wav")):
-                        shutil.copy(wav_file, dst_dir)
-                    
-            logger.info(f"📁 Copied reference files from repository")
+        # Check if the reference audio directory exists in the repository
+        if os.path.exists(REFERENCE_AUDIO_DIR):
+            logger.info(f"✅ Found reference audio directory: {REFERENCE_AUDIO_DIR}")
+            
+            # Log the contents to verify
+            pattern_dir = os.path.join(REFERENCE_AUDIO_DIR, reference_word)
+            if not os.path.exists(pattern_dir):
+                logger.warning(f"⚠️ Cannot write to repository directory: {pattern_dir}")
+                # Create a temporary directory instead
+                temp_dir = os.path.join(OUTPUT_DIR, "references", reference_word)
+                os.makedirs(temp_dir, exist_ok=True)
+                pattern_dir = temp_dir
+                logger.info(f"📁 Created temporary reference directory: {pattern_dir}")
+        else:
+            logger.warning(f"⚠️ Reference audio directory not found: {REFERENCE_AUDIO_DIR}")
     except Exception as e:
         logger.error(f"❌ Failed to set up reference audio directory: {str(e)}")