Update app.py

app.py

@@ -59,10 +59,15 @@ def medical_assistant_interaction(pulse_rate, blood_pressure_systolic, blood_pre
             #max_output_tokens=150
         )
         #assistant_message = response.last['content'].strip()
-        if 'candidates' in response:
-            assistant_message = response['candidates'][0]['content'].strip()
+        if hasattr(response, 'candidates'):
+            candidates = response.candidates
+            if len(candidates) > 0 and hasattr(candidates[0], 'content'):
+                assistant_message = candidates[0].content.strip()
+            else:
+                assistant_message = "Error: No valid response from the model."
         else:
-            assistant_message = "Error: No valid response from the model."
+            assistant_message = "Error: Response object does not have 'candidates' attribute."
+
             
         # Combine the assistant's message with the model's prediction
         if prediction == 1: