Update app.py

app.py

@@ -160,16 +160,20 @@ def analyze_symptoms(text):
             return "Error: No valid transcription", 0.0
         with torch.no_grad():
             result = symptom_classifier(text)
-        if result and isinstance(result, list) and len(result) > 0:
-            prediction = result[0]["label"]
-            score = result[0]["score"]
-            if is_fallback_model:
-                logger.warning("Using fallback model")
-                prediction = f"{prediction} (fallback)"
-            logger.info(f"Prediction: {prediction}, Score: {score:.4f}")
-            return prediction, score
-        logger.warning("No prediction returned")
-        return "No health condition detected", 0.0
+        logger.debug(f"Model output: {result}")
+        if not result or not isinstance(result, list) or len(result) == 0:
+            logger.warning("Invalid model output: empty or not a list")
+            return "No health condition detected", 0.0
+        if not isinstance(result[0], dict) or "label" not in result[0] or "score" not in result[0]:
+            logger.warning(f"Invalid result structure: {result[0]}")
+            return "No health condition detected", 0.0
+        prediction = result[0]["label"]
+        score = result[0]["score"]
+        if is_fallback_model:
+            logger.warning("Using fallback model")
+            prediction = f"{prediction} (fallback)"
+        logger.info(f"Prediction: {prediction}, Score: {score:.4f}")
+        return prediction, score
     except Exception as e:
         logger.error(f"Symptom analysis failed: {str(e)}")
         return f"Error: {str(e)}", 0.0
@@ -182,13 +186,11 @@ 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(
             temp_dir,
             f"gradio_{datetime.now().strftime('%Y%m%d%H%M%S%f')}_{os.path.basename(audio_file)}"
@@ -198,8 +200,8 @@ def analyze_voice(audio_file):
             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)}"
+            logger.error(f"Failed to move audio file: {str(e)}")
+            return f"Error: Failed to move audio file: {str(e)}"
         
         file_hash = compute_file_hash(audio_file)
         logger.info(f"Processing audio, Hash: {file_hash}")
@@ -241,12 +243,12 @@ def analyze_voice(audio_file):
 
 def test_with_sample_audio():
     """Test with sample or synthetic audio."""
-    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"
     
+    sample_audio_path = os.path.join(temp_dir, "sample.wav")
     if not os.path.exists(sample_audio_path):
         logger.warning("Sample audio not found; generating synthetic audio")
         sr = 16000
@@ -263,7 +265,6 @@ 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}"