Update app.py

app.py

@@ -104,7 +104,7 @@ def fit_outputs_constraints(x, hardness_target, ys_target, request: gr.Request):
           error_hardness, error_ys)
     return error_hardness + error_ys
 
-def predict_inverse(hardness_original_target, ys_original_target, request: gr.Request):
+def predict_inverse(hardness_original_target, ys_original_target, metals_to_use, request: gr.Request):
 
     one_hot_columns = utils.return_feature_names()
     min_df_hardness, max_df_hardness = scaling_factors["PROPERTY: HV"]
@@ -131,7 +131,7 @@ def predict_inverse(hardness_original_target, ys_original_target, request: gr.Re
 
     # Metals constraints 
     metals_elements = [c for c in continuous_variables if c.startswith("PROPERTY: Metal")]
-    metals_to_use = ['Al', 'Co', 'Fe', 'Cr']
+    # metals_to_use = ['Al', 'Co', 'Fe', 'Cr']
     metals_to_use = ["PROPERTY: Metal " + metals_to_use[i] for i in range(len(metals_to_use))]
 
     # Domain 
@@ -231,6 +231,13 @@ with gr.Blocks(css=css_styling, title=page_title, theme=osium_theme) as demo:
             gr.Markdown("### The target performance of your alloy")
             input_hardness = gr.Text(label="Enter your target hardness (in HV)")
             input_yield_strength = gr.Text(label="Enter your target yield strength (MPa)")
+            gr.Markdown('### Your metallic elements constraints')
+            metals_constraints = gr.CheckboxGroup(
+                choices=['Al', 'Co', 'Fe', 'Ni', 'Si', 'Cr', 'Nb', 'Ti',
+                        'Mn', 'V', 'Mo', 'Cu', 'Ta', 'Zr',
+                        'Hf', 'W', 'Zn', 'Sn', 'Re', 'C',
+                        'Pd', 'Sc', 'Y'], label="Your metals constraints",
+            )
         with gr.Column():
             with gr.Row():
                 #with gr.Column():
@@ -257,7 +264,7 @@ with gr.Blocks(css=css_styling, title=page_title, theme=osium_theme) as demo:
 
     prediction_button.click(
         fn=predict_inverse,
-        inputs=[input_hardness, input_yield_strength],
+        inputs=[input_hardness, input_yield_strength, metals_constraints],
         outputs=[optimal_parameters
         ],
         show_progress=True,