add legend

app.py

@@ -80,8 +80,10 @@ def molecule(input_pdb):
     <div style="display: flex; justify-content: space-between; margin-top: 5px;">
         <span>Low Energy</span>
     </div>
+    <div style="display: flex; justify-content: space-between; margin-top: 5px;">
+        <a href="data:application/octet-stream;base64,{base64_content}" download="structure.pdb">Download File</a>
+    </div>
     </div>
-    <a href="data:application/octet-stream;base64,{base64_content}" download="structure.pdb">Download File</a>
     """
 
 
@@ -126,21 +128,30 @@ h_chain_example = "EVQLVQSGPEVKKPGTSVKVSCKASGFTFMSSAVQWVRQARGQRLEWIGWIVIGSGNTNYA
 l_chain_example = "DVVMTQTPFSLPVSLGDQASISCRSSQSLVHSNGNTYLHWYLQKPGQSPKLLIYKVSNRFSGVPDRFSGSGSGTDFTLKISRVEAEDLGVYFCSQSTHVPYTFGGGTKLEIK"
 
 with gr.Blocks() as demo:
-    header = gr.Label('Antibody Structure Prediction')
+    gr.Markdown('## Antibody Structure Prediction')
+    gr.Markdown("## Sequences Examples")
+    html = """
+    <p>
+       H-chain: <tt>QVQLKESGPGLVAPSQSLSITCTVSGFSLSSYGVSWVRQPPGKGLEWLGVIWGDGSTNYHPNLMSRLSISKDISKSQVLFKLNSLQTDDTATYYCVTLDYWGQGTSVTVSS</tt>
+       <br/>
+       L-chain: <tt>DVVMTQTPLSLPVSLGDQASISCRSSQSLVHRNGNTYLHWYLQKPGQSPKLLIYKVSNRFSGVPDRFSGSGSGTDFTLKISRVEAEDLGLYFCFQTTYVPNTFGGGTKLEIK</tt>
+    </p>
+
+    <p>
+       H-chain: <tt>EVQLLESGGGLVQPGGSLRLSCAASGFTFSLYWMGWVRQAPGKGLEWVSSISSSGGVTPYADSVKGRFTISRDNSKNTLYLQMNSLRAEDTAVYYCAKLGELGWFDPWGQGTLVTVSS</tt>
+       <br/>
+       L-chain: <tt>DIQMTQSPSSLSASVGDRVTITCRASQGISSYLNWYQQKPGKAPKLLIYYASNLQNGVPSRFSGSGSGTDFTLTISSLQPEDFATYYCQQSYSTPLTFGGGTKVEIK</tt>
+    </p>
+
+    <p>
+       H-chain: <tt>EVQLVQSGPEVKKPGTSVKVSCKASGFTFMSSAVQWVRQARGQRLEWIGWIVIGSGNTNYAQKFQERVTITRDMSTSTAYMELSSLRSEDTAVYYCAAPYCSSISCNDGFDIWGQGTMVTVS</tt>
+       <br/>
+       L-chain: <tt>DVVMTQTPFSLPVSLGDQASISCRSSQSLVHSNGNTYLHWYLQKPGQSPKLLIYKVSNRFSGVPDRFSGSGSGTDFTLKISRVEAEDLGVYFCSQSTHVPYTFGGGTKLEIK</tt>
+    </p>
+    """
     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()