Create main.py

main.py

@@ -0,0 +1,129 @@
+import streamlit as st
+import google.generativeai as genai
+from streamlit_react import ReactComponent
+
+# Set up Gemini API
+api_key = "AIzaSyAHD0FwX-Ds6Y3eI-i5Oz7IdbJqR6rN7pg"
+genai.configure(api_key=api_key)
+
+# Configure the Gemini model
+model = genai.GenerativeModel(
+    model_name="gemini-1.5-pro-latest",
+    generation_config=genai.GenerationConfig(
+        temperature=0.2,
+        max_output_tokens=8192
+    )
+)
+
+# Streamlit UI
+st.title("Lecture Notes Mindmap Generator")
+
+# Text area for input
+lecture_notes = st.text_area("Paste your lecture notes here:", height=300)
+
+# Function to generate mindmap data
+def generate_mindmap(notes):
+    prompt = f"""
+    Create a hierarchical mindmap structure from the following lecture notes. 
+    The structure should be in JSON format, with each node having a 'name' and optional 'children' array.
+    Ensure all content from the notes is included in the mindmap.
+    Here are the lecture notes:
+
+    {notes}
+
+    Return only the JSON structure, without any additional text or explanation.
+    """
+    
+    response = model.generate_content(prompt)
+    return response.text
+
+# Button to generate mindmap
+if st.button("Generate Mindmap"):
+    if lecture_notes:
+        with st.spinner("Generating mindmap..."):
+            mindmap_data = generate_mindmap(lecture_notes)
+            st.session_state.mindmap_data = mindmap_data
+    else:
+        st.warning("Please enter your lecture notes first.")
+
+# Display the mindmap if data is available
+if 'mindmap_data' in st.session_state:
+    st.subheader("Generated Mindmap")
+    
+    # React component for mindmap visualization
+    ReactComponent(
+        component_name="MindMap",
+        props={"data": st.session_state.mindmap_data},
+        key="mindmap_viewer"
+    )
+
+# React component definition (you'll need to save this in a separate file)
+st.write("""
+To use this app, you'll need to create a file named `mindmap_component.jsx` in your Streamlit app's directory with the following content:
+
+```jsx
+import React, { useState } from 'react';
+import { ChevronRight, ChevronDown } from 'lucide-react';
+
+const TreeNode = ({ node }) => {
+  const [isOpen, setIsOpen] = useState(false);
+  const hasChildren = node.children && node.children.length > 0;
+
+  return (
+    <div className="ml-4">
+      <div 
+        className="flex items-center cursor-pointer hover:bg-gray-100 p-2 rounded"
+        onClick={() => setIsOpen(!isOpen)}
+      >
+        {hasChildren && (
+          <span className="mr-2">
+            {isOpen ? <ChevronDown size={16} /> : <ChevronRight size={16} />}
+          </span>
+        )}
+        <span className={`${hasChildren ? 'font-semibold' : ''}`}>{node.name}</span>
+      </div>
+      {isOpen && hasChildren && (
+        <div className="ml-4 border-l-2 border-gray-200">
+          {node.children.map((child, index) => (
+            <TreeNode key={index} node={child} />
+          ))}
+        </div>
+      )}
+    </div>
+  );
+};
+
+const MindMap = ({ data }) => {
+  const parsedData = typeof data === 'string' ? JSON.parse(data) : data;
+  
+  return (
+    <div className="p-4 bg-white rounded-lg shadow-lg max-w-4xl mx-auto overflow-auto max-h-[600px]">
+      <h2 className="text-2xl font-bold mb-4 text-center">Lecture Notes Mindmap</h2>
+      <TreeNode node={parsedData} />
+    </div>
+  );
+};
+
+export default MindMap;
+```
+""")
+
+st.write("""
+To set up this Streamlit app with the React component:
+
+1. Install required packages:
+   ```
+   pip install streamlit google-generativeai streamlit-react
+   ```
+
+2. Save the main Python code in a file named `app.py`.
+
+3. Create a file named `mindmap_component.jsx` with the React component code provided above.
+
+4. Run the Streamlit app:
+   ```
+   streamlit run app.py
+   ```
+
+This setup will allow you to paste your lecture notes, generate a mindmap using Gemini, and visualize it using the React component.
+""")