Update app.py

app.py

@@ -29,6 +29,13 @@ def main():
 
     uploaded_file = st.file_uploader('Upload an MRI image...', type=['jpg', 'png', 'jpeg'])
 
+    example_images = {
+        "MILD DEMENTED": "mild_468_0_4983.jpg",
+        "MODERATE DEMENTED": "moderate_2_0_72.jpg",
+        "NON DEMENTED": "non_61.jpg",
+        "VERY MILD DEMENTED": "verymild_37_0_2606.jpg"
+    }
+
     if uploaded_file is not None:
         predicted_class = predict_image_class(uploaded_file)
         
@@ -37,25 +44,22 @@ def main():
         st.markdown(f"<div style='text-align: center; font-size: 30px;'>Predicted Alzheimer's stage is: <b>{class_names[predicted_class]}</b></div>", unsafe_allow_html=True)
         st.image(uploaded_file, use_column_width=False, width=300)
     else:
-        # Display four example images if no image is uploaded
-        example_images = ["mild_468_0_4983.jpg", "moderate_2_0_72.jpg", "non_61.jpg", "verymild_37_0_2606.jpg"]
-        col1, col2, col3, col4 = st.columns(4)
-        
-        with col1:
-            image1 = Image.open(example_images[0])
-            st.image(image1, caption="Example 1", use_column_width=True)
+        st.write("Or select an example image below:")
+        selected_example = st.selectbox("Select an example image:", list(example_images.keys()))
+
+        if selected_example:
+            example_image_path = example_images[selected_example]
+            example_image = Image.open(example_image_path)
+            
+            # Display the selected example image
+            st.image(example_image, caption=selected_example, use_column_width=True)
             
-        with col2:
-            image2 = Image.open(example_images[1])
-            st.image(image2, caption="Example 2", use_column_width=True)
+            # Predict based on the selected example image
+            predicted_class = predict_image_class(example_image_path)
             
-        with col3:
-            image3 = Image.open(example_images[2])
-            st.image(image3, caption="Example 3", use_column_width=True)
+            class_names = ["MILD DEMENTED", "MODERATE DEMENTED", "NON DEMENTED", "VERY MILD DEMENTED"]
             
-        with col4:
-            image4 = Image.open(example_images[3])
-            st.image(image4, caption="Example 4", use_column_width=True)
+            st.markdown(f"<div style='text-align: center; font-size: 30px;'>Predicted Alzheimer's stage is: <b>{class_names[predicted_class]}</b></div>", unsafe_allow_html=True)
 
 if __name__ == '__main__':
     main()