Update app.py

app.py

@@ -135,15 +135,17 @@ def analyze_symptoms(text):
         return f"Error analyzing symptoms: {str(e)}", 0.0
 
 def handle_health_query(query, language="en"):
-    """Handle health-related queries with a simple response."""
+    """Handle health-related queries with a general response."""
     if not query:
         return "Please provide a valid health query."
     # Placeholder for Q&A logic (could integrate a model like BERT for Q&A)
-    response = f"Response to query '{query}': For accurate health information, consult a healthcare provider."
-    return response
+    restricted_terms = ["medicine", "treatment", "drug", "prescription"]
+    if any(term in query.lower() for term in restricted_terms):
+        return "This tool does not provide medication or treatment advice. Please ask about symptoms or general health information (e.g., 'What are symptoms of asthma?')."
+    return f"Response to query '{query}': For accurate health information, consult a healthcare provider."
 
 def analyze_voice(audio_file, language="en"):
-    """Analyze voice for health indicators."""
+    """Analyze voice for health indicators and handle queries."""
     try:
         # Ensure unique file name
         unique_path = f"/tmp/gradio/{datetime.now().strftime('%Y%m%d%H%M%S%f')}_{os.path.basename(audio_file)}"
@@ -163,25 +165,39 @@ def analyze_voice(audio_file, language="en"):
         if "Error transcribing" in transcription:
             return transcription
         
-        # 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
+        # Split transcription into symptom and query parts
+        symptom_text = transcription
+        query_text = None
+        restricted_terms = ["medicine", "treatment", "drug", "prescription"]
+        for term in restricted_terms:
+            if term in transcription.lower():
+                # Split at the first restricted term
+                split_index = transcription.lower().find(term)
+                symptom_text = transcription[:split_index].strip()
+                query_text = transcription[split_index:].strip()
+                break
         
-        # Analyze symptoms
-        prediction, score = analyze_symptoms(transcription)
-        if "Error analyzing" in prediction:
-            return prediction
+        feedback = ""
         
-        # Generate feedback
-        if prediction == "No health condition predicted":
-            feedback = "No significant health indicators detected."
+        # Analyze symptoms if present
+        if symptom_text:
+            prediction, score = analyze_symptoms(symptom_text)
+            if "Error analyzing" in prediction:
+                feedback += prediction + "\n"
+            elif prediction == "No health condition predicted":
+                feedback += "No significant health indicators detected.\n"
+            else:
+                feedback += f"Possible health condition: {prediction} (confidence: {score:.4f}). Consult a doctor.\n"
         else:
-            feedback = f"Possible health condition: {prediction} (confidence: {score:.4f}). Consult a doctor."
+            feedback += "No symptoms detected in the audio.\n"
         
-        feedback += f"\n\n**Debug Info**: Transcription = '{transcription}', Prediction = {prediction}, Confidence = {score:.4f}, File Hash = {file_hash}"
+        # Handle query if present
+        if query_text:
+            feedback += f"\nQuery detected: '{query_text}'\n"
+            feedback += handle_health_query(query_text, language) + "\n"
+        
+        # Add debug info and disclaimer
+        feedback += f"\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."
         
         # Clean up temporary audio file
@@ -202,9 +218,10 @@ def create_gradio_interface():
             """
             # Health Voice Analyzer
             Record or upload a voice sample describing symptoms in English, Spanish, Hindi, or Mandarin (e.g., 'I have a fever').  
-            Ask health questions in the text box below. Supports WAV, 16kHz audio.  
+            Ask health questions in the text box below (e.g., 'What are symptoms of asthma?').  
+            **Note**: Do not ask for medication or treatment advice; focus on symptoms or general health questions.  
             **Disclaimer**: This is not a diagnostic tool. Consult a healthcare provider for medical advice.  
-            **Note**: Text-to-speech is available in the web frontend (Salesforce Sites) using the browser's Web Speech API.
+            **Text-to-Speech**: Available in the web frontend (Salesforce Sites) using the browser's Web Speech API.
             """
         )
         with gr.Row():