Updated Handler to have better debugging output when encountering errors trying to generate mesh

handler.py

@@ -64,6 +64,15 @@ class EndpointHandler():
 
     def __call__(self, input_data: Any) -> Any:
 
+         # Check if input_data is a string and deserialize if necessary
+        if isinstance(input_data, str):
+            print("input_data is a string, attempting to deserialize...")
+            try:
+                input_data = json.loads(input_data)  # Convert JSON string to dictionary
+            except json.JSONDecodeError as e:
+                print(f"Failed to parse JSON: {e}")
+                return None  # Handle the error as appropriate
+
         command = "null"
 
         if "command" in input_data:
@@ -79,7 +88,20 @@ class EndpointHandler():
         elif command == "generate_pc":
             return self.generate_point_cloud(input_data)  
         elif command == "generate_mesh":
-            return self.generate_mesh_from_pc(input_data["raw_pc"])
+            print("generate_mesh command received...")
+            raw_pc = input_data.get("raw_pc")
+    
+            if raw_pc is None:
+                print("raw_pc not found in input_data!")
+                return None
+    
+            # Check if raw_pc is a string and deserialize if necessary
+            if isinstance(raw_pc, str):
+                print("raw_pc is a string, attempting to deserialize...")
+                raw_pc = json.loads(raw_pc)
+    
+            print("Calling generate_mesh_from_pc...")
+            return self.generate_mesh_from_pc(raw_pc)
         elif command == "status":
             return self.check_status()
             
@@ -144,7 +166,6 @@ class EndpointHandler():
         #image = self.pipe(inputs, guidance_scale=7.5)["sample"][0]  
 
         pc = sampler.output_to_point_clouds(samples)[0]
-
         pc_dict = {}
 
         data_list = pc.coords.tolist()