Update app.py

app.py

@@ -46,27 +46,29 @@ if uploaded_file is not None:
         predicted_class = np.argmax(prediction, axis=1)[0]  # Get the class index
         class_labels = ['0', 'MR', 'MRMS', 'MS', 'R', 'S']  # Update based on your classes
         
+        st.header("Predicted Severity Class")
+        
         if predicted_class == 0:
-            st.write("Predicted Severity Class: Healthy")
-            st.write("**Advice:** The leaf appears healthy. Continue monitoring as needed.")
+            st.markdown('<p style="color: green; font-size: 20px;">Healthy</p>', unsafe_allow_html=True)
+            st.write("The leaf appears healthy. There is no immediate action required. Continue monitoring as needed.")
         elif predicted_class == 1:
-            st.write("Predicted Severity Class: Mild Rust (MR)")
-            st.write("**Advice:** Apply fungicides to control mild rust development.")
+            st.markdown('<p style="color: orange; font-size: 20px;">Mild Rust (MR)</p>', unsafe_allow_html=True)
+            st.write("Mild rust detected. Applying fungicides will help control further spread.")
         elif predicted_class == 2:
-            st.write("Predicted Severity Class: Moderate Rust (MRMS)")
-            st.write("**Advice:** Monitor regularly and treat with appropriate fungicides.")
+            st.markdown('<p style="color: #FFA500; font-size: 20px;">Moderate Rust (MRMS)</p>', unsafe_allow_html=True)
+            st.write("Moderate rust detected. Monitor regularly and treat with fungicides.")
         elif predicted_class == 3:
-            st.write("Predicted Severity Class: Severe Rust (MS)")
-            st.write("**Advice:** Apply fungicides promptly and continue monitoring.")
+            st.markdown('<p style="color: #FF4500; font-size: 20px;">Severe Rust (MS)</p>', unsafe_allow_html=True)
+            st.write("Severe rust detected. Prompt fungicide application and continued monitoring are recommended.")
         elif predicted_class == 4:
-            st.write("Predicted Severity Class: Very Severe Rust (R)")
-            st.write("**Advice:** Implement intensive control measures and frequent monitoring.")
+            st.markdown('<p style="color: red; font-size: 20px;">Very Severe Rust (R)</p>', unsafe_allow_html=True)
+            st.write("Very severe rust detected. Intensive control measures and frequent monitoring are required.")
         elif predicted_class == 5:
-            st.write("Predicted Severity Class: Extremely Severe Rust (S)")
-            st.write("**Advice:** Apply aggressive control strategies and seek expert advice.")
+            st.markdown('<p style="color: darkred; font-size: 20px;">Extremely Severe Rust (S)</p>', unsafe_allow_html=True)
+            st.write("Extremely severe rust detected. Apply aggressive control strategies and seek expert advice.")
         
         confidence = np.max(prediction) * 100
         st.write(f"**Confidence Level:** {confidence:.2f}%")
 
 # Footer
-st.sidebar.info("MPHIL Final Year Project By Mr. Asim Khattak")
+st.info("MPHIL Final Year Project By Mr. Asim Khattak")