Update app.py

app.py

@@ -8,6 +8,7 @@ from transformers import pipeline
 import soundfile as sf
 import torch
 from tenacity import retry, stop_after_attempt, wait_fixed
+from gtts import gTTS
 
 # Initialize local models with retry logic
 @retry(stop=stop_after_attempt(3), wait=wait_fixed(2))
@@ -134,8 +135,21 @@ def analyze_symptoms(text):
     except Exception as e:
         return f"Error analyzing symptoms: {str(e)}", 0.0
 
+def generate_voice_feedback(text):
+    """Generate voice feedback from text using gTTS."""
+    try:
+        # Remove debug info and disclaimer for cleaner voice output
+        clean_text = text.split("\n\n**Debug Info**")[0]
+        tts = gTTS(text=clean_text, lang='en')
+        output_file = f"/tmp/feedback_{datetime.now().strftime('%Y%m%d%H%M%S%f')}.mp3"
+        tts.save(output_file)
+        return output_file
+    except Exception as e:
+        print(f"Error generating voice feedback: {str(e)}")
+        return None
+
 def analyze_voice(audio_file):
-    """Analyze voice for health indicators."""
+    """Analyze voice for health indicators and provide text and voice feedback."""
     try:
         # Ensure unique file name to avoid Gradio reuse
         unique_path = f"/tmp/gradio/{datetime.now().strftime('%Y%m%d%H%M%S%f')}_{os.path.basename(audio_file)}"
@@ -153,19 +167,22 @@ def analyze_voice(audio_file):
         # Transcribe audio
         transcription = transcribe_audio(audio_file)
         if "Error transcribing" in transcription:
-            return transcription
+            voice_file = generate_voice_feedback(transcription)
+            return transcription, voice_file
         
         # Check for medication-related queries
         if "medicine" in transcription.lower() or "treatment" in transcription.lower():
             feedback = "Error: This tool does not provide medication or treatment advice. Please describe symptoms only (e.g., 'I have a fever')."
             feedback += f"\n\n**Debug Info**: Transcription = '{transcription}', File Hash = {file_hash}"
             feedback += "\n**Disclaimer**: This is not a diagnostic tool. Consult a healthcare provider for medical advice."
-            return feedback
+            voice_file = generate_voice_feedback(feedback)
+            return feedback, voice_file
         
         # Analyze symptoms
         prediction, score = analyze_symptoms(transcription)
         if "Error analyzing" in prediction:
-            return prediction
+            voice_file = generate_voice_feedback(prediction)
+            return prediction, voice_file
         
         # Generate feedback
         if prediction == "No health condition predicted":
@@ -176,6 +193,9 @@ def analyze_voice(audio_file):
         feedback += f"\n\n**Debug Info**: Transcription = '{transcription}', Prediction = {prediction}, Confidence = {score:.4f}, File Hash = {file_hash}"
         feedback += "\n**Disclaimer**: This is not a diagnostic tool. Consult a healthcare provider for medical advice."
         
+        # Generate voice feedback
+        voice_file = generate_voice_feedback(feedback)
+        
         # Clean up temporary audio file
         try:
             os.remove(audio_file)
@@ -183,9 +203,11 @@ def analyze_voice(audio_file):
         except Exception as e:
             print(f"Failed to delete audio file: {str(e)}")
         
-        return feedback
+        return feedback, voice_file
     except Exception as e:
-        return f"Error processing audio: {str(e)}"
+        feedback = f"Error processing audio: {str(e)}"
+        voice_file = generate_voice_feedback(feedback)
+        return feedback, voice_file
 
 def test_with_sample_audio():
     """Test the app with sample audio files."""
@@ -193,7 +215,8 @@ def test_with_sample_audio():
     results = []
     for sample in samples:
         if os.path.exists(sample):
-            results.append(analyze_voice(sample))
+            text, voice = analyze_voice(sample)
+            results.append(f"Text: {text}\nVoice: {voice}")
         else:
             results.append(f"Sample not found: {sample}")
     return "\n".join(results)
@@ -202,7 +225,10 @@ def test_with_sample_audio():
 iface = gr.Interface(
     fn=analyze_voice,
     inputs=gr.Audio(type="filepath", label="Record or Upload Voice"),
-    outputs=gr.Textbox(label="Health Assessment Feedback"),
+    outputs=[
+        gr.Textbox(label="Health Assessment Feedback"),
+        gr.Audio(label="Voice Feedback", type="filepath")
+    ],
     title="Health Voice Analyzer",
     description="Record or upload a voice sample describing symptoms (e.g., 'I have a fever') for preliminary health assessment. Supports English only. Use clear audio (WAV, 16kHz). Do not ask for medication or treatment advice."
 )