Update app.py

app.py

@@ -16,13 +16,9 @@ def predict(input_mol, view_str, chains):
     chain_str = ""
     chain_dict = json.loads(chains)
 
-    print(chain_dict)
-    print(type(chain_dict))
-
     # sort keys in dict and add colors to chain_str
     for chain in sorted(chain_dict.keys()):
-        chain_str += f" {chain_dict[chain]}"
-    print(chain_str, "chainstring")
+        chain_str += f" '{chain_dict[chain]}'"
     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")
 
     #read content of file