Updated UI

app.py

@@ -60,13 +60,12 @@ def main():
     uploaded_file = st.sidebar.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
     if uploaded_file is not None:
-        # Load model
-        model = load_model()
-        
         # Display uploaded image
         image = Image.open(io.BytesIO(uploaded_file.read()))
         st.image(image, caption="Uploaded Image", use_column_width=True)
-        processed_image = preprocess_image(image)
+        
+        # Load model
+        model = load_model()
         
         # Predict button
         predict_button = st.sidebar.button("Predict", key="predict_button")
@@ -78,6 +77,8 @@ def main():
         if predict_button:
             # Display processing message with spinner
             with st.spinner("Please wait...Processing the image and predicting..."):
+                # Preprocess image
+                processed_image = preprocess_image(image)
                 
                 # Classify image
                 predicted_class, confidence_score = predict(model, image)
@@ -85,17 +86,17 @@ def main():
                 # Explain image classification
                 explanation_image = explain_image(processed_image, model)
                 
-                # Display explanation image
-                st.image(explanation_image, caption="Explanation Image", use_column_width=True)
+                # Display images side by side
+                col1, col2 = st.columns(2)
+                with col1:
+                    st.image(image, caption="Uploaded Image", use_column_width=True)
+                with col2:
+                    st.image(explanation_image, caption="Explanation Image", use_column_width=True)
                 
                 # Display prediction
                 st.subheader("Prediction:")
-                # Create a table for prediction results
-                prediction_table = pd.DataFrame({
-                    "Predicted Class": [predicted_class],
-                    "Confidence": [f"{confidence_score}%"]
-                })
-                st.table(prediction_table)
+                st.write(f"Predicted Class: {predicted_class}")
+                st.write(f"Confidence Score: {confidence_score}%")
 
 if __name__ == "__main__":
     main()