use blue/purple theme

folding_studio_demo/app.py

@@ -13,6 +13,7 @@ from folding_studio_demo.correlate import (
     make_correlation_plot,
 )
 from folding_studio_demo.predict import predict, predict_comparison
+from folding_studio_demo.config import BLUE, PURPLE
 
 logger = logging.getLogger(__name__)
 
@@ -102,7 +103,13 @@ def simple_prediction(api_key: str) -> None:
         with gr.Column():
             sequence = sequence_input()
 
-    predict_btn = gr.Button("Predict")
+    predict_btn = gr.Button(
+        "Predict",
+        elem_classes="gradient-button",
+        elem_id="predict-btn",
+        variant="primary",
+        # css=f".gradio-container #predict-btn {{background: linear-gradient(90deg, {BLUE}, {PURPLE});}}",
+    )
 
     with gr.Row():
         mol_output = Molecule3D(label="Protein Structure", reps=MOLECULE_REPS)
@@ -134,7 +141,13 @@ def model_comparison(api_key: str) -> None:
         with gr.Column():
             sequence = sequence_input()
 
-    predict_btn = gr.Button("Compare Models")
+    predict_btn = gr.Button(
+        "Compare Models",
+        elem_classes=["gradient-button"],
+        elem_id="compare-models-btn",
+        variant="primary",
+        # css=f".gradio-container #compare-models-btn {{background: linear-gradient(90deg, {BLUE}, {PURPLE});}}"
+    )
 
     with gr.Row():
         mol_outputs = Molecule3D(
@@ -193,7 +206,12 @@ def create_correlation_tab():
 
     gr.Markdown("# Prediction and correlation")
     with gr.Row():
-        fake_predict_btn = gr.Button("Predict structures of all complexes")
+        fake_predict_btn = gr.Button(
+            "Predict structures of all complexes",
+            elem_classes="gradient-button",
+            variant="primary",
+            # css=f".gradio-container #fake-predict-btn {{background: linear-gradient(90deg, {BLUE}, {PURPLE});}}",
+        )
     with gr.Row():
         prediction_dataframe = gr.Dataframe(label="Predicted Structures Data")
     with gr.Row():
@@ -236,7 +254,12 @@ def create_correlation_tab():
 
 
 def __main__():
-    with gr.Blocks(title="Folding Studio Demo") as demo:
+
+    theme = gr.themes.Ocean(
+        primary_hue="blue",
+        secondary_hue="purple",
+    )
+    with gr.Blocks(theme=theme, title="Folding Studio Demo") as demo:
         gr.Markdown(
             """
             # Folding Studio: Harness the Power of Protein Folding 🧬