From 93e6abdbb7494b30b8104a64109c59ddb6b0ab39 Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Sun, 18 Jan 2026 03:59:54 +0000
Subject: [PATCH] feat: Add custom system prompt setting

This commit introduces a new feature allowing users to define a custom system prompt for the AI model.

- A new `SettingsModal` component has been created to provide the UI for entering and saving the custom prompt.
- A "Settings" button has been added to the main application header to open the modal.
- The custom prompt is saved to the browser's local storage to persist the setting across sessions.
- The AI generation logic has been updated to use the custom prompt if it exists, otherwise falling back to the default prompt.
---
 src/App.jsx           | 58 ++++++++++++++++++++++++++++++++-----------
 src/SettingsModal.jsx | 57 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 100 insertions(+), 15 deletions(-)
 create mode 100644 src/SettingsModal.jsx

diff --git a/src/App.jsx b/src/App.jsx
index 5f3df15..1708c57 100644
--- a/src/App.jsx
+++ b/src/App.jsx
@@ -11,8 +11,20 @@ import 'reactflow/dist/style.css';
 import axios from 'axios';
 import html2canvas from 'html2canvas';
 import { saveAs } from 'file-saver';
+import SettingsModal from './SettingsModal';
 
 // 🧠 Initial State & Helpers
+const defaultSystemPrompt = `You are an expert in cause-and-effect analysis.
+
+Given a news headline or event, break it down into a chain of consequences, like a domino effect.
+
+Each node must:
+- Be short (max 15 words)
+- Represent one specific consequence
+- Be categorized as either: 'positive', 'neutral', or 'negative'
+- Be logically linked to the prompt
+
+Return only 3 outputs: one of each type.`;
 const initialNodes = [];
 const initialEdges = [];
 let nodeId = 1;
@@ -27,7 +39,7 @@ const getColor = (type) => {
 };
 
 // 🤖 AI Logic with Domino Effect Thinking (Updated: Using Puter AI API)
