Update app.py

app.py

@@ -23,7 +23,7 @@ def __init__(self):
     self.patient_data = {}
 
 @tool
-def collect_symptoms(self, patient_id: str, symptoms: str) -> str:
+def collect_symptoms(self: Self, patient_id: str, symptoms: str) -> str:
     """
     Nurse tool to collect symptoms and ask follow-up questions.
 
@@ -59,7 +59,7 @@ def collect_symptoms(self, patient_id: str, symptoms: str) -> str:
         return f"Unexpected Error: {str(e)}"
 
 @tool
-def diagnose_patient(self, patient_id: str) -> str:
+def diagnose_patient(self: Self, patient_id: str) -> str:
     """
     Doctor tool to diagnose the patient based on symptoms.
 
@@ -89,7 +89,7 @@ def diagnose_patient(self, patient_id: str) -> str:
         return f"Unexpected Error: {str(e)}"
 
 @tool
-def fetch_diagnosis(self, symptoms: str) -> str:
+def fetch_diagnosis(self: Self, symptoms: str) -> str:
     """
     AI tool to retrieve a diagnosis based on symptoms.
 
@@ -113,7 +113,7 @@ def fetch_diagnosis(self, symptoms: str) -> str:
         return f"Unexpected Error: {str(e)}"
 
 @tool
-def fetch_medication(self, symptoms: str) -> str:
+def fetch_medication(self: Self, symptoms: str) -> str:
     """
     AI tool to suggest medications based on symptoms.
 
@@ -137,7 +137,7 @@ def fetch_medication(self, symptoms: str) -> str:
         return f"Unexpected Error: {str(e)}"
 
 @tool
-def fetch_treatment_advice(self, symptoms: str) -> str:
+def fetch_treatment_advice(self: Self, symptoms: str) -> str:
     """
     AI tool to provide treatment recommendations.