Update app.py

app.py

@@ -12,6 +12,9 @@ import logging
 import threading
 import queue
 
+# Global timeout variable
+TIMEOUT_SECONDS = 300
+
 # Initialize workflow
 graph = create_workflow()
 
@@ -19,6 +22,9 @@ 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):
@@ -48,30 +54,31 @@ def run_graph(input_message, history, user_details):
             ]
 
             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}")
 
             # Check and return tailored response
             if "messages" in response and len(response["messages"]) > 1:
-                assistant_message = response["messages"][1].get("content", "").strip()
-                if assistant_message:
-                    q.put(assistant_message)
-                else:
-                    q.put("The assistant did not provide specific advice. Please refine your query.")
+                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))
+            debug_logs.append(f"Error: {str(e)}")
             q.put(f"An error occurred: {e}")
 
     q = queue.Queue()
     thread = threading.Thread(target=invoke_workflow, args=(q,))
     thread.start()
-    thread.join(timeout=300)
+    thread.join(timeout=TIMEOUT_SECONDS)
 
     if thread.is_alive():
-        logger.error("Workflow timed out.")
-        return "The request is taking longer than expected. Please try again later."
+        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()
 
 
@@ -157,6 +164,7 @@ 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():
@@ -194,20 +202,21 @@ with gr.Blocks() as demo:
                         f"- Activity Level: {user_details['activity_level']}\n"
                         f"- BMI: {bmi:.2f} ({status})\n"
                         f"- Daily Caloric Needs: {calories:.2f} kcal\n"
-                        f"Provide tailored advice based on these metrics."
+                        f"\nProvide tailored advice based on these metrics."
                     )
 
                     response = run_graph(user_prompt, history, user_details)
                     history.append((f"User", message))
                     history.append(("FIT.AI", response))
-                    return history, chart_path
+                    return history, chart_path, "\n".join(debug_logs)
 
                 except Exception as e:
                     logger.error("Error in submit_message: %s", str(e))
-                    return history + [("FIT.AI", "An error occurred. Please try again.")], None
+                    debug_logs.append(f"Error: {str(e)}")
+                    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])
-            clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, bmi_chart])
+            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])
 
         # Calculator + Visualization Tab
         with gr.Tab("Calculator + Visualization"):