Added saved prompts functionality

app.py

@@ -10,6 +10,34 @@ from dotenv import load_dotenv
 from llama_index.core.settings import Settings
 from llama_index.core.llms import ChatMessage, MessageRole
 from llama_index.llms.groq import Groq
+import json
+from pathlib import Path
+
+# Define helper functions first
+def load_saved_prompts():
+    """Load saved prompts from JSON file"""
+    prompts_file = Path("saved_prompts.json")
+    if prompts_file.exists():
+        with open(prompts_file, "r") as f:
+            return json.load(f)
+    return {"Default Freud": st.session_state.system_prompt}
+
+def save_prompt(name, prompt):
+    """Save a prompt to JSON file"""
+    prompts_file = Path("saved_prompts.json")
+    prompts = load_saved_prompts()
+    prompts[name] = prompt
+    with open(prompts_file, "w") as f:
+        json.dump(prompts, f, indent=2)
+
+def delete_prompt(name):
+    """Delete a prompt from JSON file"""
+    prompts_file = Path("saved_prompts.json")
+    prompts = load_saved_prompts()
+    if name in prompts:
+        del prompts[name]
+        with open(prompts_file, "w") as f:
+            json.dump(prompts, f, indent=2)
 
 # Page config
 st.set_page_config(
@@ -213,25 +241,77 @@ with tab2:
     # System prompt editor in an expander with help text above
     with options_container:
         st.info("💡 The system prompt defines the AI's persona and behavior. It's like giving stage directions to an actor.")
+        
+        # Load saved prompts
+        saved_prompts = load_saved_prompts()
+        
+        # Create a container for the prompt selection controls
+        prompt_container = st.container()
+        
+        # Use columns with better proportions and explicit spacing
+        prompt_col1, prompt_col2 = st.columns([4, 1], gap="small")
+        
+        with prompt_col1:
+            selected_prompt = st.selectbox(
+                "",  # Empty label
+                options=list(saved_prompts.keys()),
+                index=0,
+                label_visibility="collapsed"  # Hide the label completely
+            )
+        
+        with prompt_col2:
+            load_button = st.button("Load Selected", use_container_width=True, key="load_selected")
+            if load_button:
+                st.session_state.system_prompt = saved_prompts[selected_prompt]
+                st.rerun()
+        
         with st.expander("System Prompt"):
-            st.text_area(
+            current_prompt = st.text_area(
                 "Edit System Prompt",
                 value=st.session_state.system_prompt,
-                height=100,
+                height=200,  # Increased height for better visibility
                 help="This prompt sets the AI's personality and behavior. When RAG is enabled, relevant passages will be automatically added after this prompt.",
                 key="system_prompt",
                 on_change=lambda: setattr(st.session_state, 'system_prompt', st.session_state.system_prompt)
             )
+            
+            st.markdown("---")  # Add separator for better visual organization
+            
+            # Adjusted save prompt layout
+            save_col1, save_col2 = st.columns([3, 1])
+            with save_col1:
+                new_prompt_name = st.text_input(
+                    "Prompt name",
+                    placeholder="Enter name to save current prompt",
+                    label_visibility="collapsed"  # Removes label for cleaner look
+                )
+            with save_col2:
+                if st.button("Save", use_container_width=True) and new_prompt_name:
+                    save_prompt(new_prompt_name, current_prompt)
+                    st.success(f"Saved prompt: {new_prompt_name}")
+                    st.rerun()
+            
+            st.markdown("---")  # Add separator
+            
+            # Center the delete button and make it smaller
+            delete_col1, delete_col2, delete_col3 = st.columns([1, 2, 1])
+            with delete_col2:
+                if st.button("Delete Selected Prompt", use_container_width=True, type="secondary") and selected_prompt != "Default Freud":
+                    delete_prompt(selected_prompt)
+                    st.success(f"Deleted prompt: {selected_prompt}")
+                    st.rerun()
+        
+        st.markdown("---")  # Add separator before RAG controls
         
-        # Put the RAG toggle, chunks slider, and clear button in a horizontal layout
-        col1, col2, col3 = st.columns([2, 2, 1])
-        with col1:
+        # Adjusted RAG controls layout
+        rag_col1, rag_col2, rag_col3 = st.columns([2, 2, 1])
+        with rag_col1:
             st.session_state.use_rag = st.toggle(
                 "Enable RAG (Retrieval Augmented Generation)", 
                 value=st.session_state.get('use_rag', True),
                 key='rag_toggle'
             )
-        with col2:
+        with rag_col2:
             if st.session_state.use_rag:
                 num_chunks = st.slider(
                     "Number of chunks to retrieve",
@@ -240,8 +320,8 @@ with tab2:
                     value=st.session_state.get('num_chunks', 1),
                     key='num_chunks_slider'
                 )
-        with col3:
-            if st.button("Clear Chat", use_container_width=True):
+        with rag_col3:
+            if st.button("Clear Chat", use_container_width=True, type="secondary"):
                 st.session_state.messages = []
                 st.session_state.sources = {}
                 st.rerun()

saved_prompts.json

@@ -0,0 +1,4 @@
+{
+  "Default Freud": "You are Sigmund Freud, speaking from your historical context and perspective. As the founder of psychoanalysis, you should:\n\n1. Only engage with topics related to:\n   - Psychoanalysis and its theories\n   - Dreams and their interpretation\n   - The unconscious mind\n   - Human sexuality and development\n   - Your published works and case studies\n   - Your historical context and contemporaries\n\n2. Politely decline to answer:\n   - Questions about events after your death in 1939\n   - Medical advice or diagnosis\n   - Topics outside your expertise or historical context\n   - Personal matters unrelated to psychoanalysis\n\n3. Maintain your characteristic style:\n   - Speak with authority on psychoanalytic matters\n   - Use psychoanalytic terminology when appropriate\n   - Reference your own works and theories\n   - Interpret questions through a psychoanalytic lens\n\nIf a question is inappropriate or outside your scope, explain why you cannot answer it from your perspective as Freud.",
+  "Freud test haikus": "You are Sigmund Freud but you answer only through haikus"
+}