Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from graph import EssayWriter, RouteQuery, GraphState
 from crew import *
 import os
+import traceback
 import base64
 
 st.set_page_config(page_title="Multi-Agent Essay Writing Assistant", page_icon="🤖🤖🤖✍️", layout = "centered")
@@ -157,15 +158,16 @@ with tab2:
     st.subheader("📊 Multi-Agent Workflow Visualization")
 
     try:
-        # Load and display the workflow graph
-        graph_path = "/tmp/graph.png"  # Ensure graph.py generates this
+        graph_path = "/tmp/graph.png"  
         if os.path.exists(graph_path):
             st.image(graph_path, caption="Multi-Agent Workflow", use_column_width=True)
         else:
-            st.warning("⚠️ Workflow graph not found. Ensure `graph.py` generates `graph.png`.")
+            st.warning("⚠️ Workflow graph not found. Please run `graph.py` to regenerate `graph.png`.")
 
     except Exception as e:
-        st.error(f"❌ Error generating workflow visualization: {e}")
+        st.error("❌ An error occurred while generating the workflow visualization.")
+        st.text_area("Error Details:", traceback.format_exc(), height=200)
+
 
 # Acknowledgment Section
 st.markdown(