Update app.py

app.py

@@ -117,8 +117,9 @@ def spr_sidebar():
     )
     return menu
 
+
 # Home Page Content
-def home_page():
+def home_page(selected_category):  # Accept selected_category as a parameter
     st.title("Disease Detection Web App")
     
     uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
@@ -151,29 +152,18 @@ def home_page():
                 # Implement Eye Disease prediction (similar to others)
                 pass
 
-# About Page Content
-def about_page():
-    st.header('About the Project')
-    st.write("""
-        This web app detects different diseases using machine learning models.
-        - **Brain Tumor Detection**: Classifies different types of brain tumors.
-        - **Alzheimer Detection**: Classifies stages of Alzheimer.
-        - **Skin Disease Classification**: Classifies types of skin lesions.
-        - **Eye Disease Classification**: TBD.
-    """)
-
 # Main Function to Run the App
 def main():
     selected_category = spr_sidebar()
     
     if selected_category == "Brain Tumor":
-        home_page()
+        home_page(selected_category)  # Pass selected_category here
     elif selected_category == "Alzheimer":
-        home_page()
+        home_page(selected_category)  # Pass selected_category here
     elif selected_category == "Skin Disease":
-        home_page()
+        home_page(selected_category)  # Pass selected_category here
     elif selected_category == "Eye Disease":
-        home_page()  # Eye Disease detection page
+        home_page(selected_category)  # Pass selected_category here
     elif selected_category == "About":
         about_page()
 
@@ -181,4 +171,3 @@ if __name__ == '__main__':
     main()
 
 
-