Update app.py

app.py

@@ -76,6 +76,9 @@ def safe_parse_json_from_text(text: str):
 # -------------------------------
 # TRUE CORE FUNCTION
 # -------------------------------
+# -------------------------------
+# TRUE CORE FUNCTION (with objects)
+# -------------------------------
 def robot_watch_core(payload: Dict[str, Any]):
     if isinstance(payload, str):
         try:
@@ -103,7 +106,8 @@ Respond in STRICT JSON ONLY.
 {
  "description": "...",
  "human": "...",
- "environment": "..."
+ "environment": "...",
+ "objects": []  // list of detected objects
 }
 """
 
@@ -121,7 +125,7 @@ Respond in STRICT JSON ONLY.
         resp = client.chat.completions.create(
             model=HF_VLM_MODEL,
             messages=messages,
-            max_tokens=300,
+            max_tokens=500,
             temperature=0.1
         )
     except Exception as e:
@@ -137,6 +141,11 @@ Respond in STRICT JSON ONLY.
             "message": "Invalid JSON returned"
         }
 
+    # Ensure "objects" is a list
+    objects = parsed.get("objects", [])
+    if not isinstance(objects, list):
+        objects = []
+
     return {
         "status": "success",
         "robot_id": robot_id,
@@ -145,10 +154,12 @@ Respond in STRICT JSON ONLY.
         "description": parsed.get("description"),
         "human": parsed.get("human"),
         "environment": parsed.get("environment"),
+        "objects": objects,   # ← new field
         "vlm_raw": vlm_output
     }
 
 
+
 # -------------------------------
 # REGISTER MCP TOOL (wrapper)
 # -------------------------------