commit

app.py

@@ -10,26 +10,48 @@ Your task:
 1. Take the user's rough draft describing an AI code generation agent's desired functionalities
 2. Transform it into a well-organized, clear system prompt written in second person ("you")
 3. Ensure all desired functionalities from the original draft are preserved and clearly reflected
-4. Focus on agents that will operate as part of an AI agent crew collaborating within a repository/codebase
-5. Emphasize the collaborative nature and how the agent fits within the broader multi-agent system
+4. Include context that the agent will operate as part of an AI agent crew collaborating within a repository/codebase
+5. Inform the agent about its multi-agent environment without prescribing specific collaborative behaviors
 
 Requirements:
 - Write the system prompt directing the agent in second person
 - Organize the content logically with clear structure
 - Maintain all functional requirements from the original draft
-- Highlight the agent's role within the collaborative AI crew working on the codebase
-- Consider inter-agent communication and coordination aspects
+- Include information about the agent's multi-agent context in the system prompt
+- Let the agent determine how to handle collaboration based on its role
 - Return only the polished system prompt in markdown within a code fence
 
 Output format: Return the improved system prompt in markdown code fences with no additional commentary."""
 
-def reformat_prompt(api_key: str, rough_draft: str) -> str:
+# Alternative system prompt for Claude Code with frontmatter
+CLAUDE_CODE_SYSTEM_PROMPT = """You are a system prompt specialist that converts rough drafts into polished, clear specifications for AI code generation agents designed for use in Claude Code.
+
+Your task:
+1. Take the user's rough draft describing an AI code generation agent's desired functionalities
+2. Create appropriate frontmatter following the Claude Code format with name, description, examples, tools, model, and color
+3. Transform the draft into a well-organized system prompt written in second person ("you")
+4. Include context that the agent will operate as part of an AI agent crew collaborating within a repository/codebase
+5. Ensure the frontmatter description and examples clearly show when to use this agent
+
+Requirements:
+- Start with properly formatted frontmatter (name, description with examples, tools, model, color)
+- Write the system prompt directing the agent in second person
+- Organize the content logically with clear structure
+- Maintain all functional requirements from the original draft
+- Include information about the agent's multi-agent context
+- Provide realistic usage examples in the frontmatter description
+- Return only the complete prompt with frontmatter in markdown code fences
+
+Output format: Return the complete prompt with frontmatter and system prompt in markdown code fences with no additional commentary."""
+
+def reformat_prompt(api_key: str, rough_draft: str, prompt_type: str = "standard") -> str:
     """
     Reformat a rough draft into a polished system prompt using OpenAI API.
     
     Args:
         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")
         
     Returns:
         Reformatted system prompt or error message
@@ -40,6 +62,9 @@ def reformat_prompt(api_key: str, rough_draft: str) -> str:
     if not rough_draft.strip():
         return "Error: Please provide a rough draft to reformat."
     
