Update app.py

app.py

@@ -51,7 +51,11 @@ def run_graph(input_message, history, user_details):
             logger.debug("Invoking workflow with messages: %s", messages)
             response = graph.invoke({"messages": messages})
             logger.debug("Workflow response: %s", response)
-            q.put(response["messages"][1]["content"])
+            # Check for proper response structure
+            if "messages" in response and len(response["messages"]) > 1:
+                q.put(response["messages"][1].get("content", "No content in response."))
+            else:
+                q.put("The workflow did not return a valid response. Please try again.")
         except Exception as e:
             logger.error("Error in run_graph: %s", str(e))
             q.put(f"An error occurred while processing your request: {e}")
@@ -60,11 +64,11 @@ def run_graph(input_message, history, user_details):
     q = queue.Queue()
     thread = threading.Thread(target=invoke_workflow, args=(q,))
     thread.start()
-    thread.join(timeout=300)  # Wait for 30 seconds
+    thread.join(timeout=300)  # Increased timeout to 60 seconds
 
     if thread.is_alive():
         logger.error("Workflow timed out.")
-        return "The request is taking longer than expected. Please try again later or modify your query."
+        return "The request is taking longer than expected. Please try again later."
     return q.get()
 
 
@@ -83,21 +87,20 @@ def calculate_bmi(height, weight):
 
 
 def visualize_bmi_and_calories(bmi, calories):
-    # Create a combined visualization
     categories = ["Underweight", "Normal Weight", "Overweight", "Obese"]
     bmi_values = [18.5, 24.9, 29.9, 40]
     calorie_range = [1500, 2000, 2500, 3000]
 
     fig, ax1 = plt.subplots(figsize=(10, 6))
 
-    # BMI visualization
+    # BMI Visualization
     ax1.bar(categories, bmi_values, color=['blue', 'green', 'orange', 'red'], alpha=0.6, label="BMI Ranges")
     ax1.axhline(y=bmi, color='purple', linestyle='--', linewidth=2, label=f"Your BMI: {bmi:.2f}")
     ax1.set_ylabel("BMI Value")
     ax1.set_title("BMI and Caloric Needs Visualization")
     ax1.legend(loc="upper left")
 
-    # Calorie visualization
+    # Calorie Visualization
     ax2 = ax1.twinx()
     ax2.plot(categories, calorie_range, 'o-', color='magenta', label="Calorie Ranges")
     ax2.axhline(y=calories, color='cyan', linestyle='--', linewidth=2, label=f"Your Calorie Needs: {calories:.2f} kcal")
@@ -106,7 +109,7 @@ def visualize_bmi_and_calories(bmi, calories):
 
     plt.tight_layout()
 
-    # Save to a temporary file
+    # Save visualization to a temporary file
     temp_file = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
     plt.savefig(temp_file.name)
     plt.close()
@@ -136,7 +139,7 @@ with gr.Blocks() as demo:
 
     with gr.Tabs():
         with gr.Tab("Visualization + Chat"):
-            # User Info
+            # User Input
             with gr.Row():
                 user_name = gr.Textbox(placeholder="Enter your name", label="Name")
                 user_age = gr.Number(label="Age (years)", value=25, precision=0)
@@ -177,11 +180,6 @@ with gr.Blocks() as demo:
                     )
                     chart_path = visualize_bmi_and_calories(bmi, calories)
 
-                    # Compose input for the LLM dynamically
-                    system_prompt = (
-                        "You are a fitness and health assistant. Provide detailed advice based on the user's metrics, including BMI and caloric needs."
-                    )
-
                     user_prompt = (
                         f"User Details:\n"
                         f"- Name: {user_details['name']}\n"
@@ -195,38 +193,29 @@ with gr.Blocks() as demo:
                         f"\nProvide tailored advice based on these metrics."
                     )
 
-                    messages = [
-                        {"role": "system", "content": system_prompt},
-                        {"role": "user", "content": user_prompt}
-                    ]
-
-                    logger.debug("Submitting messages to LLM: %s", messages)
                     response = run_graph(message, history, user_details)
-                    logger.debug("Received LLM response: %s", response)
-
                     history.append((f"User", message))
                     history.append(("FIT.AI", response))
                     return history, chart_path
 
                 except Exception as e:
                     logger.error("Error in submit_message: %s", str(e))
-                    return history + [("FIT.AI", "An error occurred while processing your request." + str(e))], None
+                    return history + [("FIT.AI", "An error occurred. Please try again.")], None
 
             submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, bmi_chart])
             clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, bmi_chart])
 
+        # Calculator + Visualization Tab
         with gr.Tab("Calculator + Visualization"):
-            # Calculator Tab
-            with gr.Row():
-                user_age_calc = gr.Number(label="Age (years)", value=25, precision=0)
-                user_gender_calc = gr.Dropdown(choices=["Male", "Female"], label="Gender", value="Male")
-                user_weight_calc = gr.Number(label="Weight (kg)", value=70, precision=1)
-                user_height_calc = gr.Number(label="Height (cm)", value=170, precision=1)
-                activity_level_calc = gr.Dropdown(
-                    choices=["Sedentary", "Lightly active", "Moderately active", "Very active", "Extra active"],
-                    label="Activity Level",
-                    value="Moderately active"
-                )
+            user_age_calc = gr.Number(label="Age (years)", value=25, precision=0)
+            user_gender_calc = gr.Dropdown(choices=["Male", "Female"], label="Gender", value="Male")
+            user_weight_calc = gr.Number(label="Weight (kg)", value=70, precision=1)
+            user_height_calc = gr.Number(label="Height (cm)", value=170, precision=1)
+            activity_level_calc = gr.Dropdown(
+                choices=["Sedentary", "Lightly active", "Moderately active", "Very active", "Extra active"],
+                label="Activity Level",
+                value="Moderately active"
+            )
 
             bmi_output = gr.Label(label="BMI Result")
             calorie_output = gr.Label(label="Calorie Needs")
@@ -237,9 +226,9 @@ with gr.Blocks() as demo:
             def calculate_metrics(age, weight, height, gender, activity_level):
                 try:
                     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)
-                    return f"Your BMI is {bmi:.2f}, considered {status}.", f"Daily calorie needs: {calories:.2f} kcal", bmi_path
+                    chart_path = visualize_bmi_and_calories(bmi, calories)
+                    return f"Your BMI is {bmi:.2f}, considered {status}.", f"Daily calorie needs: {calories:.2f} kcal", chart_path
                 except Exception as e:
                     logger.error("Error in calculate_metrics: %s", str(e))
                     return "An error occurred.", "", None