Update app.py

app.py

@@ -45,6 +45,14 @@ def generate_mindmap_step(notes_chunk, current_mindmap=None):
     New lecture notes chunk:
     {notes_chunk}
     
+    Follow these guidelines to create a well-structured mindmap:
+    1. Create clear, concise node names.
+    2. Maintain a logical hierarchy with main topics and subtopics.
+    3. Ensure consistent depth across similar levels of information.
+    4. Group related concepts together.
+    5. Use parallel structure in naming sibling nodes.
+    6. Limit the number of main topics to 3-7 for better organization.
+    
     Return only the JSON structure of the entire updated mindmap, without any additional text or explanation.
     Ensure the JSON is valid and complete.
     """
@@ -72,18 +80,21 @@ def generate_mindmap_step(notes_chunk, current_mindmap=None):
         return None
 
 def display_mindmap(data, indent=0):
-    if isinstance(data, str):
-        try:
-            data = json.loads(data)
-        except json.JSONDecodeError:
-            st.error("Failed to parse the mindmap data. Please try again.")
-            return
-    
     if isinstance(data, dict):
-        st.markdown('  ' * indent + f"- **{data['name']}**")
+        if indent == 0:
+            st.markdown(f"# {data['name']}")
+        else:
+            prefix = "  " * (indent - 1) + ("└─ " if indent > 0 else "")
+            st.markdown(f"{prefix}**{data['name']}**")
+        
         if 'children' in data and isinstance(data['children'], list):
-            for child in data['children']:
-                display_mindmap(child, indent + 1)
+            for i, child in enumerate(data['children']):
+                if i == len(data['children']) - 1:
+                    display_mindmap(child, indent + 1)
+                else:
+                    display_mindmap(child, indent + 1)
+                    if indent == 0:
+                        st.markdown("---")  # Add a horizontal line between main topics
     elif isinstance(data, list):
         for item in data:
             display_mindmap(item, indent)
@@ -126,6 +137,16 @@ if st.button("Clear and Restart"):
 if st.session_state.current_mindmap:
     st.subheader("Current Mindmap")
     display_mindmap(st.session_state.current_mindmap)
+    
+    # Add an option to download the mindmap as JSON
+    if st.button("Download Mindmap as JSON"):
+        json_string = json.dumps(st.session_state.current_mindmap, indent=2)
+        st.download_button(
+            label="Click here to download",
+            file_name="mindmap.json",
+            mime="application/json",
+            data=json_string
+        )
 
 st.write("""
 To use this app:
@@ -134,6 +155,7 @@ To use this app:
 3. The mindmap will be generated in chunks, and you can see the progress.
 4. If the process is interrupted, you can continue from where it left off.
 5. Use the "Clear and Restart" button to start over with new notes.
+6. You can download the generated mindmap as a JSON file for future reference.
 
 This app uses Google's Gemini AI to structure your notes into a hierarchical mindmap format.
 It processes the notes in multiple steps to handle large inputs and complex structures.