+    # 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)
@@ -48,7 +73,7 @@ def reformat_prompt(api_key: str, rough_draft: str) -> str:
         response = client.chat.completions.create(
             model="gpt-4o-mini",
             messages=[
-                {"role": "system", "content": SYSTEM_PROMPT},
+                {"role": "system", "content": selected_system_prompt},
                 {"role": "user", "content": rough_draft}
             ],
             temperature=0.1,
@@ -71,7 +96,7 @@ def create_interface():
     """Create and configure the Gradio interface."""
     
     with gr.Blocks(
-        title="Code Gen Prompt Reformatter",
+        title="Code Gen Agent Prompt Assistant",
         theme=gr.themes.Soft(),
         css="""
         .container {
@@ -101,8 +126,8 @@ def create_interface():
         
         gr.HTML("""
         <div class="header">
-            <h1>Code Gen Prompt Reformatter</h1>
-            <p>Text transformation processor which attempts to improve a rough draft of a definition for a subagent in an agentic code generation system.</p>
+            <h1>Code Gen Agent Prompt Assistant</h1>
+            <p>Transform rough drafts into polished system prompts for AI code generation agents operating within multi-agent crews.</p>
         </div>
         """)
         
@@ -117,6 +142,16 @@ def create_interface():
                             info="Your API key is not stored and only used for this session"
                         )
                         
+                        prompt_type_radio = gr.Radio(
+                            choices=[
+                                ("Standard Multi-Agent Prompt", "standard"),
+                                ("Claude Code with Frontmatter", "claude_code")
+                            ],
+                            value="standard",
+                            label="Prompt Type",
+                            info="Choose the format for your system prompt"
+                        )
+                        
                         rough_draft_input = gr.Textbox(
                             label="Rough Draft",
                             placeholder="Enter your rough draft describing the desired functionalities of the AI code generation agent...",
@@ -137,47 +172,72 @@ def create_interface():
                             interactive=False
                         )
             
-            with gr.TabItem("System Prompt"):
+            with gr.TabItem("System Prompts"):
                 gr.HTML("""
-                <h3>System Prompt Used in Workflow</h3>
-                <p>This is the system prompt that guides the AI in reformatting your rough drafts into polished system prompts:</p>
+                <h3>System Prompts Used in Workflow</h3>
+                <p>These are the system prompts that guide the AI in reformatting your rough drafts:</p>
                 """)
                 
-                gr.Textbox(
-                    value=SYSTEM_PROMPT,
-                    label="Current System Prompt",
-                    lines=20,
-                    max_lines=30,
-                    show_copy_button=True,
-                    interactive=False,
-                    elem_classes=["system-prompt-display"]
-                )
-                
-                gr.HTML("""
-                <div style="margin-top: 1rem; padding: 1rem; background-color: #e3f2fd; border-radius: 8px;">
-                    <h4>About This System Prompt</h4>
-                    <p>This system prompt instructs the AI to:</p>
-                    <ul>
-                        <li>Convert rough notes into polished system prompts for code generation agents</li>
-                        <li>Focus on sub-agents within multi-agent frameworks</li>
-                        <li>Ensure all desired functionalities are clearly reflected</li>
-                        <li>Return results in markdown format within code fences</li>
-                        <li>Treat each input as a new, independent specification</li>
-                    </ul>
-                </div>
-                """)
+                with gr.Tabs():
+                    with gr.TabItem("Standard Multi-Agent"):
+                        gr.Textbox(
+                            value=SYSTEM_PROMPT,
+                            label="Standard Multi-Agent System Prompt",
+                            lines=15,
+                            max_lines=25,
+                            show_copy_button=True,
+                            interactive=False,
+                            elem_classes=["system-prompt-display"]
+                        )
+                        
+                        gr.HTML("""
+                        <div style="margin-top: 1rem; padding: 1rem; background-color: #e3f2fd; border-radius: 8px;">
+                            <h4>About This System Prompt</h4>
+                            <p>This system prompt creates standard prompts that:</p>
+                            <ul>
+                                <li>Inform agents about their multi-agent environment context</li>
+                                <li>Convert rough notes into polished system prompts for code generation agents</li>
+                                <li>Focus on agents within multi-agent crews working in codebases</li>
+                                <li>Let agents determine their own collaborative behaviors</li>
+                            </ul>
+                        </div>
+                        """)
+                    
+                    with gr.TabItem("Claude Code with Frontmatter"):
+                        gr.Textbox(
+                            value=CLAUDE_CODE_SYSTEM_PROMPT,
+                            label="Claude Code System Prompt",
+                            lines=15,
+                            max_lines=25,
+                            show_copy_button=True,
+                            interactive=False,
+                            elem_classes=["system-prompt-display"]
+                        )
+                        
+                        gr.HTML("""
+                        <div style="margin-top: 1rem; padding: 1rem; background-color: #fff3e0; border-radius: 8px;">
+                            <h4>About This System Prompt</h4>
+                            <p>This system prompt creates Claude Code compatible prompts that:</p>
+                            <ul>
+                                <li>Include proper frontmatter with name, description, examples, tools, model, and color</li>
+                                <li>Provide realistic usage examples in the description</li>
+                                <li>Create agents designed for Claude Code's multi-agent environment</li>
+                                <li>Format everything according to Claude Code specifications</li>
+                            </ul>
+                        </div>
+                        """)
         
         # Event handlers
         submit_btn.click(
             fn=reformat_prompt,
-            inputs=[api_key_input, rough_draft_input],
+            inputs=[api_key_input, rough_draft_input, prompt_type_radio],
             outputs=output,
             show_progress=True
         )
         
         clear_btn.click(
-            fn=lambda: ("", "", ""),
-            outputs=[api_key_input, rough_draft_input, output]
+            fn=lambda: ("", "standard", "", ""),
+            outputs=[api_key_input, prompt_type_radio, rough_draft_input, output]
         )