Update app.py

app.py

@@ -129,50 +129,61 @@ with gr.Blocks() as demo:
                     value="Moderately active"
                 )
 
-            # Outputs
-            chatbot = gr.Chatbot(label="Chat with FIT.AI")
-            text_input = gr.Textbox(placeholder="Type your question here...", label="Your Question")
+            # Visualization Output
+            bmi_chart = gr.Image(label="BMI and Calorie Chart")
+
+            # Chat Outputs
+            with gr.Row():
+                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()])
+                try:
+                    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
+                    }
+                    if None in user_details.values():
+                        raise ValueError("Please fill in all the details for personalized advice.")
+
+                    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
+                except ValueError as ve:
+                    return history + [("FIT.AI", str(ve))], None
+
+            submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, bmi_chart])
+            clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, bmi_chart])
 
         with gr.Tab("Calculator + Visualization"):
             # Calculator Tab
@@ -189,11 +200,14 @@ with gr.Blocks() as demo:
 
             bmi_output = gr.Label(label="BMI Result")
             calorie_output = gr.Label(label="Calorie Needs")
-            bmi_chart = gr.Image(label="BMI and Calorie Chart")
+            bmi_chart_calc = gr.Image(label="BMI and Calorie Chart")
 
             calculate_button = gr.Button("Calculate")
 
             def calculate_metrics(age, weight, height, gender, activity_level):
+                if not all([age, weight, height, gender, activity_level]):
+                    return "Please fill in all fields.", "", None
+
                 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)
@@ -202,7 +216,7 @@ with gr.Blocks() as demo:
             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]
+                outputs=[bmi_output, calorie_output, bmi_chart_calc]
             )
 
 demo.launch(share=True)