add legend

app.py

@@ -57,7 +57,7 @@ def molecule(input_pdb):
                 let config = { backgroundColor: "white" };
                 let viewer = $3Dmol.createViewer(element, config);
                 viewer.addModel(pdb, "pdb");
-                viewer.getModel(0).setStyle({}, {sphere:{color:"spectrum"}});
+                viewer.getModel(0).setStyle({}, {cartoon:{color:"spectrum"}});
                 viewer.zoomTo();
                 viewer.render();
                 viewer.zoom(0.8, 2000);
@@ -121,12 +121,22 @@ ex_seq = "QVQLVESGGGSVQAGGSLRLSCTASGYTIGPYCMGWFRQAPGGEREAVAAINMGGGITYYADSVKGRFTI
 
 h_chain_example = "EVQLVQSGPEVKKPGTSVKVSCKASGFTFMSSAVQWVRQARGQRLEWIGWIVIGSGNTNYAQKFQERVTITRDMSTSTAYMELSSLRSEDTAVYYCAAPYCSSISCNDGFDIWGQGTMVTVS"
 l_chain_example = "DVVMTQTPFSLPVSLGDQASISCRSSQSLVHSNGNTYLHWYLQKPGQSPKLLIYKVSNRFSGVPDRFSGSGSGTDFTLKISRVEAEDLGVYFCSQSTHVPYTFGGGTKLEIK"
-inputs = [gr.Textbox(lines=5, label="Heavy chain"),
-          gr.Textbox(lines=5, label="Light chain")
-]
-iface = gr.Interface(fn=pred_seq, 
-                     inputs=inputs,
-                     outputs=gr.HTML(),
-                     title="Antibody structure prediction")
-
-iface.launch()
+
+with gr.Blocks() as demo:
+
+    h_text = gr.Textbox(lines=5, label="Heavy chain")
+    l_text = gr.Textbox(lines=5, label="Light chain")
+    btn = gr.Button(value="Submit")
+    btn.click(pred_seq, inputs=[h_text, l_text], outputs=gr.HTML())
+
+    gr.Markdown("## Text Examples")
+    gr.Examples(
+        [[h_chain_example, l_chain_example]],
+        [h_text, l_text],
+        gr.HTML(),
+        pred_seq,
+        cache_examples=True,
+    )
+
+if __name__ == "__main__":
+    demo.launch()