Update app.py

app.py

@@ -320,17 +320,25 @@ APA Citations:
         # Mind Map Generator (Visual)
         st.markdown("---")
         st.subheader("🧭 Visual Mind Map")
-        if st.button("🗺 Generate Mind Map"):
-            mindmap_prompt = [
-                {"role": "system", "content": "You are a researcher who converts reports into Mermaid.js mind maps."},
-                {"role": "user", "content": f"Create a mermaid mind map for this research:\n{full_output}"}
-            ]
-            mindmap_code = ""
-            for chunk in call_llm(mindmap_prompt):
-                mindmap_code += chunk
-
-            mindmap_code = mindmap_code.replace("```mermaid", "").replace("```", "").strip()
-            st.markdown(f"<div class='mermaid'>{mindmap_code}</div>", unsafe_allow_html=True)
+        
+        if "last_report" not in st.session_state or not st.session_state["last_report"].strip():
+            st.warning("⚠️ Please generate a report first.")
+        else:
+            if st.button("🗺 Generate Mind Map"):
+                mindmap_prompt = [
+                    {"role": "system", "content": "You are a researcher who converts reports into mermaid mind maps."},
+                    {"role": "user", "content": f"Create a mermaid.js mind map of the following:\n\n{st.session_state['last_report']}"}
+                ]
+                mindmap_code = ""
+                for chunk in call_llm(mindmap_prompt):
+                    mindmap_code += chunk
+        
+                # Remove wrapping markdown if exists
+                mindmap_code = mindmap_code.replace("```mermaid", "").replace("```", "").strip()
+        
+                # Render mermaid directly
+                st.markdown(f"<div class='mermaid'>{mindmap_code}</div>", unsafe_allow_html=True)
+
 
         # Follow-up Q&A
         st.markdown("---")