Update medapp.py

Sudhendra · Oct 17, 2024 · 4114ef2 · 4114ef2
1 parent eaa01f5
commit 4114ef2
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/medapp.py b/medapp.py
@@ -195,11 +195,12 @@ def main():
             st.subheader("Generated Answer:")
             st.write(answer)
 
-            # Add the mindmap visualization
-            st.subheader("Interactive Mindmap")
-            nodes, edges = build_mindmap(user_query, answer, related_topics)
-            config = Config(width=750, height=500, directed=True, physics=True, hierarchical=False)
-            agraph(nodes=nodes, edges=edges, config=config)
+            # Add the mindmap visualization within an expander
+            with st.expander("View Interactive Mindmap"):
+                st.subheader("Interactive Mindmap")
+                nodes, edges = build_mindmap(user_query, answer, related_topics)
+                config = Config(width=750, height=500, directed=True, physics=True, hierarchical=False)
+                agraph(nodes=nodes, edges=edges, config=config)
 
             with st.expander("View Relevant Passages"):
                 for passage in relevant_passages: