Update app.py

app.py

@@ -57,7 +57,10 @@ def run_graph(input_message, history, user_details):
         return f"An error occurred while processing your request: {e}"
 
 
-def calculate_bmi(height, weight):
+def calculate_bmi(height, weight, gender):
+    """
+    Calculate BMI and determine the category based on height, weight, and gender.
+    """
     try:
         height_m = height / 100
         bmi = weight / (height_m ** 2)
@@ -69,14 +72,47 @@ def calculate_bmi(height, weight):
             status = "overweight"
         else:
             status = "obese"
-        return f"Your BMI is {bmi:.2f}, which is considered {status}."
-    except:
-        return "Invalid height or weight provided."
 
-def calculate_calories(age, weight, height, activity_level):
+        return bmi, status
+    except:
+        return None, "Invalid height or weight provided."
+
+
+def visualize_bmi(bmi):
+    """
+    Create a visualization for BMI, showing the user's value against standard categories.
+    """
+    categories = ["Underweight", "Normal Weight", "Overweight", "Obese"]
+    bmi_ranges = [18.5, 24.9, 29.9, 40]
+    x_pos = np.arange(len(categories))
+    user_position = min(len(bmi_ranges), sum(bmi > np.array(bmi_ranges)))
+
+    plt.figure(figsize=(8, 4))
+    plt.bar(x_pos, bmi_ranges, color=['blue', 'green', 'orange', 'red'], alpha=0.6, label="BMI Categories")
+    plt.axhline(y=bmi, color='purple', linestyle='--', linewidth=2, label=f"Your BMI: {bmi:.2f}")
+    plt.xticks(x_pos, categories)
+    plt.ylabel("BMI Value")
+    plt.title("BMI Visualization")
+    plt.legend()
+    plt.tight_layout()
+
+    plt_path = "bmi_chart.png"
+    plt.savefig(plt_path)
+    plt.close()
+
+    return plt_path
+
+
+def calculate_calories(age, weight, height, activity_level, gender):
+    """
+    Estimate daily calorie needs based on user inputs.
+    """
     try:
-        # Harris-Benedict Equation for calorie calculation
-        bmr = 10 * weight + 6.25 * height - 5 * age + 5  # Male (adjust for female by subtracting 161)
+        if gender.lower() == "male":
+            bmr = 10 * weight + 6.25 * height - 5 * age + 5
+        else:
+            bmr = 10 * weight + 6.25 * height - 5 * age - 161
+
         activity_multipliers = {
             "Sedentary": 1.2,
             "Lightly active": 1.375,
@@ -84,8 +120,9 @@ def calculate_calories(age, weight, height, activity_level):
             "Very active": 1.725,
             "Extra active": 1.9,
         }
-        calories = bmr * activity_multipliers[activity_level]
-        return f"Your estimated daily calorie needs are {calories:.2f} kcal."
+
+        calories = bmr * activity_multipliers.get(activity_level, 1.2)
+        return calories
     except Exception:
         return "Invalid inputs for calorie calculation."
 
@@ -97,6 +134,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         user_name = gr.Textbox(placeholder="Enter your name", label="Name")
         user_age = gr.Number(label="Age (years)", value=25, precision=0)
+        user_gender = gr.Dropdown(choices=["Male", "Female"], label="Gender", value="Male")
         user_weight = gr.Number(label="Weight (kg)", value=70, precision=1)
         user_height = gr.Number(label="Height (cm)", value=170, precision=1)
         activity_level = gr.Dropdown(
@@ -105,38 +143,27 @@ with gr.Blocks() as demo:
             value="Moderately active"
         )
 
-    # Chatbot Section
-    chatbot = gr.Chatbot(label="Chat with FIT.AI")
-    text_input = gr.Textbox(placeholder="Type your question here...", label="Your Question")
-    submit_button = gr.Button("Submit")
-    clear_button = gr.Button("Clear Chat")
-
-    # Examples
-    examples = gr.Examples(
-        examples=[
-            "Create a personalized workout plan for me",
-            "What should I eat to lose 5 kg in two months?",
-            "Suggest a yoga routine for beginners",
-            "How much water should I drink daily based on my weight?",
-        ],
-        inputs=text_input,
-    )
+    # Outputs
+    bmi_output = gr.Label(label="BMI Result")
+    calorie_output = gr.Label(label="Calorie Needs")
+    bmi_chart = gr.Image(label="BMI Chart")
 
-    # Chat Logic
-    def submit_message(message, history=[]):
-        user_details = {
-            "name": user_name.value,
-            "age": user_age.value,
-            "weight": user_weight.value,
-            "height": user_height.value,
-            "activity_level": activity_level.value
-        }
-        response = run_graph(message, history, user_details)
-        history.append(("User", message))
-        history.append(("FIT.AI", response))
-        return history, ""
+    # Actions
+    calculate_button = gr.Button("Calculate")
 
-    submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, text_input])
-    clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, text_input])
+    def calculate_metrics(age, weight, height, gender, activity_level):
+        bmi, status = calculate_bmi(height, weight, gender)
+        if bmi:
+            bmi_path = visualize_bmi(bmi)
+            calories = calculate_calories(age, weight, height, activity_level, gender)
+            return f"Your BMI is {bmi:.2f}, considered {status}.", f"Daily calorie needs: {calories:.2f} kcal", bmi_path
+        else:
+            return "Invalid inputs.", "", ""
+
+    calculate_button.click(
+        calculate_metrics,
+        inputs=[user_age, user_weight, user_height, user_gender, activity_level],
+        outputs=[bmi_output, calorie_output, bmi_chart]
+    )
 
 demo.launch(share=True)