Update app.py

app.py

@@ -4,6 +4,21 @@ from transformers.models.esm.openfold_utils.feats import atom14_to_atom37
 from proteins_viz import *
 import gradio as gr
 import spaces
+from gradio_molecule3d import Molecule3D
+
+reps =    [
+    {
+      "model": 0,
+      "chain": "",
+      "resname": "",
+      "style": "stick",
+      "color": "whiteCarbon",
+      "residue_range": "",
+      "around": 0,
+      "byres": False,
+      "visible": False
+    }
+]
 
 def read_mol(molpath):
     with open(molpath, "r") as fp:
@@ -114,7 +129,7 @@ def fold_protein(test_protein):
         f.write("".join(pdb))
     image = take_care("output_structure.pdb")
     html = molecule("output_structure.pdb")
-    return image, html
+    return image, html, "output_structure.pdb"
 
 iface = gr.Interface(
     title="everything-ai-proteinfold",
@@ -125,7 +140,7 @@ iface = gr.Interface(
             lines=5,
             value=f"Paste or write amino-acidic sequence here",
         ),
-    outputs=[gr.Image(label="Protein static image"), gr.HTML(label="Protein 3D model")], 
+    outputs=[gr.Image(label="Protein static image"), gr.HTML(label="Protein 3D model"), Molecule3D(label="Molecular 3D model", reps=reps)], 
     examples=[
         "MVHLTPEEKSAVTALWGKVNVDEVGGEALGRLLVVYPWTQRFFESFGDLSTPDAVMGNPKVKAHGKKVLGAFSDGLAHLDNLKGTFATLSELHCDKLHVDPENFRLLGNVLVCVLAHHFGKEFTPPVQAAYQKVVAGVANALAHKYH",
         "MTEYKLVVVGAGGVGKSALTIQLIQNHFVDEYDPTIEDSYRKQVVIDGETCLLDILDTAGQEEYSAMRDQYMRTGEGFLCVFAINNTKSFEDIHQYREQIKRVKDSDDVPMVLVGNKCDLAARTVESRQAQDLARSYGIPYIETSAKTRQGVEDAFYTLVREIRQHKLRKLNPPDESGPGCMSCKCVLS",