Update app.py

app.py

@@ -114,81 +114,95 @@ def calculate_calories(age, weight, height, activity_level, gender):
 with gr.Blocks() as demo:
     gr.Markdown("<strong>FIT.AI - Your Fitness and Wellbeing Coach</strong>")
 
-    # User Info
-    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(
-            choices=["Sedentary", "Lightly active", "Moderately active", "Very active", "Extra active"],
-            label="Activity Level",
-            value="Moderately active"
-        )
-
-    # Outputs
-    bmi_output = gr.Label(label="BMI Result")
-    calorie_output = gr.Label(label="Calorie Needs")
-    bmi_chart = gr.Image(label="BMI and Calorie Chart")
-
-    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")
-
-    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,
-            "gender": user_gender.value
-        }
-        bmi, status = calculate_bmi(user_details['height'], user_details['weight'])
-        calories = calculate_calories(
-            user_details['age'], user_details['weight'], user_details['height'], user_details['activity_level'], user_details['gender']
-        )
-        chart_path = visualize_bmi_and_calories(bmi, calories)
-
-        # Append metrics to response with actionable advice
-        personalized_metrics = (
-            f"Based on your metrics:\n"
-            f"- BMI: {bmi:.2f} ({status})\n"
-            f"- Daily Caloric Needs: {calories:.2f} kcal\n"
-            f"\nAdvice: "
-        )
-        if status == "underweight":
-            personalized_metrics += "Your BMI indicates you are underweight. Focus on consuming more calories from nutrient-dense foods, including lean proteins, healthy fats, and whole grains. Consider strength training exercises to build muscle mass."
-        elif status == "normal weight":
-            personalized_metrics += "Your BMI is in the normal range. Maintain your weight by balancing calorie intake with regular physical activity. Incorporate a mix of cardio and strength training for overall fitness."
-        elif status == "overweight":
-            personalized_metrics += "Your BMI indicates you are overweight. Aim for a calorie deficit by reducing portion sizes and increasing physical activity. Focus on a diet rich in vegetables, lean protein, and healthy fats."
-        else:
-            personalized_metrics += "Your BMI indicates you are obese. Consult a healthcare provider for a personalized weight loss plan. Start with small, sustainable changes, such as reducing sugary foods and increasing daily physical activity."
-
-        response = personalized_metrics
-        history.append((f"User", message))
-        history.append(("FIT.AI", response))
-        return history, chart_path
-
-    submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, bmi_chart])
-    clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, bmi_chart])
-
-    # Actions
-    calculate_button = gr.Button("Calculate")
-
-    def calculate_metrics(age, weight, height, gender, activity_level):
-        bmi, status = calculate_bmi(height, weight)
-        bmi_path = visualize_bmi_and_calories(bmi, calculate_calories(age, weight, height, activity_level, gender))
-        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
-
-    calculate_button.click(
-        calculate_metrics,
-        inputs=[user_age, user_weight, user_height, user_gender, activity_level],
-        outputs=[bmi_output, calorie_output, bmi_chart]
-    )
+    with gr.Tabs():
+        with gr.Tab("Visualization + Chat"):
+            # User Info
+            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(
+                    choices=["Sedentary", "Lightly active", "Moderately active", "Very active", "Extra active"],
+                    label="Activity Level",
+                    value="Moderately active"
+                )
+
+            # Outputs
+            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")
+
+            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,
+                    "gender": user_gender.value
+                }
+                bmi, status = calculate_bmi(user_details['height'], user_details['weight'])
+                calories = calculate_calories(
+                    user_details['age'], user_details['weight'], user_details['height'], user_details['activity_level'], user_details['gender']
+                )
+                chart_path = visualize_bmi_and_calories(bmi, calories)
+
+                # Append metrics to response with actionable advice
+                personalized_metrics = (
+                    f"Based on your metrics:\n"
+                    f"- BMI: {bmi:.2f} ({status})\n"
+                    f"- Daily Caloric Needs: {calories:.2f} kcal\n"
+                    f"\nAdvice: "
+                )
+                if status == "underweight":
+                    personalized_metrics += "Your BMI indicates you are underweight. Focus on consuming more calories from nutrient-dense foods, including lean proteins, healthy fats, and whole grains. Consider strength training exercises to build muscle mass."
+                elif status == "normal weight":
+                    personalized_metrics += "Your BMI is in the normal range. Maintain your weight by balancing calorie intake with regular physical activity. Incorporate a mix of cardio and strength training for overall fitness."
+                elif status == "overweight":
+                    personalized_metrics += "Your BMI indicates you are overweight. Aim for a calorie deficit by reducing portion sizes and increasing physical activity. Focus on a diet rich in vegetables, lean protein, and healthy fats."
+                else:
+                    personalized_metrics += "Your BMI indicates you are obese. Consult a healthcare provider for a personalized weight loss plan. Start with small, sustainable changes, such as reducing sugary foods and increasing daily physical activity."
+
+                response = personalized_metrics
+                history.append((f"User", message))
+                history.append(("FIT.AI", response))
+                return history, chart_path
+
+            submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, gr.Image()])
+            clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, gr.Image()])
+
+        with gr.Tab("Calculator + Visualization"):
+            # Calculator Tab
+            with gr.Row():
+                user_age_calc = gr.Number(label="Age (years)", value=25, precision=0)
+                user_gender_calc = gr.Dropdown(choices=["Male", "Female"], label="Gender", value="Male")
+                user_weight_calc = gr.Number(label="Weight (kg)", value=70, precision=1)
+                user_height_calc = gr.Number(label="Height (cm)", value=170, precision=1)
+                activity_level_calc = gr.Dropdown(
+                    choices=["Sedentary", "Lightly active", "Moderately active", "Very active", "Extra active"],
+                    label="Activity Level",
+                    value="Moderately active"
+                )
+
+            bmi_output = gr.Label(label="BMI Result")
+            calorie_output = gr.Label(label="Calorie Needs")
+            bmi_chart = gr.Image(label="BMI and Calorie Chart")
+
+            calculate_button = gr.Button("Calculate")
+
+            def calculate_metrics(age, weight, height, gender, activity_level):
+                bmi, status = calculate_bmi(height, weight)
+                bmi_path = visualize_bmi_and_calories(bmi, calculate_calories(age, weight, height, activity_level, gender))
+                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
+
+            calculate_button.click(
+                calculate_metrics,
+                inputs=[user_age_calc, user_weight_calc, user_height_calc, user_gender_calc, activity_level_calc],
+                outputs=[bmi_output, calorie_output, bmi_chart]
+            )
 
 demo.launch(share=True)