gr banner

app.py

@@ -1,15 +1,16 @@
 import os
 
 import gradio as gr
-from gradio_banner import Banner
 import numpy as np
 import plotly.graph_objs as go
 from datasets import load_dataset
 from pymatgen.analysis.phase_diagram import PDPlotter, PhaseDiagram
 from pymatgen.core import Composition, Structure
 from pymatgen.core.composition import Composition
-from pymatgen.entries.computed_entries import (ComputedStructureEntry,
-                                               GibbsComputedStructureEntry)
+from pymatgen.entries.computed_entries import (
+    ComputedStructureEntry,
+    GibbsComputedStructureEntry,
+)
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
@@ -36,8 +37,13 @@ del dataset
 
 
 def create_phase_diagram(
-    elements, max_e_above_hull, color_scheme, plot_style, functional, finite_temp,
-    **kwargs
+    elements,
+    max_e_above_hull,
+    color_scheme,
+    plot_style,
+    functional,
+    finite_temp,
+    **kwargs,
 ):
     # Split elements and remove any whitespace
     element_list = [el.strip() for el in elements.split("-")]
@@ -125,15 +131,21 @@ plot_style_dropdown = gr.Dropdown(choices=["2D", "3D"], label="Plot Style")
 functional_dropdown = gr.Dropdown(choices=["PBE", "PBESol", "SCAN"], label="Functional")
 finite_temp_toggle = gr.Checkbox(label="Enable Finite Temperature Estimation")
 
-banner = Banner(value="This application uses energy correction schemes directly" + 
-                " from the data providers (Alexandria, MP) and has the 2020 MP"+
-                " Compatibility scheme applied to OQMD. However, because we did"+
-                " not directly apply the compatibility schemes to Alexandria, MP"+
-                " we have noticed discrepencies in the data. While the correction"+
-                " scheme will be standardized in a soon to be released update, for"+
-                " now please take caution when analyzing the results of this"+
-                " application.", 
-                variant="warning")
+warning_message = "This application uses energy correction schemes directly"
+warning_message += " from the data providers (Alexandria, MP) and has the 2020 MP"
+warning_message += " Compatibility scheme applied to OQMD. However, because we did"
+warning_message += " not directly apply the compatibility schemes to Alexandria, MP"
+warning_message += " we have noticed discrepencies in the data. While the correction"
+warning_message += " scheme will be standardized in a soon to be released update, for"
+warning_message += " now please take caution when analyzing the results of this"
+warning_message += " application."
+
+with gr.Blocks() as banner:
+    message = gr.HTML(
+        '<div class="alert"><span class="closebtn" onclick="this.parentElement.style.display="none";">&times;</span>{}</div>'.format(
+            warning_message
+        )
+    )
 
 # Create Gradio interface
 iface = gr.Interface(
@@ -145,7 +157,7 @@ iface = gr.Interface(
         plot_style_dropdown,
         functional_dropdown,
         finite_temp_toggle,
-        banner,
+        message,
     ],
     outputs=gr.Plot(label="Phase Diagram"),
     title="LeMaterial - Phase Diagram Viewer",

requirements.txt

@@ -1,5 +1,4 @@
 plotly
 pymatgen
 gradio==5.8.0
-datasets==3.1.0
-gradio_banner
+datasets==3.1.0