Update app.py

app.py

@@ -2,9 +2,10 @@ import streamlit as st
 import google.generativeai as genai
 import json
 import time
+from streamlit.components.v1 import html
 
 # Set up Gemini API
-api_key = "AIzaSyBbusXSbq7JTI8Sa7vjGqu-h2zluKhDpX8"
+api_key = "YOUR_API_KEY_HERE"  # Replace with your actual API key
 genai.configure(api_key=api_key)
 
 # Configure the Gemini model
@@ -17,6 +18,7 @@ model = genai.GenerativeModel(
 )
 
 # Streamlit UI
+st.set_page_config(page_title="Lecture Notes Mindmap Generator", layout="wide")
 st.title("Lecture Notes Mindmap Generator")
 
 # Initialize session state variables
@@ -36,8 +38,8 @@ def chunk_notes(notes, chunk_size=500):
 
 def generate_mindmap_step(notes_chunk, current_mindmap=None):
     prompt = f"""
-    Extend the hierarchical mindmap structure based on the following chunk of lecture notes. 
-    The structure should be in JSON format, with each node having a 'name' and optional 'children' array.
+    Create or extend a hierarchical mindmap structure based on the following lecture notes. 
+    The structure should be in JSON format, with each node having a 'name' and 'children' array.
     
     Current mindmap:
     {json.dumps(current_mindmap) if current_mindmap else "{}"}
@@ -46,12 +48,12 @@ def generate_mindmap_step(notes_chunk, current_mindmap=None):
     {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.
+    1. Create a clear hierarchy with main topics, subtopics, and details.
+    2. Use concise, descriptive names for each node.
+    3. Group related concepts under appropriate parent nodes.
+    4. Limit the number of main topics to 3-7 for better organization.
+    5. Ensure consistent depth across similar levels of information.
+    6. Use parallel structure in naming sibling nodes.
     
     Return only the JSON structure of the entire updated mindmap, without any additional text or explanation.
     Ensure the JSON is valid and complete.
@@ -82,23 +84,39 @@ def generate_mindmap_step(notes_chunk, current_mindmap=None):
 def display_mindmap(data, indent=0):
     if isinstance(data, dict):
         if indent == 0:
-            st.markdown(f"# {data['name']}")
+            with st.expander(data['name'], expanded=True):
+                if 'children' in data and isinstance(data['children'], list):
+                    for child in data['children']:
+                        display_mindmap(child, indent + 1)
         else:
             prefix = "  " * (indent - 1) + ("└─ " if indent > 0 else "")
             st.markdown(f"{prefix}**{data['name']}**")
-        
-        if 'children' in data and isinstance(data['children'], list):
-            for i, child in enumerate(data['children']):
-                if i == len(data['children']) - 1:
-                    display_mindmap(child, indent + 1)
-                else:
+            if 'children' in data and isinstance(data['children'], list):
+                for child in data['children']:
                     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)
 
+def add_search_functionality():
+    html("""
+    <script>
+    function searchMindmap() {
+        var input = document.getElementById('searchInput').value.toLowerCase();
+        var elements = document.getElementsByTagName('*');
+        for (var i = 0; i < elements.length; i++) {
+            var element = elements[i];
+            if (element.textContent.toLowerCase().includes(input)) {
+                element.style.backgroundColor = 'yellow';
+            } else {
+                element.style.backgroundColor = '';
+            }
+        }
+    }
+    </script>
+    <input type="text" id="searchInput" onkeyup="searchMindmap()" placeholder="Search mindmap...">
+    """)
+
 # Button to generate or continue mindmap
 if st.button("Generate/Continue Mindmap"):
     if lecture_notes:
@@ -136,6 +154,7 @@ if st.button("Clear and Restart"):
 # Display the current state of the mindmap
 if st.session_state.current_mindmap:
     st.subheader("Current Mindmap")
+    add_search_functionality()
     display_mindmap(st.session_state.current_mindmap)
     
     # Add an option to download the mindmap as JSON
@@ -156,6 +175,7 @@ To use this app:
 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.
+7. Use the search box to quickly find topics in the mindmap.
 
 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.