Update app.py

app.py

@@ -23,7 +23,7 @@ def __init__(self):
     self.patient_data = {}
 
 @tool
-def collect_symptoms(self: CodeAgent, patient_id: str, symptoms: str) -> str:
+def collect_symptoms(patient_id: str, symptoms: str) -> str:
     """
     Nurse tool to collect symptoms and ask follow-up questions.
 
@@ -64,6 +64,7 @@ def diagnose_patient(self: CodeAgent, 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:
@@ -89,7 +90,7 @@ def diagnose_patient(self: CodeAgent, patient_id: str) -> str:
         return f"Unexpected Error: {str(e)}"
 
 @tool
-def fetch_diagnosis(self: CodeAgent, symptoms: str) -> str:
+def fetch_diagnosis(symptoms: str) -> str:
     """
     AI tool to retrieve a diagnosis based on symptoms.
 
@@ -113,7 +114,7 @@ def fetch_diagnosis(self: CodeAgent, symptoms: str) -> str:
         return f"Unexpected Error: {str(e)}"
 
 @tool
-def fetch_medication(self: CodeAgent, symptoms: str) -> str:
+def fetch_medication(symptoms: str) -> str:
     """
     AI tool to suggest medications based on symptoms.
 
@@ -137,7 +138,7 @@ def fetch_medication(self: CodeAgent, symptoms: str) -> str:
         return f"Unexpected Error: {str(e)}"
 
 @tool
-def fetch_treatment_advice(self: CodeAgent, symptoms: str) -> str:
+def fetch_treatment_advice(symptoms: str) -> str:
     """
     AI tool to provide treatment recommendations.