Update app.py

Morrreeeeeeeee

app.py

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