commit

app.py

@@ -44,7 +44,7 @@ Requirements:
 
 Output format: Return the complete prompt with frontmatter and system prompt as plain text with no markdown formatting or code fences."""
 
-def reformat_prompt(api_key: str, rough_draft: str, prompt_type: str = "standard") -> str:
+def reformat_prompt(api_key: str, rough_draft: str, prompt_type: str = "standard", api_key_state: str = "") -> tuple[str, str]:
     """
     Reformat a rough draft into a polished system prompt using OpenAI API.
     
@@ -52,22 +52,26 @@ def reformat_prompt(api_key: str, rough_draft: str, prompt_type: str = "standard
         api_key: OpenAI API key
         rough_draft: The rough draft text to be reformatted
         prompt_type: Type of prompt to generate ("standard" or "claude_code")
+        api_key_state: State variable for API key persistence
         
     Returns:
-        Reformatted system prompt or error message
+        Tuple of (reformatted system prompt or error message, updated api_key_state)
     """
-    if not api_key.strip():
-        return "Error: Please provide your OpenAI API key."
+    # Use provided API key or fall back to state
+    effective_api_key = api_key.strip() or api_key_state.strip()
+    
+    if not effective_api_key:
+        return "Error: Please provide your OpenAI API key.", api_key_state
     
     if not rough_draft.strip():
-        return "Error: Please provide a rough draft to reformat."
+        return "Error: Please provide a rough draft to reformat.", effective_api_key
     
     # Select the appropriate system prompt based on type
     selected_system_prompt = CLAUDE_CODE_SYSTEM_PROMPT if prompt_type == "claude_code" else SYSTEM_PROMPT
     
     try:
-        # Initialize OpenAI client with the provided API key
-        client = openai.OpenAI(api_key=api_key)
+        # Initialize OpenAI client with the effective API key
+        client = openai.OpenAI(api_key=effective_api_key)
         
         # Make API call to reformat the prompt
         response = client.chat.completions.create(
@@ -81,39 +85,20 @@ def reformat_prompt(api_key: str, rough_draft: str, prompt_type: str = "standard
         )
         
         reformatted_text = response.choices[0].message.content
-        return reformatted_text
+        return reformatted_text, effective_api_key
         
     except openai.AuthenticationError:
-        return "Error: Invalid API key. Please check your OpenAI API key."
+        return "Error: Invalid API key. Please check your OpenAI API key.", api_key_state
     except openai.RateLimitError:
-        return "Error: Rate limit exceeded. Please try again later."
+        return "Error: Rate limit exceeded. Please try again later.", effective_api_key
     except openai.APIError as e:
-        return f"Error: OpenAI API error - {str(e)}"
+        return f"Error: OpenAI API error - {str(e)}", effective_api_key
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error: {str(e)}", effective_api_key
 
 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(),
@@ -148,8 +133,7 @@ def create_interface():
             font-size: 0.9em;
             color: #2d5a2d;
         }
-        """,
-        js=js_code
+        """
     ) as interface:
         
         gr.HTML("""
@@ -163,23 +147,16 @@ def create_interface():
             with gr.TabItem("Prompt Reformatter"):
                 with gr.Row():
                     with gr.Column():
+                        # Create a state variable for API key persistence
+                        api_key_state = gr.State("")
+                        
                         api_key_input = gr.Textbox(
                             label="OpenAI API Key",
                             placeholder="sk-...",
                             type="password",
-                            info="Your API key is stored locally in your browser and persists across sessions"
+                            info="Your API key will be remembered during this session"
                         )
                         
-                        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"),
@@ -268,41 +245,14 @@ def create_interface():
         # Event handlers
         submit_btn.click(
             fn=reformat_prompt,
-            inputs=[api_key_input, rough_draft_input, prompt_type_radio],
-            outputs=output,
+            inputs=[api_key_input, rough_draft_input, prompt_type_radio, api_key_state],
+            outputs=[output, api_key_state],
             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(
-            fn=lambda: ("", "standard", "", ""),
-            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()"
+            fn=lambda: ("", "standard", "", "", ""),
+            outputs=[api_key_input, prompt_type_radio, rough_draft_input, output, api_key_state]
         )