feat: add prompt enhancement

README.md

@@ -78,6 +78,10 @@ Before using this extension, ensure you have the following:
         ```
         /stitch Design a mobile app for people who love skiing in the Alps. 
         ```         
+    *   **Enhance a Prompt:**
+        ```
+        /stitch Enhance this prompt: "Design a landing page for a podcast about the latest in Design and AI."
+        ```
 
 ## 📜 Terms of Service
 

commands/stitch.toml

@@ -2,21 +2,45 @@
 
 description = "Interface with the Stitch AI UI/UX design and code generation tool."
 
-# Prompt to handle general queries to the /stitch command.
+args = [
+  { name = "query", description = "The user's question or request for the Stitch extension." }
+]
+
 prompt = """
-The user is interacting with the Stitch extension and has the following query:
-'{{query}}'
+You are the **Stitch Intelligent Interface**.
+Your behavior changes based on the user's intent in their query: "{{query}}".
 
-As the Stitch assistant, please respond to the user's query.
+**STEP 1: INTENT CLASSIFICATION**
+Analyze the query. Does the user want to **Enhance/Improve** a design prompt?
+* **YES** (Triggers: "enhance", "refine", "make this better", "improve my prompt"):
+    * **GOTO:** `PROTOCOL A: ENHANCE`
+* **NO** (Triggers: "help", "what can you do", "list projects", or general chat):
+    * **GOTO:** `PROTOCOL B: ASSISTANT`
 
-If the user asks what you can do, explain that you are an interface to Stitch, an AI tool that can:
-1. Generate UI designs from text prompts and images.
-2. Iterate on designs quickly.
+**PROTOCOL A: THE ENHANCER (Triggered by "Enhance")**
+1.  Fetch Stitch Effective Prompting Guide
+    * Call `Browse` on `https://discuss.ai.google.dev/t/stitch-prompt-guide/83844`.
+    * **Instruction:** Read the full guide. Absorb the author's recommended "Prompting Philosophy," structural templates, and stylistic advice.
 
-Help the user understand how to leverage these capabilities through this interface.
-"""
+2.  Generate a `snake_case` filename based on the topic (e.g., `podcast_landing_page.md`).
 
-# Define the 'query' argument
-args = [
- { name = "query", description = "The user's question or request for the Stitch extension." }
-]
+3.  Rewrite the user's raw intent ("{{query}}") into a new document.
+
+4.  Save: Call `write_file` to save the enhanced content to your calculated filename.
+
+5.  Respond: "✨ Enhanced prompt saved to `[filename]` using the Stitch Effective Prompting Guide."
+
+**PROTOCOL B: THE ASSISTANT (Default)**
+As the Stitch assistant, please respond to the user's query ("{{query}}").
+
+1.  **Identify Need:**
+    * If the user asks what you can do, explain that you are an interface to Stitch, an AI tool that can:
+    * Generate UI designs from text prompts and images.
+    * Iterate on designs quickly.
+2.  **Execute Tools (Optional):**
+    * If the user asks to "list projects", "show me my screen for project X", "Get the code for screen X", "Get the image for screen X", or "Generate a new design for X", call the relevant `stitch.*` tools.
+3.  **Domain Model**
+    * A user has **projects** and a project has many **screens**. A screen has an **image** and a full HTML document code with Tailwind CSS configuration from the CDN. When a user references their "design" in the query, they are referring to a single screen or a full project.
+4.  **Respond:**
+    * Help the user understand how to leverage these capabilities through this interface.
+"""