commit

app.py

@@ -95,6 +95,25 @@ def reformat_prompt(api_key: str, rough_draft: str, prompt_type: str = "standard
 def create_interface():
     """Create and configure the Gradio interface."""
     
+    # JavaScript for localStorage API key persistence
+    js_code = """
+    function saveApiKey(api_key) {
+        if (api_key && api_key.trim()) {
+            localStorage.setItem('openai_api_key', api_key);
+        }
+        return api_key;
+    }
+    
+    function loadApiKey() {
+        return localStorage.getItem('openai_api_key') || '';
+    }
+    
+    function clearApiKey() {
+        localStorage.removeItem('openai_api_key');
+        return '';
+    }
+    """
+    
     with gr.Blocks(
         title="Code Gen Agent Prompt Assistant",
         theme=gr.themes.Soft(),
@@ -121,7 +140,16 @@ def create_interface():
             white-space: pre-wrap;
             border: 1px solid #dee2e6;
         }
-        """
+        .persistence-info {
+            background-color: #e8f5e8;
+            padding: 0.75rem;
+            border-radius: 6px;
+            margin-top: 0.5rem;
+            font-size: 0.9em;
+            color: #2d5a2d;
+        }
+        """,
+        js=js_code
     ) as interface:
         
         gr.HTML("""
@@ -139,9 +167,19 @@ def create_interface():
                             label="OpenAI API Key",
                             placeholder="sk-...",
                             type="password",
-                            info="Your API key is not stored and only used for this session"
+                            info="Your API key is stored locally in your browser and persists across sessions"
                         )
                         
+                        with gr.Row():
+                            load_key_btn = gr.Button("Load Saved Key", variant="secondary", size="sm")
+                            clear_key_btn = gr.Button("Clear Saved Key", variant="secondary", size="sm")
+                        
+                        gr.HTML("""
+                        <div class="persistence-info">
+                            <strong>🔒 Privacy:</strong> Your API key is stored only in your browser's local storage and never sent to our servers.
+                        </div>
+                        """)
+                        
                         prompt_type_radio = gr.Radio(
                             choices=[
                                 ("Standard Multi-Agent Prompt", "standard"),
@@ -233,6 +271,25 @@ def create_interface():
             inputs=[api_key_input, rough_draft_input, prompt_type_radio],
             outputs=output,
             show_progress=True
+        ).then(
+            fn=None,
+            inputs=[api_key_input],
+            outputs=None,
+            js="(api_key) => saveApiKey(api_key)"
+        )
+        
+        load_key_btn.click(
+            fn=None,
+            inputs=None,
+            outputs=[api_key_input],
+            js="() => loadApiKey()"
+        )
+        
+        clear_key_btn.click(
+            fn=None,
+            inputs=None,
+            outputs=[api_key_input],
+            js="() => clearApiKey()"
         )
         
         clear_btn.click(
@@ -240,6 +297,14 @@ def create_interface():
             outputs=[api_key_input, prompt_type_radio, rough_draft_input, output]
         )
         
+        # Auto-load API key on interface load
+        interface.load(
+            fn=None,
+            inputs=None,
+            outputs=[api_key_input],
+            js="() => loadApiKey()"
+        )
+        
     
     return interface