Update app.py

app.py

@@ -67,17 +67,15 @@ deep_model = NeuralNet(input_size, hidden_size, output_size)
 criterion = nn.BCELoss()
 optimizer = optim.Adam(deep_model.parameters(), lr=0.001)
 
-# Skip training loop here for real-time prediction
-
 # Title and description in Streamlit
 st.title("BERTO AI😊 : Personalized Diabetes Treatment Predictor")
 st.write("""
-BERTO  AI  is a personalized diabetes prediction tool built to assist users in better understanding their health risks. Berto honors the legacy 
-of one of our founders dad ,Roberto Ferrer who unfortunately succumbed to Type 2 Diabetes .
+BERTO AI is a personalized diabetes prediction tool built to assist users in better understanding their health risks. 
+Berto honors the legacy of one of our founder's dad, Roberto Ferrer, who unfortunately succumbed to Type 2 Diabetes.
 This application uses a deep learning model powered by PyTorch to predict whether you may have diabetes based on various health indicators. 
-We aim to provide insights and suggestions for personalizing diabetes treatment. The Application allows you to choose between two options ,Yes and No 
-which have been give binary values .Also there is a slider for various features which have ranges as indicated . Berto is part of DiabeTrek Health 
-mission to help manage ,prevent and provide personalized treatment to Type 2 Diabetic Patients.
+We aim to provide insights and suggestions for personalizing diabetes treatment. The Application allows you to choose between two options, Yes and No, 
+which have been given binary values. There is also a slider for various features, which have ranges as indicated. Berto is part of DiabeTrek Health's 
+mission to help manage, prevent, and provide personalized treatment to Type 2 Diabetic Patients.
 """)
 
 # Input form for user to enter health information with descriptions
@@ -94,11 +92,18 @@ HeartDiseaseorAttack = st.selectbox("History of Heart Disease or Attack (1: Yes,
 Age = st.slider("Age", 18, 100, 30)
 HighChol = st.selectbox("High Cholesterol (1: Yes, 0: No)", [0, 1])
 DiffWalk = st.selectbox("Difficulty Walking (1: Yes, 0: No)", [0, 1])
-BMI = st.slider("BMI (Body Mass Index)", 10.0, 50.0, 25.0)
 HighBP = st.selectbox("High Blood Pressure (1: Yes, 0: No)", [0, 1])
 GenHlth = st.slider("General Health (1=Excellent, 5=Poor)", 1, 5, 3)
 
-# Create a feature array from the inputs
+# Get height and weight input
+height_cm = st.slider("Height (in cm)", 100, 250, 170)
+weight_kg = st.slider("Weight (in kg)", 30, 200, 70)
+
+# Calculate BMI
+BMI = weight_kg / (height_cm / 100) ** 2
+st.write(f"Your BMI is: {BMI:.2f}")
+
+# Create a feature array from the inputs, including the calculated BMI
 user_input = np.array([[AnyHealthcare, Sex, Smoker, MentHlth, CholCheck, Stroke,
                         PhysHlth, HeartDiseaseorAttack, Age, HighChol, DiffWalk,
                         BMI, HighBP, GenHlth]])
@@ -120,6 +125,29 @@ if st.button("Predict"):
 
     # Display prediction result
     if prediction == 1:
-        st.success("The model predicts that you likely ** have diabetes**.")
+        st.success("The model predicts that you likely **have diabetes**.")
+        
+        # Provide tips for managing diabetes
+        st.subheader("Tips for Managing Diabetes")
+        st.markdown("""
+        - **Maintain a balanced diet**: Focus on eating whole grains, vegetables, lean protein, and healthy fats.
+        - **Exercise regularly**: Physical activity helps control your blood sugar.
+        - **Monitor blood sugar levels**: Regular monitoring helps you keep track of your glucose levels.
+        - **Take medications as prescribed**: Follow your doctor's instructions regarding medications.
+        - **Manage stress**: Chronic stress can raise blood sugar levels, so practice relaxation techniques.
+        - **Get regular check-ups**: Regular health check-ups are crucial for managing diabetes effectively.
+        """)
     else:
-        st.success("The model predicts that you are ** unlikely to get diabetes**.")
+        st.success("✅ Lower risk detected. Keep up the good work!")
+        
+        # Provide tips for maintaining low diabetes risk
+        st.subheader("Tips to Maintain Low Diabetes Risk")
+        st.markdown("""
+        - **Keep a healthy weight**
+        - **Exercise often**
+        - **Eat well-balanced meals**
+        - **Cut down on sugar and unhealthy fats**
+        - **Check your blood sugar**
+        - **Manage stress**
+        - **Get regular check-ups**
+        """)