Update app/src/main/java/com/example/cuida/ui/sns24/Sns24Fragment.java

app/src/main/java/com/example/cuida/ui/sns24/Sns24Fragment.java

@@ -74,15 +74,19 @@ public class Sns24Fragment extends Fragment {
                 "Se os sintomas indicarem perigo de vida ou necessidade de observação urgente, OBRIGATORIAMENTE começa a tua primeira linha com a palavra [GRAVE]. " +
                 "Sintomas do paciente: " + symptoms;
 
-        ollama.fazerPergunta(prompt, new Ollama().OllamaCallback() {
+        ollama.fazerPergunta(prompt, new OllamaService.OllamaCallback() {
     @Override
     public void onSuccess(String result) {
+        // Já estamos na Main Thread graças ao Handler no serviço!
         if (getActivity() != null && binding != null) {
-                    getActivity().runOnUiThread(() -> {
+            
+            // Limpa o marcador [GRAVE] para não aparecer ao utilizador
             String displayResult = result.replace("[GRAVE]", "").trim();
+            
             binding.textAiResult.setText(displayResult);
             binding.buttonAiTriage.setEnabled(true);
             
+            // Verifica se a string ORIGINAL continha o marcador
             if (result.contains("[GRAVE]")) {
                 binding.buttonFindHospital.setVisibility(View.VISIBLE);
                 binding.buttonFindHospital.setOnClickListener(v -> {
@@ -92,21 +96,20 @@ public class Sns24Fragment extends Fragment {
                     startActivity(mapIntent);
                 });
             }
+            
             saveTriageToHistory(symptoms, displayResult);
-                    });
         }
     }
 
     @Override
     public void onError(Throwable t) {
+        // Já estamos na Main Thread
         if (getActivity() != null && binding != null) {
-                    getActivity().runOnUiThread(() -> {
             binding.textAiResult.setText("Erro na ligação: " + t.getMessage());
             binding.buttonAiTriage.setEnabled(true);
-                    });
         }
     }
-        });
+});
     }
 
     private void saveTriageToHistory(String symptoms, String result) {