Update app.py

app.py

@@ -43,8 +43,11 @@ def sample_info(sample_id_str, time_step, fieldn):
     plaid_sample = Sample.model_validate(pickle.loads(sample_))
     # plaid_sample = Sample.load_from_dir(f"Tensile2d/dataset/samples/sample_"+str(sample_id_str).zfill(9))
 
-    nodes = plaid_sample.get_nodes(time=time)
-    field = plaid_sample.get_field(fieldn, time=time)
+    nodes = plaid_sample.get_nodes()
+    if fieldn in ["U_x", "U_y"]:
+        field = plaid_sample.get_field(fieldn, time=time)
+    else:
+        field = plaid_sample.get_field(fieldn, location="CellCenter", time=time)        
     # if nodes.shape[1] == 2:
     #     nodes__ = np.zeros((nodes.shape[0],nodes.shape[1]+1))
     #     nodes__[:,:-1] = nodes