Update app.py

app.py

@@ -79,7 +79,16 @@ def greet(name):
                     )
     sf_output = [tokenizer.decode(g, skip_special_tokens=True, clean_up_tokenization_spaces=True).replace(" ","") for g in molecules]
     sf_output = list(set(sf_output))
-    return sf_output
+    input_plogp = plogp(input_sm)
+    plogp_improve = [plogp(i)-input_plogp for i in sm_output]
+    
+    
+    sim = [sim(i,input_sm) for i in sm_output]
+    
+    candidate_selfies = {"candidates": sf_output, "improvement": plogp_improve, "sim": sim}
+    data = pd.DataFrame(candidate_selfies)
+    
+    return data[(data['improvement']> 0) & (data['sim']>0.4)]