Update pages/Brain.py

pages/Brain.py

@@ -19,6 +19,8 @@ text_model = BioGptForCausalLM.from_pretrained("microsoft/biogpt")
 tokenizer = BioGptTokenizer.from_pretrained("microsoft/biogpt")
 labels_path = 'brain_labels.json'
 from captum.attr import DeepLift , visualization
+import os
+import base64
 
 with open(labels_path) as json_data:
     idx_to_labels = json.load(json_data)
@@ -86,7 +88,51 @@ class FineTuneModel(pl.LightningModule):
    
     
     
-    
+# Get the current working directory
+current_dir = os.getcwd()
+
+# Construct the absolute path to the logo.png file
+logo_path = os.path.join(current_dir, "logo.png")
+
+with open(logo_path, "rb") as f:
+    image_data = f.read()
+    image_base64 = base64.b64encode(image_data).decode("utf-8")
+
+# Add custom CSS for the header
+header_css = """
+<style>
+.header {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    padding: 20px;
+    font-size: 18px;
+}
+
+.header img {
+    margin-right: 10px;
+    width: 80px;
+    height: 50px;
+}
+
+.header p {
+    font-size: 14px;
+}
+</style>
+"""
+
+# Render the custom CSS
+st.markdown(header_css, unsafe_allow_html=True)
+
+# Render the header
+header_html = f"""
+<div class="header">
+    <img src='data:image/jpeg;base64,{image_base64}' alt="Logo"/>
+    <p>Disclaimer: This web app is for demonstration purposes only and not intended for commercial use. Contact: contact@1001epochs.co.uk for full solution.</p>
+</div>
+"""
+
+st.markdown(header_html, unsafe_allow_html=True)
 
 st.markdown("<h1 style='text-align: center; '>Brain Tumor Diagnosis</h1>",unsafe_allow_html=True)
 
@@ -125,32 +171,6 @@ with st.container():
                     input_text = f"Patient has {predicted_label} and is advised to take the following medicines:"
                     with st.spinner('Generating Text'):
                         generator(input_text, max_length=300, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=1)
-                    st.markdown(generator(input_text, max_length=300, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=1)[0]['generated_text'])
-                
-                    
-            
-            
-            
-            
-
-    
-        
-        
-    
-        
-        
+                    st.markdown(generator(input_text, max_length=300, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=1)[0]['generated_text'])        
     else:
-        st.success("Please upload an image file 🧠")
-        
-        
-    
-    
-    
-## Model Explainibilty Dashboard using Captum 
-
-
-
-
-    
-
-
+        st.success("Please upload an image file 🧠")