Update app.py

app.py

@@ -1,4 +1,3 @@
-# add tab 3 --> workflow diagram
 import gradio as gr
 import matplotlib.pyplot as plt
 import pandas as pd
@@ -60,6 +59,9 @@ def run_graph(input_message, history, user_details):
 
 
 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)
@@ -71,31 +73,41 @@ def calculate_bmi(height, weight, gender):
             status = "overweight"
         else:
             status = "obese"
+
         return bmi, status
-    except Exception:
+    except:
         return None, "Invalid height or weight provided."
 
+
 def visualize_bmi(bmi):
-    try:
-        categories = ["Underweight", "Normal Weight", "Overweight", "Obese"]
-        colors = ['blue', 'green', 'orange', 'red']
-
-        plt.figure(figsize=(8, 4))
-        plt.bar(categories, [18.5, 24.9, 29.9, 40], color=colors, alpha=0.6)
-        plt.axhline(y=bmi, color='purple', linestyle='--', linewidth=2, label=f"Your BMI: {bmi:.2f}")
-        plt.ylabel("BMI Value")
-        plt.title("BMI Visualization")
-        plt.legend(loc="upper left")
-        plt.tight_layout()
-
-        plt_path = "bmi_chart.png"
-        plt.savefig(plt_path)
-        plt.close()
-        return plt_path
-    except Exception as e:
-        return f"Error creating BMI visualization: {e}"
+    """
+    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:
         if gender.lower() == "male":
             bmr = 10 * weight + 6.25 * height - 5 * age + 5
@@ -109,69 +121,50 @@ def calculate_calories(age, weight, height, activity_level, gender):
             "Very active": 1.725,
             "Extra active": 1.9,
         }
-        return round(bmr * activity_multipliers[activity_level], 2)
-    except Exception as e:
-        return f"Error in calorie calculation: {e}"
+
+        calories = bmr * activity_multipliers.get(activity_level, 1.2)
+        return calories
+    except Exception:
+        return "Invalid inputs for calorie calculation."
 
 # Interface Components
 with gr.Blocks() as demo:
-    gr.Markdown("# FIT.AI - Your Fitness and Wellbeing Coach")
+    gr.Markdown("<strong>FIT.AI - Your Fitness and Wellbeing Coach</strong>")
 
+    # User Info
     with gr.Row():
-        with gr.Column():
-            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"
-            )
-        with gr.Column():
-            chatbot = gr.Chatbot(label="Chat with FIT.AI", type="messages", height=500)
-            text_input = gr.Textbox(placeholder="Type your question here...", label="Your Question")
-            bmi_result = gr.Label(label="BMI Result")
-            calorie_result = gr.Label(label="Calorie Needs")
-            bmi_chart = gr.Image(label="BMI Chart")
-            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
-        }
-        try:
-            bmi, status = calculate_bmi(user_details["height"], user_details["weight"], user_details["gender"])
-            if bmi:
-                bmi_path = visualize_bmi(bmi)
-                calories = calculate_calories(
-                    user_details["age"], user_details["weight"], user_details["height"], user_details["activity_level"], user_details["gender"]
-                )
-                response = run_graph(message, history, user_details)
-                history.append({"role": "user", "content": message})
-                history.append({
-                    "role": "assistant",
-                    "content": f"{response}\n\nYour BMI is {bmi:.2f}, considered {status}.\nDaily calorie needs: {calories} kcal.",
-                    "image": bmi_path
-                })
-                return history, "", f"Your BMI is {bmi:.2f}, considered {status}.", f"Daily calorie needs: {calories} kcal.", bmi_path
-            else:
-                return history, "", "Invalid input for BMI.", "", ""
-        except Exception as e:
-            return history, "", f"Error occurred: {e}", "", ""
-
-    submit_button.click(
-        submit_message, 
-        inputs=[text_input, chatbot], 
-        outputs=[chatbot, text_input, bmi_result, calorie_result, bmi_chart]
+        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 Chart")
+
+    # Actions
+    calculate_button = gr.Button("Calculate")
+
+    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]
     )
-    clear_button.click(lambda: ([], "", "", "", ""), inputs=None, outputs=[chatbot, text_input, bmi_result, calorie_result, bmi_chart])
 
 demo.launch(share=True)