1. Removed hardcoded "cough" question

2. Added generic follow-up about symptom timing
3. Included transcript in error responses for debugging
4. Maintained JSON structure but made content dynamic

src/app.py

@@ -144,19 +144,29 @@ def process_speech(new_transcript, history):
         return history
     
     try:
+        # Query the symptom index with user's transcript
         response = symptom_index.as_query_engine().query(new_transcript)
         
+        # Extract the ICD code and description
+        icd_info = str(response).strip()
+        
+        # Structure the initial response
         formatted_response = {
-            "diagnoses": [str(response).split(":")[0]],
-            "confidences": [0.8],
-            "follow_up": "Is the cough productive or dry?"
+            "diagnoses": [icd_info],
+            "confidences": [0.8],  # We could make this more dynamic
+            "follow_up": f"Could you tell me more about your symptoms? When did they start?"
         }
         
+        # Add to conversation history
         history.append({"role": "user", "content": new_transcript})
         history.append({"role": "assistant", "content": json.dumps(formatted_response, indent=2)})
         
     except Exception as e:
-        error_response = {"error": str(e), "status": "error"}
+        error_response = {
+            "error": str(e), 
+            "status": "error",
+            "transcript": new_transcript  # Include what we heard for debugging
+        }
         history.append({"role": "user", "content": new_transcript})
         history.append({"role": "assistant", "content": json.dumps(error_response, indent=2)})