Update app.py

app.py

@@ -15,32 +15,36 @@ import os
 os.environ["PYOPENGL_PLATFORM"] = "egl"
 
 
-hf_dataset = load_dataset("PLAID-datasets/2D_profile", split="all_samples")
-
+hf_dataset = load_dataset("fabiencasenave/2D_ElastoPlastoDynamics", split="all_samples")
+sample0 = hf_dataset[0]["sample"]
+plaid_sample0 = Sample.model_validate(pickle.loads(sample_))
+times_steps = list(plaid_sample0._meshes.keys()) 
 
 ids_train = hf_dataset.description["split"]["train"]
 nb_samples = len(ids_train)
 
 
-field_names_train = ["Mach", "Pressure", "Velocity-x", "Velocity-y"]
+field_names_train = ["U_x", "U_y", "Sigma_xx", "Sigma_xy", "Sigma_yy", "Epsilon_xx", "Epsilon_xy", "Epsilon_yy", "EROSION_STATUS"]
 
 
 _HEADER_ = '''
-<h2><b>Visualization demo of <a href='https://huggingface.co/datasets/PLAID-datasets/2D_profile' target='_blank'><b>2D_profile dataset</b></b></h2>
+<h2><b>Visualization demo of <a href='https://huggingface.co/datasets/PLAID-datasets/2D_ElastoPlastoDynamics' target='_blank'><b>2D_profile 2D_ElastoPlastoDynamics</b></b></h2>
 '''
 
 
 def round_num(num)->str:
     return '%s' % float('%.3g' % num)
 
-def sample_info(sample_id_str, fieldn):
+def sample_info(sample_id_str, time_step, fieldn):
+
+    time = times_steps[int(time_step)]
 
     sample_ = hf_dataset[ids_train[int(sample_id_str)]]["sample"]
     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()
-    field = plaid_sample.get_field(fieldn)
+    nodes = plaid_sample.get_nodes(time=time)
+    field = plaid_sample.get_field(fieldn, time=time)
     # if nodes.shape[1] == 2:
     #     nodes__ = np.zeros((nodes.shape[0],nodes.shape[1]+1))
     #     nodes__[:,:-1] = nodes
@@ -132,14 +136,16 @@ if __name__ == "__main__":
         with gr.Row(variant="panel"):
             with gr.Column(scale=1):
                 d1 = gr.Slider(0, nb_samples-1, value=0, label="Training sample id", info="Choose between 0 and "+str(nb_samples-1))
+                d2 = gr.Slider(0, len(times_steps)-1, value=0, label="Time step", info="Choose between 0 and "+str(len(times_steps)-1))
                 output1 = gr.Text(label="Training sample info")
             with gr.Column(scale=2, min_width=300):
-                d2 = gr.Dropdown(field_names_train, value=field_names_train[0], label="Field name")        
+                d3 = gr.Dropdown(field_names_train, value=field_names_train[0], label="Field name")        
                 # output2 = gr.Image(label="Training sample visualization")    
                 output2 = gr.Model3D(label="Training sample visualization")
                 
-        d1.input(sample_info, [d1, d2], [output1, output2])
-        d2.input(sample_info, [d1, d2], [output1, output2]) 
+        d1.input(sample_info, [d1, d2, d3], [output1, output2])
+        d2.input(sample_info, [d1, d2, d3], [output1, output2]) 
+        d3.input(sample_info, [d1, d2, d3], [output1, output2]) 
 
     demo.launch()