Update app.py

app.py

@@ -60,9 +60,15 @@ def run_graph(input_message, history, user_details):
             logger.debug("Workflow response: %s", response)
             debug_logs.append(f"Workflow Response: {response}")
 
-            # Check and return tailored response
-            if "messages" in response and len(response["messages"]) > 1:
-                q.put(response["messages"][1].get("content", "No content in response."))
+            # Extract LLM response
+            llm_response = None
+            for msg in response.get("messages", []):
+                if isinstance(msg, HumanMessage) and msg.name in ["nutritionist", "workout_coach"]:
+                    llm_response = msg.content
+                    break
+
+            if llm_response:
+                q.put(llm_response)
             else:
                 q.put("The workflow did not return a valid response. Please try again.")
         except Exception as e:
@@ -121,8 +127,10 @@ def visualize_bmi_and_calories(bmi, calories):
 
     # Save visualization to a temporary file
     temp_file = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
-    plt.savefig(temp_file.name)
-    plt.close()
+    try:
+        plt.savefig(temp_file.name)
+    finally:
+        plt.close()
 
     return temp_file.name
 
@@ -134,20 +142,21 @@ def calculate_calories(age, weight, height, activity_level, gender):
         bmr = 10 * weight + 6.25 * height - 5 * age - 161
 
     activity_multipliers = {
-        "Sedentary": 1.2,
-        "Lightly active": 1.375,
-        "Moderately active": 1.55,
-        "Very active": 1.725,
-        "Extra active": 1.9,
+        "sedentary": 1.2,
+        "lightly active": 1.375,
+        "moderately active": 1.55,
+        "very active": 1.725,
+        "extra active": 1.9,
     }
 
+    activity_level = activity_level.lower()
     return bmr * activity_multipliers.get(activity_level, 1.2)
 
 # Interface Components
 with gr.Blocks() as demo:
     gr.Markdown("<strong>FIT.AI - Your Fitness and Wellbeing Coach</strong>")
 
-    debug_logs = []  # Store debug logs
+    debug_logs = []
 
     with gr.Tabs():
         with gr.Tab("Visualization + Chat"):