Update app.py

app.py

@@ -10,6 +10,7 @@ import torch
 from tenacity import retry, stop_after_attempt, wait_fixed
 import logging
 import tempfile
+import shutil
 
 # Set up logging
 logging.basicConfig(
@@ -88,6 +89,20 @@ def compute_file_hash(file_path):
         logger.error(f"Failed to compute file hash: {str(e)}")
         return "unknown"
 
+def ensure_writable_dir(directory):
+    """Ensure directory exists and is writable."""
+    try:
+        os.makedirs(directory, exist_ok=True)
+        test_file = os.path.join(directory, "test_write")
+        with open(test_file, "w") as f:
+            f.write("test")
+        os.remove(test_file)
+        logger.debug(f"Directory {directory} is writable")
+        return True
+    except Exception as e:
+        logger.error(f"Directory {directory} not writable: {str(e)}")
+        return False
+
 def transcribe_audio(audio_file):
     """Transcribe audio using Whisper model."""
     if not whisper:
@@ -95,6 +110,9 @@ def transcribe_audio(audio_file):
         return "Error: Whisper model not loaded"
     try:
         logger.debug(f"Transcribing audio: {audio_file}")
+        if not os.path.exists(audio_file):
+            logger.error(f"Audio file not found: {audio_file}")
+            return "Error: Audio file not found"
         audio, sr = librosa.load(audio_file, sr=16000)
         if len(audio) < 1600:
             logger.error("Audio too short")
@@ -164,13 +182,24 @@ def analyze_voice(audio_file):
             logger.error(f"Audio file not found: {audio_file}")
             return "Error: Audio file not found"
         
+        # Ensure temp directory is writable
+        temp_dir = os.path.join(tempfile.gettempdir(), "gradio")
+        if not ensure_writable_dir(temp_dir):
+            logger.error(f"Temp directory {temp_dir} not writable")
+            return f"Error: Temp directory {temp_dir} not writable"
+        
+        # Rename file to unique path
         unique_path = os.path.join(
-            tempfile.gettempdir(),
+            temp_dir,
             f"gradio_{datetime.now().strftime('%Y%m%d%H%M%S%f')}_{os.path.basename(audio_file)}"
         )
-        os.rename(audio_file, unique_path)
-        audio_file = unique_path
-        logger.debug(f"Renamed to: {audio_file}")
+        try:
+            shutil.move(audio_file, unique_path)
+            audio_file = unique_path
+            logger.debug(f"Moved to: {audio_file}")
+        except Exception as e:
+            logger.error(f"Failed to rename audio file: {str(e)}")
+            return f"Error: Failed to rename audio file: {str(e)}"
         
         file_hash = compute_file_hash(audio_file)
         logger.info(f"Processing audio, Hash: {file_hash}")
@@ -212,7 +241,12 @@ def analyze_voice(audio_file):
 
 def test_with_sample_audio():
     """Test with sample or synthetic audio."""
-    sample_audio_path = "audio_samples/sample.wav"
+    sample_audio_path = os.path.join("audio_samples", "sample.wav")
+    temp_dir = os.path.join(tempfile.gettempdir(), "audio_samples")
+    if not ensure_writable_dir(temp_dir):
+        logger.error(f"Temp directory {temp_dir} not writable")
+        return f"Error: Temp directory {temp_dir} not writable"
+    
     if not os.path.exists(sample_audio_path):
         logger.warning("Sample audio not found; generating synthetic audio")
         sr = 16000
@@ -221,8 +255,7 @@ def test_with_sample_audio():
         amplitude_mod = 0.5 + 0.1 * np.sin(2 * np.pi * 0.3 * t)
         noise = 0.01 * np.random.normal(0, 1, len(t))
         dummy_audio = amplitude_mod * np.sin(2 * np.pi * freq_mod * t) + noise
-        sample_audio_path = os.path.join(tempfile.gettempdir(), "dummy_test.wav")
-        os.makedirs(os.path.dirname(sample_audio_path), exist_ok=True)
+        sample_audio_path = os.path.join(temp_dir, "dummy_test.wav")
         try:
             soundfile.write(dummy_audio, sr, sample_audio_path)
             logger.info(f"Generated synthetic audio: {sample_audio_path}")
@@ -230,6 +263,11 @@ def test_with_sample_audio():
             logger.error(f"Failed to write synthetic audio: {str(e)}")
             return f"Error: Failed to generate synthetic audio: {str(e)}"
     
+    # Verify file existence
+    if not os.path.exists(sample_audio_path):
+        logger.error(f"Synthetic audio not created: {sample_audio_path}")
+        return f"Error: Synthetic audio not created: {sample_audio_path}"
+    
     mock_transcription = "I have a cough and sore throat"
     logger.info(f"Mock transcription: {mock_transcription}")
     prediction, score = analyze_symptoms(mock_transcription)