improved design and background image

app.py

@@ -25,26 +25,26 @@ def add_bg_from_local(image_file):
         unsafe_allow_html=True
     )
 
-def header_white_bg(text, fontsize = 40, bold = True):
+def header_white(text, fontsize = 40, bold = True):
     st.markdown(
         f"""
-        <span style="background:rgba(255, 255, 255, 0.8); font-size:{fontsize}px; font-weight:{"bold" if bold else "normal"}; line-height: 2em">{text}</span>
+        <span style="display:inline-block; background:rgba(255, 255, 255, 1); outline: 1px solid; border-radius: 50px; padding: 10px; font-size:{fontsize}px; font-weight:{"bold" if bold else "normal"}; line-height: 2em">{text}</span>
         """,
         unsafe_allow_html=True
     )
 
-def header_red_bg(text, fontsize = 40, bold = True):
+def header_red(text, fontsize = 40, bold = True):
     st.markdown(
         f"""
-        <span style="background:rgba(184, 35, 35, 0.5); font-size:{fontsize}px; font-weight:{"bold" if bold else "normal"}; line-height: 1.25">{text}</span>
+        <span style="display:inline-block; background:rgba(255, 255, 255, 1); color:red; outline: 1px solid; border-radius: 30px; padding: 10px;  font-size:{fontsize}px; font-weight:{"bold" if bold else "normal"}">{text}</span>
         """,
         unsafe_allow_html=True
     )
 
-def header_green_bg(text, fontsize = 40, bold = True):
+def header_green(text, fontsize = 40, bold = True):
     st.markdown(
         f"""
-        <span style="background:rgba(26, 153, 58, 0.5); font-size:{fontsize}px; font-weight:{"bold" if bold else "normal"}; line-height: 1.25">{text}</span>
+        <span style="display:inline-block; background:rgba(255, 255, 255, 1); color:green; outline: 1px solid; border-radius: 30px; padding: 10px; font-size:{fontsize}px; font-weight:{"bold" if bold else "normal"}; line-height: 1.25">{text}</span>
         """,
         unsafe_allow_html=True
     )
@@ -115,14 +115,15 @@ def diagnose_health(file):
 def app():
 
     add_bg_from_local('assets/background.png')
-    header_white_bg(f'<span style="color:green">Plant</span><span style="color:orange">Dx</span><span style="color:green">: Diagnosis in a Snap!</span> ')
+    header_white(f'<span style="color:green">Plant</span><span style="color:orange">Dx</span><span style="color:green">: Diagnosis in a Snap!</span> ')
 
     # Upload image of plant
-    header_white_bg("Upload an image of your plant:", fontsize=32)
+    header_white("Upload an image of your plant below:", fontsize=32)
+    header_white("For best results, remove the leaf from the plant and take the image against a dark background.", fontsize=16, bold=False)
     uploaded_file = st.file_uploader("", type=["jpg", "jpeg", "png"])
 
     if uploaded_file:
-        header_white_bg("Preview of the selected image:", fontsize=28, bold=False)
+        header_white("Preview of the selected image:", fontsize=28, bold=False)
         st.image(uploaded_file)
 
     # Get diagnosis button
@@ -132,34 +133,34 @@ def app():
             results = diagnose_health(uploaded_file)
 
             if results[0][1] == 'healthy':
-                header_green_bg(f"We believe this is a healthy {results[0][0]} plant with {results[0][2]} confidence. Keep up the good work with proper watering, sunlight, and nutrients.", fontsize=32, bold=False)
+                header_green(f"We believe this is a healthy {results[0][0]} plant with {results[0][2]} confidence. Keep up the good work with proper watering, sunlight, and nutrients.", fontsize=32, bold=False)
             else:
-                header_red_bg(f"We believe this is an unhealthy {results[0][0]} plant with {results[0][1]}, with {results[0][2]} confidence. {results[0][3] if results [0][3] else ''}", fontsize=32, bold=False)
+                header_red(f"We believe this is an unhealthy {results[0][0]} plant with {results[0][1]}, with {results[0][2]} confidence. {results[0][3] if results [0][3] else ''}", fontsize=32, bold=False)
 
             if len(results) > 1:
-                header_white_bg("Other potential diagnoses: ", fontsize=24)
+                header_white("Other potential diagnoses: ", fontsize=24)
 
                 for p in range(1, len(results)):
                     if results[p][1] == 'healthy':
-                        header_white_bg(
+                        header_white(
                             f"A healthy {results[p][0]} plant, {results[p][2]} confidence.",
                             fontsize=20, bold=False)
                     else:
-                        header_white_bg(
+                        header_white(
                             f"An unhealthy {results[p][0]} plant with {results[p][1]}, {results[p][2]} confidence. {results[p][3] if results [p][3] else ''}",
                             fontsize=20, bold=False)
 
         else:
             st.warning("Please upload an image of your plant first")
 
-    # Create user profile button
-    if st.button("Create User Profile"):
-        st.subheader("User Profile")
-        # Prompt user to add their name and the plant they own
-        user_name = st.text_input("Enter your name:")
-        plant_name = st.text_input("Enter the plant you own:")
-        if user_name and plant_name:
-            st.success(f"User profile created for {user_name} with plant {plant_name}")
+    # # Create user profile button
+    # if st.button("Create User Profile"):
+    #     st.subheader("User Profile")
+    #     # Prompt user to add their name and the plant they own
+    #     user_name = st.text_input("Enter your name:")
+    #     plant_name = st.text_input("Enter the plant you own:")
+    #     if user_name and plant_name:
+    #         st.success(f"User profile created for {user_name} with plant {plant_name}")
 
 
 # Run Streamlit app

predictor.py

@@ -77,7 +77,7 @@ def predict(image_file):
         for j in range(topk):
             tmp_dct = {}
             label_indx = indices[0][j]
-            # print(label_indx)
+            print("index:", label_indx)
             class_name = class_names[label_indx]
             tmp_dct["predicted"] = class_name
             tmp_dct["probability"] = probs[0][j]