Update app.py

app.py

@@ -22,9 +22,6 @@ graph = create_workflow()
 logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
-# Debug logs for displaying in UI
-debug_logs = []
-
 # Helper Functions
 def run_graph(input_message, history, user_details):
     def invoke_workflow(q):
@@ -53,12 +50,7 @@ def run_graph(input_message, history, user_details):
                 {"role": "user", "content": input_message}
             ]
 
-            logger.debug("Invoking workflow with messages: %s", messages)
-            debug_logs.append(f"Workflow Input: {messages}")
-
             response = graph.invoke({"messages": messages})
-            logger.debug("Workflow response: %s", response)
-            debug_logs.append(f"Workflow Response: {response}")
 
             # Extract LLM response
             llm_response = None
@@ -72,8 +64,6 @@ def run_graph(input_message, history, user_details):
             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))
-            debug_logs.append(f"Error: {str(e)}")
             q.put(f"An error occurred: {e}")
 
     q = queue.Queue()
@@ -82,8 +72,6 @@ def run_graph(input_message, history, user_details):
     thread.join(timeout=TIMEOUT_SECONDS)
 
     if thread.is_alive():
-        logger.error(f"Workflow timed out after {TIMEOUT_SECONDS} seconds.")
-        debug_logs.append(f"Workflow timed out after {TIMEOUT_SECONDS} seconds.")
         return f"The request took longer than {TIMEOUT_SECONDS} seconds and timed out. Please try again."
     return q.get()
 
@@ -156,8 +144,6 @@ def calculate_calories(age, weight, height, activity_level, gender):
 with gr.Blocks() as demo:
     gr.Markdown("<strong>FIT.AI - Your Fitness and Wellbeing Coach</strong>")
 
-    debug_logs = []
-
     with gr.Tabs():
         with gr.Tab("Visualization + Chat"):
             # User Input
@@ -175,7 +161,6 @@ with gr.Blocks() as demo:
 
             # Visualization Output
             bmi_chart = gr.Image(label="BMI and Calorie Chart")
-            debug_output = gr.Textbox(label="Debug Logs", interactive=False, lines=10)
 
             # Chat Outputs
             with gr.Row():
@@ -186,62 +171,47 @@ with gr.Blocks() as demo:
             clear_button = gr.Button("Clear Chat")
 
             def submit_message(message, history=[]):
-                try:
-                    # Debug logging
-                    debug_logs.clear()
-                    debug_logs.append("Processing user request...")
-
-                    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
-                    }
-                    debug_logs.append(f"User Details: {user_details}")
-
-                    bmi, status = calculate_bmi(user_details['height'], user_details['weight'])
-                    calories = calculate_calories(
-                        user_details['age'], user_details['weight'], user_details['height'], user_details['activity_level'], user_details['gender']
-                    )
-                    chart_path = visualize_bmi_and_calories(bmi, calories)
-
-                    user_prompt = (
-                        f"User wants advice on: {message}\n"
-                        f"User Details:\n"
-                        f"- Name: {user_details['name']}\n"
-                        f"- Age: {user_details['age']}\n"
-                        f"- Gender: {user_details['gender']}\n"
-                        f"- Weight: {user_details['weight']} kg\n"
-                        f"- Height: {user_details['height']} cm\n"
-                        f"- Activity Level: {user_details['activity_level']}\n"
-                        f"- BMI: {bmi:.2f} ({status})\n"
-                        f"- Daily Caloric Needs: {calories:.2f} kcal\n"
-                        f"\nProvide tailored advice based on these metrics."
-                    )
-                    debug_logs.append(f"Composed user prompt: {user_prompt}")
-
-                    response = run_graph(user_prompt, history, user_details)
-                    debug_logs.append(f"Received response: {response}")
-
-                    # Append the tailored response to the chatbot history
-                    history.append(("User", message))
-                    if isinstance(response, str):
-                        history.append(("FIT.AI", response))
-                    else:
-                        history.append(("FIT.AI", "An unexpected response was received. Check debug logs for details."))
-
-                    return history, chart_path, "\n".join(debug_logs)
-
-                except Exception as e:
-                    error_message = f"Error: {str(e)}"
-                    logger.error(error_message)
-                    debug_logs.append(error_message)
-                    return history + [("FIT.AI", "An error occurred. Please try again.")], None, "\n".join(debug_logs)
-
-            submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, bmi_chart, debug_output])
-            clear_button.click(lambda: ([], "", ""), inputs=None, outputs=[chatbot, bmi_chart, debug_output])
+                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'])
+                calories = calculate_calories(
+                    user_details['age'], user_details['weight'], user_details['height'], user_details['activity_level'], user_details['gender']
+                )
+                chart_path = visualize_bmi_and_calories(bmi, calories)
+
+                user_prompt = (
+                    f"User wants advice on: {message}\n"
+                    f"User Details:\n"
+                    f"- Name: {user_details['name']}\n"
+                    f"- Age: {user_details['age']}\n"
+                    f"- Gender: {user_details['gender']}\n"
+                    f"- Weight: {user_details['weight']} kg\n"
+                    f"- Height: {user_details['height']} cm\n"
+                    f"- Activity Level: {user_details['activity_level']}\n"
+                    f"- BMI: {bmi:.2f} ({status})\n"
+                    f"- Daily Caloric Needs: {calories:.2f} kcal\n"
+                    f"\nProvide tailored advice based on these metrics."
+                )
+
+                response = run_graph(user_prompt, history, user_details)
+
+                history.append(("User", message))
+                if isinstance(response, str):
+                    history.append(("FIT.AI", response))
+                else:
+                    history.append(("FIT.AI", "An unexpected response was received."))
+
+                return history, chart_path
+
+            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"):
@@ -262,22 +232,11 @@ with gr.Blocks() as demo:
             calculate_button = gr.Button("Calculate")
 
             def calculate_metrics(age, weight, height, gender, activity_level):
-                try:
-                    debug_logs.clear()
-                    debug_logs.append("Processing calculator request...")
-
-                    bmi, status = calculate_bmi(height, weight)
-                    calories = calculate_calories(age, weight, height, activity_level, gender)
-                    chart_path = visualize_bmi_and_calories(bmi, calories)
-
-                    debug_logs.append(f"Calculated metrics: BMI={bmi:.2f}, Calories={calories:.2f}")
-                    return f"Your BMI is {bmi:.2f}, considered {status}.", f"Daily calorie needs: {calories:.2f} kcal", chart_path
-
-                except Exception as e:
-                    error_message = f"Error: {str(e)}"
-                    logger.error(error_message)
-                    debug_logs.append(error_message)
-                    return "An error occurred.", "", None
+                bmi, status = calculate_bmi(height, weight)
+                calories = calculate_calories(age, weight, height, activity_level, gender)
+                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
 
             calculate_button.click(
                 calculate_metrics,