Update deployer/simulator_interface.py

deployer/simulator_interface.py

@@ -1,32 +1,54 @@
+# simulator_interface.py - Robot app simulator for Hugging Face Spaces (CPU only)
 import time
 import logging
-from typing import Optional
+
+logging.basicConfig(level=logging.INFO)
 
 class VirtualRobot:
+    """
+    Simulated robot that can wave, speak, and perform simple commands.
+    Useful for prototyping in CPU-only environments like Hugging Face Spaces.
+    """
+
     def __init__(self):
         self.state = "IDLE"
-        logging.basicConfig(level=logging.INFO)
-        logging.info("Robot initialized")
-
-    def perform_action(self, action: str) -> str:
-        action = action.lower().strip()
-        
-        if action == "wave":
-            return self._wave()
-        elif action.startswith("say"):
-            return self._speak(action[3:].strip())
-        return "❓ Unknown command"
+        logging.info("[🤖] VirtualRobot initialized: state=IDLE")
 
-    def _wave(self) -> str:
+    def wave(self) -> str:
+        """
+        Simulate an arm wave action.
+        """
+        logging.info("[🖐️] VirtualRobot: waving")
         self.state = "WAVING"
         time.sleep(0.5)
         self.state = "IDLE"
-        return "👋 Wave complete!"
+        return "🤖 *waves*"
 
-    def _speak(self, message: str) -> str:
-        if not message:
-            return "❌ No message provided"
+    def speak(self, text: str) -> str:
+        """
+        Simulate robot speech.
+        """
+        logging.info(f"[💬] VirtualRobot: speaking -> '{text}'")
         self.state = "SPEAKING"
-        time.sleep(max(0.3, len(message)*0.05))
+        time.sleep(0.5)
         self.state = "IDLE"
-        return f"🗣️ {message.capitalize()}"
+        return f"🗣️ {text}"
+
+    def perform_action(self, command: str) -> str:
+        """
+        Parse and execute a command. Supports:
+          - "wave": calls wave()
+          - "say <message>": calls speak(message)
+        Returns an error message for unknown commands.
+        """
+        parts = command.strip().split(" ", 1)
+        action = parts[0].lower()
+        arg = parts[1] if len(parts) > 1 else ""
+
+        if action == "wave":
+            return self.wave()
+        elif action == "say" and arg:
+            return self.speak(arg)
+        else:
+            logging.warning(f"[⚠️] VirtualRobot: unknown command '{command}'")
+            return f"❓ Unknown action: {command}"