Upload chat_agent.py

agents/chat_agent.py

@@ -8,18 +8,35 @@ class ChatAgent:
         self.chat_history = []
 
     def respond_to_query(self, message: str) -> str:
-        # Replace this with your actual logic or import from another agent
-        if "next pod" in message.lower():
+        message = message.lower()
+
+        if "next pod" in message:
             return "Your next pod site is right_leg."
-        elif "sensor" in message.lower():
+        elif "sensor" in message:
             return "Your next sensor site is left_arm."
-        elif "i used" in message.lower():
+        elif "i used" in message:
             site = message.split("i used")[1].strip().replace(".", "")
             self.chat_history.append((message, f"Logged {site} as your most recent site."))
             return f"Thanks, I’ve logged {site} as your most recent pod site."
+        elif "schedule" in message or "show me" in message:
+            return self.get_schedule()
         else:
             return "Try asking about your next pod or sensor site."
 
+    def get_schedule(self) -> str:
+        # Replace with dynamic logic or dataset later
+        schedule = [
+            "08/24/2025 — Sensor (left_arm)",
+            "08/24/2025 — Pod (left_arm)",
+            "08/27/2025 — Pod (left_leg)",
+            "08/30/2025 — Pod (left_stomach)",
+            "09/02/2025 — Sensor (right_leg)",
+            "09/02/2025 — Pod (right_arm)",
+            "09/05/2025 — Pod (right_leg)",
+            "09/08/2025 — Pod (right_stomach)",
+        ]
+        return "📅 Upcoming Schedule:\n" + "\n".join(schedule)
+
     def handle_message(self, user_message):
         response = self.respond_to_query(user_message)
         self.chat_history.append((user_message, response))