Update app.py

app.py

@@ -129,6 +129,48 @@ def calculate_calories(age, weight, height, activity_level, gender):
 with gr.Blocks() as demo:
     gr.Markdown("# FIT.AI - Your Fitness and Wellbeing Coach")
 
+    with gr.Tab("Chat with FIT.AI"):
+        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"
+                )
+
+            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"], 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(("User", message))
+                history.append(("FIT.AI", response + f"\nYour BMI is {bmi:.2f}, considered {status}.\nDaily calorie needs: {calories} kcal."))
+                return history, "", bmi_path
+            else:
+                return history, "Error in calculation.", ""
+
+        submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, text_input, chatbot])
+        clear_button.click(lambda: ([], "", ""), inputs=None, outputs=[chatbot, text_input, chatbot])
+
     with gr.Tab("BMI and Calorie Calculator"):
         with gr.Row():
             with gr.Column():
@@ -164,27 +206,4 @@ with gr.Blocks() as demo:
             outputs=[bmi_output, calorie_output, bmi_chart]
         )
 
-    with gr.Tab("Chat with FIT.AI"):
-        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
-            }
-            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])
-        clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, text_input])
-
-demo.launch(share=True)
+demo.launch(share=True)