Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from PIL import Image
 import numpy as np
 import torch
-from transformers import ViTForImageClassification, pipeline
+from transformers import ViTForImageClassification
 
 # Load the Hugging Face model for ECG classification
 model_name = "adilsaleem/ecg-image-multilabel-classification"
@@ -13,7 +13,6 @@ except Exception as e:
     st.error(f"Error loading model: {str(e)}")
     st.stop()
 
-
 # CSS for styling
 st.markdown(
     """
@@ -21,12 +20,17 @@ st.markdown(
         body {
             background-color: #F5F7FA;  /* Light background color */
             font-family: 'Arial', sans-serif;
+            display: flex;
+            flex-direction: column;
+            height: 100vh;
+            margin: 0;
         }
         .title {
             font-size: 40px;
             text-align: center;
             color: #2980B9;  /* Beautiful blue color */
             font-weight: bold;  /* Bold text */
+            margin: 20px 0;
         }
         .subtitle {
             font-size: 20px;
@@ -35,14 +39,19 @@ st.markdown(
             margin-bottom: 20px;
         }
         .footer {
-            background-color: #2980B9;  /* Footer color */
+            background-color: #2C3E50;  /* Darker blue for footer */
             color: white;  /* Text color in footer */
             text-align: center;
-            padding: 10px 0;  /* Padding for footer */
-            position: relative;
+            padding: 15px;  /* Padding for footer */
+            position: sticky;
             bottom: 0;
-            left: 0;
-            right: 0;
+            width: 100%;
+            font-size: 14px;  /* Font size for footer */
+        }
+        .about {
+            margin: 20px 0;
+            text-align: center;
+            color: #34495E;  /* Darker shade for about section */
         }
     </style>
     """,
@@ -52,6 +61,19 @@ st.markdown(
 # App Title
 st.markdown('<div class="title">ECG READER</div>', unsafe_allow_html=True)
 
+# About Us Section in Expander
+with st.expander("About the App", expanded=True):
+    st.write("""
+        This ECG Reader application uses AI to analyze ECG images and provide insights into heart health.
+        It helps users determine whether their ECG results are clear or unclear.
+        Developed by Hamaad Ayub Khan.
+    """)
+    st.write("Connect with me:")
+    st.markdown('- [Instagram](https://www.instagram.com/hamaadayubkhan)', unsafe_allow_html=True)
+    st.markdown('- [GitHub](https://github.com/hakgs1234)', unsafe_allow_html=True)
+    st.markdown('- [My Email](mailto:[email protected])', unsafe_allow_html=True)
+    st.markdown('- [Facebook](https://www.facebook.com/hamaadayubkhan)', unsafe_allow_html=True)
+
 def preprocess_image(image):
     try:
         image = image.convert("RGB")
@@ -86,7 +108,6 @@ def get_ecg_explanation(status):
     }
     return explanations.get(status, "No explanation available.")
 
-
 uploaded_file = st.file_uploader("Select an ECG image", type=["jpg", "jpeg", "png"])
 
 if uploaded_file:
@@ -105,22 +126,6 @@ if uploaded_file:
     except Exception as e:
         st.error(f"Error uploading or displaying image: {str(e)}")
 
-
-# About Us Section
-    with st.expander("About the App"):st.write("""
-        This ECG Reader application uses AI to analyze ECG images and provide insights into heart health.
-        It helps users determine whether their ECG results are clear or unclear.
-        
-        Developed by Hamaad Ayub Khan.
-        
-        Connect with me:
-        - [Instagram](https://www.instagram.com/hamaadayubkhan)
-        - [GitHub](https://github.com/hakgs1234)
-        - Email: [My Email](mailto:[email protected])
-        - [Facebook](https://www.facebook.com/hamaadayubkhan)
-    """)
-
-
 # Footer
 st.markdown(
     """
@@ -130,4 +135,4 @@ st.markdown(
     </div>
     """,
     unsafe_allow_html=True,
-)
+)