Update app.py

app.py

@@ -9,7 +9,64 @@ labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
 model = tf.keras.models.load_model('classify_model.h5')
 
 # Customized Streamlit layout
-# (Your existing layout code remains unchanged)
+# Customized Streamlit styles
+st.markdown(
+    """
+    <style>
+        body {
+            color: #333333;
+            background-color: #f9f9f9;
+            font-family: 'Helvetica', sans-serif;
+        }
+        .st-bb {
+            padding: 0rem;
+        }
+        .st-ec {
+            color: #666666;
+        }
+        .st-ef {
+            color: #666666;
+        }
+        .st-ei {
+            color: #333333;
+        }
+        .st-dh {
+            font-size: 36px;
+            font-weight: bold;
+            color: #4CAF50;
+            text-align: center;
+            margin-bottom: 20px;
+        }
+        .st-gf {
+            background-color: #4CAF50;
+            color: white;
+            padding: 15px 30px;
+            font-size: 18px;
+            border: none;
+            border-radius: 8px;
+            cursor: pointer;
+            transition: background-color 0.3s;
+        }
+        .st-gf:hover {
+            background-color: #45a049;
+        }
+        .st-gh {
+            text-align: center;
+            font-size: 24px;
+            font-weight: bold;
+            margin-bottom: 20px;
+        }
+        .st-logo {
+            max-width: 100%;
+            height: auto;
+            margin: 20px auto;
+            display: block;
+        }
+    </style>
+    """,
+    unsafe_allow_html=True,
+)
+
 
 # Logo
 st.image("https://ecoclimsolutions.files.wordpress.com/2024/01/rmcai-removebg.png?resize=48%2C48")
@@ -72,10 +129,6 @@ elif mode == "Train Mode":
             # Label input
             user_label = st.selectbox("Select the correct label", labels)
 
-            # Train button
-           # Inside the "Train Mode" section
-
-
             # Train button
             if st.button('Train Model'):
                 # Update the model with the user-provided image and label
@@ -94,3 +147,5 @@ elif mode == "Train Mode":
                 model.fit(dataset, epochs=1)  # You might want to adjust the number of epochs
         
                 st.success(f'Model has been trained with the new image and label.')
+        except Exception as e:
+            st.error(f"An error occurred: {e}. Please contact us EcoClim Solutions at EcoClimSolutions.wordpress.com.")