Update app.py

app.py

@@ -58,8 +58,12 @@ def medical_assistant_interaction(pulse_rate, blood_pressure_systolic, blood_pre
             messages=[{"content": response_text}],
             #max_output_tokens=150
         )
-        assistant_message = response.last['content'].strip()
-
+        #assistant_message = response.last['content'].strip()
+        if 'candidates' in response:
+            assistant_message = response['candidates'][0]['content'].strip()
+        else:
+            assistant_message = "Error: No valid response from the model."
+            
         # Combine the assistant's message with the model's prediction
         if prediction == 1:
             model_inference = "Based on the sensor readings, the classifier suggests a high likelihood of Lassa fever."