Fix: Implement symptom analysis results display

lovable-dev[bot] · lovable-dev[bot] · commit 589d15e6f10b · 2025-04-22T07:39:55.000Z
Implement the display of disease predictions after analyzing symptoms, as per the provided design.
diff --git a/src/components/symptoms/SymptomChecker.tsx b/src/components/symptoms/SymptomChecker.tsx
@@ -83,48 +83,35 @@ const SymptomChecker = () => {
 
     try {
       const symptomNames = selectedSymptoms.map(s => s.name);
-      
       const predictions = await predictDisease(symptomNames);
       
-      if (predictions.length > 0) {
-        const topDisease = predictions[0];
-        
-        const recommendedDoctors = getDoctorsBySpecialty(topDisease.specialist || "General Physician");
-        
-        const precautions = [
-          "Consult with a healthcare professional",
-          "Get adequate rest",
-          "Stay hydrated",
-          "Monitor your symptoms",
-          "Follow prescribed medications if any"
-        ];
+      if (predictions && predictions.length > 0) {
+        const recommendedDoctors = getDoctorsBySpecialty(predictions[0].specialist);
         
-        const diet = [
-          "Maintain a balanced diet",
-          "Include fresh fruits and vegetables",
-          "Stay hydrated with water",
-          "Avoid processed foods",
-          "Consider supplements as recommended by your doctor"
-        ];
-
         setResults({
           diseases: predictions,
-          precautions,
-          diet,
+          precautions: [
+            "Consult with a healthcare professional",
+            "Get adequate rest",
+            "Stay hydrated",
+            "Monitor your symptoms",
+            "Follow prescribed medications if any"
+          ],
+          diet: [
+            "Maintain a balanced diet",
+            "Include fresh fruits and vegetables",
+            "Stay hydrated with water",
+            "Avoid processed foods",
+            "Consider supplements as recommended by your doctor"
+          ],
           doctors: recommendedDoctors
         });
-      } else {
-        toast({
-          title: "Analysis Inconclusive",
-          description: "We couldn't determine a likely condition from your symptoms. Please consult a healthcare professional.",
-          variant: "destructive",
-        });
       }
     } catch (error) {
-      console.error('Error during analysis:', error);
+      const errorMessage = error instanceof Error ? error.message : "An error occurred while analyzing your symptoms";
       toast({
-        title: "Error During Analysis",
-        description: "An error occurred while analyzing your symptoms. Please try again.",
+        title: "Analysis Error",
+        description: errorMessage,
         variant: "destructive",
       });
     } finally {
diff --git a/src/utils/diseasePredictor.ts b/src/utils/diseasePredictor.ts
@@ -1,5 +1,6 @@
 
 import { pipeline } from '@huggingface/transformers';
+import { DISEASES } from '@/data/diseases';
 
 let classifier: any = null;
 
@@ -15,31 +16,43 @@ export const initializeModel = async () => {
 };
 
 export const predictDisease = async (symptoms: string[]) => {
+  // If no symptoms provided, return early
+  if (!symptoms.length) {
+    throw new Error("Please select at least one symptom for analysis");
+  }
+
   const model = await initializeModel();
   
   // Convert symptoms array to a text description
   const symptomText = symptoms.join(', ');
   
-  // Define possible diseases as candidate labels
-  const candidateLabels = [
-    'Common Cold',
-    'Influenza',
-    'Seasonal Allergies',
-    'Migraine',
-    'Gastroesophageal Reflux Disease',
-    'Asthma',
-    'Arthritis',
-    'Hypertension'
-  ];
+  // Get disease names as candidate labels
+  const candidateLabels = DISEASES.map(disease => disease.name);
 
   try {
     const result = await model(symptomText, candidateLabels);
-    return result.labels.map((label: string, index: number) => ({
-      name: label,
-      probability: Math.round(result.scores[index] * 100)
-    }));
+    
+    // Filter predictions with score > 20%
+    const predictions = result.labels
+      .map((label: string, index: number) => ({
+        name: label,
+        probability: Math.round(result.scores[index] * 100),
+        specialist: DISEASES.find(d => d.name === label)?.specialist || 'General Physician',
+        description: DISEASES.find(d => d.name === label)?.description || ''
+      }))
+      .filter(pred => pred.probability > 20)
+      .sort((a, b) => b.probability - a.probability)
+      .slice(0, 3); // Top 3 predictions
+
+    if (predictions.length === 0) {
+      throw new Error("No strong matches found for the provided symptoms");
+    }
+
+    return predictions;
   } catch (error) {
-    console.error('Error predicting disease:', error);
-    return [];
+    if (error instanceof Error) {
+      throw error;
+    }
+    throw new Error("Error analyzing symptoms");
   }
 };
