Update app.py

app.py

@@ -1,12 +1,17 @@
 import gradio as gr
+import matplotlib.pyplot as plt
+import pandas as pd
+from datetime import datetime
 from langchain_core.messages import HumanMessage
 from config import *
 from tools import tools
 from agents import *
 from workflow import create_workflow
 
+# Initialize workflow
 graph = create_workflow()
 
+# Helper Functions
 def run_graph(input_message, history, user_details):
     try:
         # Relevant fitness-related keywords to handle irrelevant user prompts
@@ -51,7 +56,6 @@ def run_graph(input_message, history, user_details):
         return f"An error occurred while processing your request: {e}"
 
 
-# Advanced Functionalities
 def calculate_bmi(height, weight):
     try:
         height_m = height / 100
@@ -66,24 +70,45 @@ def calculate_bmi(height, weight):
             status = "obese"
         return f"Your BMI is {bmi:.2f}, which is considered {status}."
     except:
-        return "Please provide valid height and weight values."
+        return "Invalid height or weight provided."
 
+def calculate_calories(age, weight, height, activity_level):
+    try:
+        # Harris-Benedict Equation for calorie calculation
+        bmr = 10 * weight + 6.25 * height - 5 * age + 5  # Male (adjust for female by subtracting 161)
+        activity_multipliers = {
+            "Sedentary": 1.2,
+            "Lightly active": 1.375,
+            "Moderately active": 1.55,
+            "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."
+    except Exception:
+        return "Invalid inputs for calorie calculation."
 
 # Interface Components
 with gr.Blocks() as demo:
-    gr.Markdown("<strong>FIT.AI - Your Personal Wellbeing Coach</strong>")
+    gr.Markdown("<strong>FIT.AI - Your Advanced Fitness and Wellbeing Coach</strong>")
 
-    # User Details Form
+    # User Info
     with gr.Row():
         user_name = gr.Textbox(placeholder="Enter your name", label="Name")
-        user_age = gr.Number(label="Age", value=25, precision=0)
+        user_age = gr.Number(label="Age (years)", value=25, precision=0)
         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"
+        )
 
+    # 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")
+    clear_button = gr.Button("Clear Chat")
 
     # Examples
     examples = gr.Examples(
@@ -92,49 +117,25 @@ with gr.Blocks() as demo:
             "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?",
-            "What are the best recovery exercises after a leg injury?",
-            "Can you help me calculate my BMI?",
-            "What are some quick tips to manage work stress?",
-            "Explain the benefits of intermittent fasting",
-            "Which exercises can help improve posture?",
-            "What should I include in my diet to build muscle?",
-            "How can I stay active while working from home?",
-            "What are the signs of overtraining?",
-            "What are some healthy snacks for weight loss?",
-            "How can I improve my sleep quality?",
-            "Suggest a daily routine to stay fit and mentally healthy",
         ],
         inputs=text_input,
     )
 
-    # Chatbot Logic
-    def submit_message(message, history=[], user_details=None):
+    # 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
+            "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, ""
 
-    submit_button.click(
-        submit_message,
-        inputs=[text_input, chatbot],
-        outputs=[chatbot, text_input]
-    )
+    submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, text_input])
     clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, text_input])
 
-    # BMI Calculator Button
-    with gr.Row():
-        calculate_bmi_button = gr.Button("Calculate BMI")
-        bmi_result = gr.Textbox(label="BMI Result", interactive=False)
-        calculate_bmi_button.click(
-            calculate_bmi,
-            inputs=[user_height, user_weight],
-            outputs=[bmi_result]
-        )
-
 demo.launch(share=True)