Update app.py

app.py

@@ -328,12 +328,20 @@ with gr.Blocks(title="FoldMark", css=custom_css) as demo:
         with gr.Row():
             view3d = Molecule3D(label="3D Visualization(Gray: Unwatermarked; Cyan: Watermarked)", reps=reps)
             
-        legend = gr.Markdown("""
-        **Color Legend:**
-
-        - <span style="color:grey">Gray: Unwatermarked Structure</span>
-        - <span style="color:cyan">Cyan: Watermarked Structure</span>
-        """)
+        # legend = gr.Markdown("""
+        # **Color Legend:**
+
+        # - <span style="color:grey">Gray: Unwatermarked Structure</span>
+        # - <span style="color:cyan">Cyan: Watermarked Structure</span>
+        # """)
+
+        legend = gr.HTML("""
+            <div>
+              <strong>Color Legend:</strong><br>
+              - <span style="color:grey;">Gray: Unwatermarked Structure</span><br>
+              - <span style="color:cyan;">Cyan: Watermarked Structure</span>
+            </div>
+            """)
         
         with gr.Row():
             cif_file = gr.File(label="Download CIF File")