Fix Layout

app.py

@@ -60,8 +60,8 @@ display_names = {
 mapping_table_idx_dataset_idx = {}
 
 map_periodic_table = {v.symbol: k for k, v in enumerate(periodictable.elements)}
+n_elements = len(map_periodic_table)
 
-dataset_index = np.zeros((len(dataset), 118))
 train_df = dataset.to_pandas()
 
 pattern = re.compile(r"(?P<element>[A-Z][a-z]?)(?P<count>\d*)")
@@ -125,20 +125,35 @@ layout = html.Div(
                 html.Div(
                     [
                         html.H3("Search Materials (eg. 'Ac,Cd,Ge' or 'Ac2CdGe3')"),
-                        dmp.MaterialsInput(
-                            allowedInputTypes=["elements", "formula"],
-                            hidePeriodicTable=False,
-                            periodicTableMode="toggle",
-                            hideWildcardButton=True,
-                            showSubmitButton=True,
-                            submitButtonText="Search",
-                            type="elements",
-                            id="materials-input",
+                        html.Div(
+                            [
+                                html.Div(
+                                    [
+                                        dmp.MaterialsInput(
+                                            allowedInputTypes=["elements", "formula"],
+                                            hidePeriodicTable=False,
+                                            periodicTableMode="toggle",
+                                            hideWildcardButton=True,
+                                            showSubmitButton=True,
+                                            submitButtonText="Search",
+                                            type="elements",
+                                            id="materials-input",
+                                        ),
+                                    ],
+                                    style={
+                                        "width": "48%",
+                                    },
+                                ),
+                            ],
+                            style={
+                                "display": "flex",
+                                "justifyContent": "center",
+                                "width": "100%",
+                            },
                         ),
                     ],
                     style={
                         "width": "100%",
-                        "display": "inline-block",
                         "verticalAlign": "top",
                     },
                 ),
@@ -191,7 +206,7 @@ layout = html.Div(
 
 
 def search_materials(query):
-    query_vector = np.zeros(118)
+    query_vector = np.zeros(n_elements)
 
     if "," in query:
         element_list = [el.strip() for el in query.split(",")]