Update demo.py

demo.py

@@ -341,7 +341,7 @@ def cSSStructureSampleDemo(style,resn):
     display(output,style,resn)
 
 #  Merging Symmetry and Substructure
-def mSSubstructureSample(composedCondtioner,output):
+def mSSubstructureSample(protein,composedCondtioner,output):
     protein, trajectories = chroma.sample(
         protein_init=protein,
         conditioner=composedCondtioner,
@@ -373,6 +373,6 @@ def mSSubstructureSampleDemo(style,resn):
     composed_cond = conditioners.ComposedConditioner([substruct_conditioner, c_symmetry])
 
     if st.sidebar.button("Run Code with Button",key="substructure"):
-        mSSubstructureSample(composed_cond,output)
+        mSSubstructureSample(protein,composed_cond,output)
     
     display(output,style,resn)