Update app.py

Made edits on diagnose_patient

app.py

@@ -59,25 +59,24 @@ def collect_symptoms(patient_id: str, symptoms: str) -> str:
         return f"Unexpected Error: {str(e)}"
 
 @tool
-def diagnose_patient(self: CodeAgent, patient_id: str) -> str:
+def diagnose_patient(patient_id: str) -> str:
     """
     Doctor tool to diagnose the patient based on symptoms.
-
+    
     Args:
-        self: Reference to the instance of the class.
         patient_id: Unique identifier for the patient.
-
+        
     Returns:
         Diagnosis and recommendations.
     """
     try:
-        if patient_id not in self.patient_data:
+        if patient_id not in CodeAgent.patient_data:
             raise ValueError("No symptoms found. Nurse must collect symptoms first.")
 
-        symptoms = self.patient_data[patient_id]["symptoms"]
-        diagnosis = self.fetch_diagnosis(symptoms)
-        medication = self.fetch_medication(symptoms)
-        advice = self.fetch_treatment_advice(symptoms)
+        symptoms = CodeAgent.patient_data[patient_id]["symptoms"]
+        diagnosis = CodeAgent.fetch_diagnosis(symptoms)
+        medication = CodeAgent.fetch_medication(symptoms)
+        advice = CodeAgent.fetch_treatment_advice(symptoms)
 
         return (f"Doctor: Based on the symptoms: {symptoms},\n"
                 f"Diagnosis: {diagnosis}\n"
@@ -89,6 +88,7 @@ def diagnose_patient(self: CodeAgent, patient_id: str) -> str:
     except Exception as e:
         return f"Unexpected Error: {str(e)}"
 
+
 @tool
 def fetch_diagnosis(symptoms: str) -> str:
     """