Update app.py

app.py

@@ -139,7 +139,10 @@ def opt_process(opt_input):
     results = data[(data['improvement']> 0) & (data['sim']>0.4)]
     opt_output  = results["candidates"].tolist()
     opt_output_imp = results["improvement"].tolist()
+    opt_output_imp = [str(i) for i in opt_output_imp]
     opt_output_sim = results["sim"].tolist()
+    opt_output_sim = [str(i) for i in opt_output_sim]
+    
 
     smis = [sf.decoder(i) for i in opt_output]
     mols = []
@@ -153,18 +156,7 @@ def opt_process(opt_input):
         subImgSize=(200,200),
         legends=['' for x in mols]
     )
-
-    return opt_input_img, "\n".join(opt_output), opt_output_imp, opt_output_sim, opt_output_img
-# examples = [
-            
-#     ['[C][C][=Branch1][C][=O][N][C][C][O][C][C][O][C][C][O][C][C][Ring1][N]'],['[C][C][S][C][C][S][C][C][C][S][C][C][S][C][Ring1][=C]']
-
-# ]
-
-
-
-# iface = gr.Interface(fn=greet, inputs="text", outputs="numpy", title="Molecular Language Model as Multi-task Generator",examples=examples)
-# iface.launch()
+    return opt_input_img, "\n".join(opt_output), "\n".join(opt_output_imp), "\n".join(opt_output_sim), opt_output_img
 
 with gr.Blocks() as demo:
     gr.Markdown("# MolGen: Molecular Language Model as Multi-task Generator")
@@ -178,7 +170,7 @@ with gr.Blocks() as demo:
 
                 with gr.Column():
                     gen_output = gr.Textbox(label="Generation Results", lines=5, placeholder="")
-                    gen_output_image = gr.Image(type='pil', label="Visualization")
+                    gen_output_image = gr.Image(label="Visualization")
        
             gr.Examples(
                 examples=[["[C][=C][C][=C][C][=C][Ring1][=Branch1]"], 
@@ -202,7 +194,7 @@ with gr.Blocks() as demo:
                     opt_output_imp = gr.Textbox(label="Optimization Property Improvements", lines=3, placeholder="")
                     opt_output_sim = gr.Textbox(label="Similarity", lines=3, placeholder="")
                     opt_output_img = gr.Textbox(label="Output Visualization", lines=3, placeholder="")
-                    opt_output_img = gr.Image(type='pil', label="Output Visualization")
+                    opt_output_img = gr.Image(label="Output Visualization")
 
             gr.Examples(
                 examples=[