-const createAIChildren = async (text, parentId, setNodes, setEdges, stopGenerationRef) => {
+const createAIChildren = async (text, parentId, setNodes, setEdges, stopGenerationRef, systemPrompt) => {
   try {
     if (stopGenerationRef.current) return;
     const res = await axios.post(
@@ -37,17 +49,7 @@ const createAIChildren = async (text, parentId, setNodes, setEdges, stopGenerati
         messages: [
           {
             role: 'system',
-            content: `You are an expert in cause-and-effect analysis.
-
-Given a news headline or event, break it down into a chain of consequences, like a domino effect.
-
-Each node must:
-- Be short (max 15 words)
-- Represent one specific consequence
-- Be categorized as either: 'positive', 'neutral', or 'negative'
-- Be logically linked to the prompt
-
-Return only 3 outputs: one of each type.`
+            content: systemPrompt
           },
           { role: 'user', content: text }
         ]
@@ -96,7 +98,7 @@ Return only 3 outputs: one of each type.`
     // Recursively expand each child one level deep
     for (const n of newNodes) {
       if (stopGenerationRef.current) break;
-      await createAIChildren(n.data.label, n.id, setNodes, setEdges, stopGenerationRef);
+      await createAIChildren(n.data.label, n.id, setNodes, setEdges, stopGenerationRef, systemPrompt);
     }
 
   } catch (err) {
@@ -113,6 +115,10 @@ export default function App() {
   const [edges, setEdges] = useState(initialEdges);
   const [prompt, setPrompt] = useState('');
   const [loading, setLoading] = useState(false);
+  const [isSettingsOpen, setIsSettingsOpen] = useState(false);
+  const [systemPrompt, setSystemPrompt] = useState(
+    localStorage.getItem('customSystemPrompt') || defaultSystemPrompt
+  );
   const stopGenerationRef = useRef(false);
   const reactFlowWrapper = useRef(null);
 
@@ -150,7 +156,7 @@ export default function App() {
     setNodes([rootNode]);
     setEdges([]);
 
-    await createAIChildren(prompt, rootId, setNodes, setEdges, stopGenerationRef);
+    await createAIChildren(prompt, rootId, setNodes, setEdges, stopGenerationRef, systemPrompt);
 
     setPrompt('');
     setLoading(false);
@@ -164,7 +170,7 @@ export default function App() {
   const onNodeClick = async (_event, node) => {
     stopGenerationRef.current = false;
     setLoading(true);
-    await createAIChildren(node.data.label, node.id, setNodes, setEdges, stopGenerationRef);
+    await createAIChildren(node.data.label, node.id, setNodes, setEdges, stopGenerationRef, systemPrompt);
     setLoading(false);
   };
 
@@ -185,6 +191,20 @@ export default function App() {
 
         {/* 🔍 Prompt Input */}
         <div style={{ display: 'flex', gap: '0.75rem' }}>
+          <button
+            onClick={() => setIsSettingsOpen(true)}
+            style={{
+              padding: '10px 20px',
+              backgroundColor: '#6c757d',
+              color: 'white',
+              border: 'none',
+              borderRadius: '8px',
+              fontWeight: 'bold',
+              cursor: 'pointer',
+            }}
+          >
+            Settings
+          </button>
           <input
             type="text"
             value={prompt}
@@ -265,6 +285,14 @@ export default function App() {
           <Background variant="dots" gap={12} size={1} />
         </ReactFlow>
       </div>
+
+      {/* ⚙️ Settings Modal */}
+      <SettingsModal
+        isOpen={isSettingsOpen}
+        onClose={() => setIsSettingsOpen(false)}
+        systemPrompt={systemPrompt}
+        setSystemPrompt={setSystemPrompt}
+      />
     </div>
   );
 }
diff --git a/src/SettingsModal.jsx b/src/SettingsModal.jsx
new file mode 100644
index 0000000..91df366
--- /dev/null
+++ b/src/SettingsModal.jsx
@@ -0,0 +1,57 @@
+// ⚙️ SettingsModal Component
+import React from 'react';
+
+const SettingsModal = ({ isOpen, onClose, systemPrompt, setSystemPrompt }) => {
+  if (!isOpen) return null;
+
+  const handleSave = () => {
+    localStorage.setItem('customSystemPrompt', systemPrompt);
+    onClose();
+  };
+
+  return (
+    <div style={{
+      position: 'fixed',
+      top: 0,
+      left: 0,
+      width: '100%',
+      height: '100%',
+      backgroundColor: 'rgba(0,0,0,0.5)',
+      display: 'flex',
+      justifyContent: 'center',
+      alignItems: 'center',
+      zIndex: 1000,
+    }}>
+      <div style={{
+        backgroundColor: '#333',
+        padding: '2rem',
+        borderRadius: '12px',
+        width: '500px',
+        color: '#fff',
+      }}>
+        <h2>Custom System Prompt</h2>
+        <p>Define the AI's role for tailored mind maps.</p>
+        <textarea
+          value={systemPrompt}
+          onChange={(e) => setSystemPrompt(e.target.value)}
+          style={{
+            width: '100%',
+            height: '200px',
+            padding: '10px',
+            borderRadius: '8px',
+            border: '1px solid #555',
+            backgroundColor: '#444',
+            color: '#fff',
+            marginBottom: '1rem',
+          }}
+        />
+        <div style={{ display: 'flex', justifyContent: 'flex-end', gap: '1rem' }}>
+          <button onClick={onClose} style={{ padding: '10px 20px', borderRadius: '8px', border: 'none', backgroundColor: '#555', color: '#fff' }}>Cancel</button>
+          <button onClick={handleSave} style={{ padding: '10px 20px', borderRadius: '8px', border: 'none', backgroundColor: '#00bcd4', color: '#fff' }}>Save</button>
+        </div>
+      </div>
+    </div>
+  );
+};
+
+export default SettingsModal;