Update app.py

app.py

@@ -8,7 +8,7 @@ from gradio_moleculeview import moleculeview
 
 import cellscape
 
-def predict(input_mol, style, view_str, chains):
+def predict(input_mol, style, contour_level, view_str, chains):
     # write view to file
     with open("view_matrix", "w") as f:
         f.write(json.loads(view_str))
@@ -22,7 +22,7 @@ def predict(input_mol, style, view_str, chains):
     if style == "Goodsell3D":
         os.system(f"cellscape cartoon --pdb {input_mol.name} --outline residue --color_by chain --depth_shading --depth_lines --colors {chain_str}  --depth flat --back_outline --view view_matrix --save outline_all.svg")
     elif style == "Contour":
-        os.system(f"cellscape cartoon --pdb {input_mol.name} --outline chain --color_by chain  --depth_contour_interval 20 --colors {chain_str}  --depth contours --back_outline --view view_matrix --save outline_all.svg")
+        os.system(f"cellscape cartoon --pdb {input_mol.name} --outline chain --color_by chain  --depth_contour_interval {contour_level} --colors {chain_str}  --depth contours --back_outline --view view_matrix --save outline_all.svg")
     else:
         os.system(f"cellscape cartoon --pdb {input_mol.name} --outline chain --colors {chain_str}  --depth flat --back_outline --view view_matrix --save outline_all.svg")
     
@@ -37,10 +37,17 @@ def predict(input_mol, style, view_str, chains):
 
 
 
+def show_contour_level(style):
+    if style=="Contour":
+        return gr.Slider(0,10,1, label="Contour level", visible=True)
+    else:
+        return gr.Slider(0,10,1, label="Contour level", visible=False)
 
 with gr.Blocks() as demo:
     gr.Markdown("# PDB2Vector")
     style = gr.Radio(value="Flat", choices=["Flat", "Contour", "Goodsell3D"], label="Style")
+    contour_level = gr.Slider(0,10,1, label="Contour level", visible=False)
+    style.change(show_contour_level, style, contour_level)
     inp = moleculeview(label="Molecule3D")
 
     view_str = gr.Textbox("viewMatrixResult", label="View Matrix", visible=False)
@@ -53,7 +60,7 @@ with gr.Blocks() as demo:
     out_file = gr.File(label="Download SVG")
     btn.click(None, style, [view_str, chains, hidden_style, timestamp], js="(style) => [document.getElementById('viewMatrixResult').value, document.getElementById('chains').value, style, Date.now()]")  #
     
-    timestamp.change(predict, [inp, style, view_str, chains], [html, out_file])
+    timestamp.change(predict, [inp, style, contour_level, view_str, chains], [html, out_file])
     # on change of chains trigger, rendering