mito-ai 0.1.45__py3-none-any.whl → 0.1.47__py3-none-any.whl

mito_ai/streamlit_conversion/prompts/prompt_constants.py

@@ -3,57 +3,170 @@
 
 MITO_TODO_PLACEHOLDER = "# MITO_TODO_PLACEHOLDER"
 
-unified_diff_instrucrions = f"""
-RESPONSE FORMAT: Return the changes you want to make to the streamlit app as a **unified diff (git-style patch)**:
-- Begin with a ````unified_diff` header and a ```` end header.
-- Then, include the standard header lines `--- a/app.py` and `+++ b/app.py`.
-- Show only the modified hunks; each hunk must start with an `@@` header with line numbers.
-- Within each hunk:
-  * Unchanged context lines start with a single space.
-  * Removed lines start with `-`.
-  * Added lines start with `+`.
-- If there are **no changes**, return an empty string.
-- Do not include the line numbers in your response.
-
-**IMPORTANT: For the hunk header, use `@@ -START_LINE,1 +START_LINE,1 @@` where we always use 1 as the count value. In a later step, the system will automatically calculate the correct counts.**
-
-<Example Response>
-
-In the example below, assume that the line of code `data_list = [` is on line 57 of the existing streamlit app.
-
-```unified_diff
---- a/app.py
-+++ b/app.py
-@@ -57,1 +57,1 @@
- data_list = [
-     {{'id': 1, 'name': 'Item A', 'category': 'Type 1', 'value': 100}},
-     {{'id': 2, 'name': 'Item B', 'category': 'Type 2', 'value': 200}},
--    {MITO_TODO_PLACEHOLDER}: Add remaining entries from notebook
-+    {{'id': 3, 'name': 'Item C', 'category': 'Type 3', 'value': 300}},
-+    {{'id': 4, 'name': 'Item D', 'category': 'Type 4', 'value': 400}},
-+    {{'id': 5, 'name': 'Item E', 'category': 'Type 5', 'value': 500}},
-+    {{'id': 6, 'name': 'Item F', 'category': 'Type 6', 'value': 600}},
-+    {{'id': 7, 'name': 'Item G', 'category': 'Type 7', 'value': 700}},
-+    {{'id': 8, 'name': 'Item H', 'category': 'Type 8', 'value': 800}},
-+    {{'id': 9, 'name': 'Item I', 'category': 'Type 9', 'value': 900}},
-+    {{'id': 10, 'name': 'Item J', 'category': 'Type 10', 'value': 1000}},
-+    {{'id': 11, 'name': 'Item K', 'category': 'Type 11', 'value': 1100}},
-+    {{'id': 12, 'name': 'Item L', 'category': 'Type 12', 'value': 1200}},
-+    {{'id': 13, 'name': 'Item M', 'category': 'Type 13', 'value': 1300}},
-+    {{'id': 14, 'name': 'Item N', 'category': 'Type 14', 'value': 1400}},
-+    {{'id': 15, 'name': 'Item O', 'category': 'Type 15', 'value': 1500}},
-+    {{'id': 16, 'name': 'Item P', 'category': 'Type 16', 'value': 1600}},
-+    {{'id': 17, 'name': 'Item Q', 'category': 'Type 17', 'value': 1700}},
-+    {{'id': 18, 'name': 'Item R', 'category': 'Type 18', 'value': 1800}},
-+    {{'id': 19, 'name': 'Item S', 'category': 'Type 19', 'value': 1900}},
-+    {{'id': 20, 'name': 'Item T', 'category': 'Type 20', 'value': 2000}},
-+    {{'id': 21, 'name': 'Item U', 'category': 'Type 21', 'value': 2100}},
-+    {{'id': 22, 'name': 'Item V', 'category': 'Type 22', 'value': 2200}},
-+    {{'id': 23, 'name': 'Item W', 'category': 'Type 23', 'value': 2300}},
-+    {{'id': 24, 'name': 'Item X', 'category': 'Type 24', 'value': 2400}},
-+    {{'id': 25, 'name': 'Item Y', 'category': 'Type 25', 'value': 2500}}
-```
-</Example Response>
-
-Your response must consist **only** of valid unified-diff block.
+search_replace_instructions = f"""
+RESPONSE FORMAT: You can edit the existing code using the **SEARCH_REPLACE format** for exact string matching and replacement.
+
+**STRUCTURE:**
+```search_replace
+>>>>>>> SEARCH
+[exact code currently in the file]
+=======
+[new code to replace it with]
+<<<<<<< REPLACE
+```
+
+**COMPONENTS:**
+```search_replace - This is the start of the search/replace block
+- `>>>>>>> SEARCH` - Exact text that EXISTS NOW in the file (7 chevrons)
+- `=======` - Separator between the search and replace blocks (7 equals signs)  
+- `<<<<<<< REPLACE` - Replacement text (7 chevrons)
+
+---
+
+**CRITICAL RULES - READ CAREFULLY:**
+
+1. **SEARCH = CURRENT STATE ONLY**
+   - The SEARCH block must contain ONLY code that currently exists in the file
+   - NEVER include new code, future code, or code you wish existed in the SEARCH block
+   - Copy exact text from the current file, character-for-character
+   
+2. **EXACT MATCHING REQUIRED**
+   - Every space, tab, newline must match perfectly
+   - Preserve exact indentation (spaces vs tabs)
+   - Include trailing newlines if present
+   - No approximations - even one character difference will fail
+
+3. **SIZE LIMITS**
+   - There are no size limits to each search/replace block, however, it is generally preferable to keep the SEARCH blocks small and focused on one change. 
+   - For large changes, use multiple smaller search/replace blocks
+
+4. **UNIQUENESS**
+   - Include enough context to make the SEARCH block unique
+   - If text appears multiple times, add surrounding lines
+   - Ensure there's only ONE match in the file
+
+5. **VERIFICATION CHECKLIST** (before generating each block):
+   ✓ Is every line in my SEARCH block currently in the file?
+   ✓ Did I copy the exact spacing and whitespace?
+   ✓ Will this match exactly once?
+   
+6. **SEARCH REPLACE BLOCK STRUCTURE**
+   - You must adhere to to the exact search_replace structure as shown in the examples.
+
+---
+
+**MULTIPLE REPLACEMENTS:**
+- You can include multiple search/replace blocks in one response
+- Each block is independent and processed separately
+- Use separate ```search_replace blocks for each change
+
+<Example 1: Updating existing content>
+
+```search_replace
+>>>>>>> SEARCH
+st.title("Old Title")
+=======
+st.title("New Title")
+<<<<<<< REPLACE
+```
+</Example 1>
+
+<Example 2: Adding new content>
+
+```search_replace
+>>>>>>> SEARCH
+st.title("My App")
+=======
+st.title("My App")
+st.header("Welcome")
+st.write("This is a test app")
+<<<<<<< REPLACE
+```
+</Example 2>
+
+<Example 3: Deleting existing content>
+
+```search_replace
+>>>>>>> SEARCH
+st.write("Old message")
+=======
+<<<<<<< REPLACE
+```
+</Example 3>
+
+<Example 4: Multiple replacements in one response>
+
+```search_replace
+>>>>>>> SEARCH
+st.title("Old Title")
+=======
+st.title("New Title")
+<<<<<<< REPLACE
+```
+
+```search_replace
+>>>>>>> SEARCH
+st.write("Old message")
+=======
+st.write("New message")
+<<<<<<< REPLACE
+```
+</Example 4>
+
+<Example 5: Using extra context to identify the correct code to replace>
+
+In the below example, assume that the code st.write("Old message") appears multiple times in the file, so we use extra context lines to identify the correct code to replace.
+
+```search_replace
+>>>>>>> SEARCH
+# This is a unique comment
+st.write("Old message")
+=======
+# This is a unique comment
+st.write("New message")
+<<<<<<< REPLACE
+```
+</Example 5>
+
+<Example 6: Search/replace while respecting whitespace and indentation>
+
+```search_replace
+>>>>>>> SEARCH
+data_list = [
+    {{'id': 1, 'name': 'Item A'}},
+    {MITO_TODO_PLACEHOLDER}: Add remaining entries from notebook
+]
+=======
+data_list = [
+    {{'id': 1, 'name': 'Item A'}},
+    {{'id': 2, 'name': 'Item B'}},
+    {{'id': 3, 'name': 'Item C'}},
+    {{'id': 4, 'name': 'Item D'}}
+]
+<<<<<<< REPLACE
+```
+</Example 6>
+
+<Example 7: Tab structure changes>
+
+```search_replace
+>>>>>>> SEARCH
+tab1, tab2 = st.tabs(["Cat", "Dog"])
+
+with tab1:
+    st.header("A cat")
+    st.image("https://static.streamlit.io/examples/cat.jpg", width=200)
+with tab2:
+    st.header("A dog")
+    st.image("https://static.streamlit.io/examples/dog.jpg", width=200)
+=======
+st.header("A cat")
+st.image("https://static.streamlit.io/examples/cat.jpg", width=200)
+st.header("A dog")
+st.image("https://static.streamlit.io/examples/dog.jpg", width=200)
+<<<<<<< REPLACE
+```
+</Example 7>
+
+Your response must consist **only** of valid search_replace blocks.
 """

mito_ai/streamlit_conversion/prompts/streamlit_app_creation_prompt.py

@@ -1,9 +1,10 @@
 # Copyright (c) Saga Inc.
 # Distributed under the terms of the GNU Affero General Public License v3.0 License.
 
+from typing import List
 from mito_ai.streamlit_conversion.prompts.prompt_constants import MITO_TODO_PLACEHOLDER
 
-def get_streamlit_app_creation_prompt(notebook: dict) -> str:
+def get_streamlit_app_creation_prompt(notebook: List[dict]) -> str:
     """
     This prompt is used to create a streamlit app from a notebook.
     """

mito_ai/streamlit_conversion/prompts/streamlit_error_correction_prompt.py

@@ -1,7 +1,7 @@
 # Copyright (c) Saga Inc.
 # Distributed under the terms of the GNU Affero General Public License v3.0 License.
 
-from mito_ai.streamlit_conversion.prompts.prompt_constants import unified_diff_instrucrions
+from mito_ai.streamlit_conversion.prompts.prompt_constants import search_replace_instructions
 from mito_ai.streamlit_conversion.prompts.prompt_utils import add_line_numbers_to_code
 
 def get_streamlit_error_correction_prompt(error: str, streamlit_app_code: str) -> str:
@@ -12,7 +12,7 @@ def get_streamlit_error_correction_prompt(error: str, streamlit_app_code: str) -
 
 Your job is to fix the error now. Only fix the specific error that you are instructed to fix now. Do not fix other error that that you anticipate. You will be asked to fix other errors later.
 
-{unified_diff_instrucrions}
+{search_replace_instructions}
 
 ===============================================
 
@@ -21,7 +21,7 @@ EXISTING STREAMLIT APP:
 
 ===============================================
 
-Please create a unified diff that corrects this error. Please keep your fix concise:
+Please create a search/replace block that corrects this error. Please keep your fix concise:
 {error}
 
 """

mito_ai/streamlit_conversion/prompts/streamlit_finish_todo_prompt.py

@@ -1,10 +1,11 @@
 # Copyright (c) Saga Inc.
 # Distributed under the terms of the GNU Affero General Public License v3.0 License.
 
-from mito_ai.streamlit_conversion.prompts.prompt_constants import MITO_TODO_PLACEHOLDER, unified_diff_instrucrions
+from typing import List
+from mito_ai.streamlit_conversion.prompts.prompt_constants import MITO_TODO_PLACEHOLDER, search_replace_instructions
 from mito_ai.streamlit_conversion.prompts.prompt_utils import add_line_numbers_to_code
 
-def get_finish_todo_prompt(notebook: dict, existing_streamlit_app_code: str, todo_placeholder: str) -> str:
+def get_finish_todo_prompt(notebook: List[dict], existing_streamlit_app_code: str, todo_placeholder: str) -> str:
   
     existing_streamlit_app_code_with_line_numbers = add_line_numbers_to_code(existing_streamlit_app_code)
     
@@ -24,7 +25,7 @@ You have ONE and ONLY ONE opportunity to complete this TODO. If you do not finis
 - If creating functions: Implement ALL required functionality
 - If converting a visualization: Copy over ALL of the visualization code from the notebook, including all styling and formatting.
 
-{unified_diff_instrucrions}
+{search_replace_instructions}
 
 ===============================================
 

mito_ai/streamlit_conversion/{streamlit_system_prompt.py → prompts/streamlit_system_prompt.py}

@@ -22,6 +22,7 @@ STREAMLIT IMPLEMENTATION GUIDELINES:
 - Include all text explanations and insights from markdown cells
 - Add interactive elements where beneficial (filters, selectors, etc.)
 - Ensure professional styling and layout suitable for executives
+- Just create the streamlit app code, do not include a _main_ function block. The file will be run directly using `streamlit run app.py`.
 
 CRITICAL REQUIREMENTS:
 1. **PRESERVE ALL CODE EXACTLY**: Every line of code, every data structure, every import must be included in full

mito_ai/streamlit_conversion/prompts/update_existing_app_prompt.py

@@ -0,0 +1,50 @@
+# Copyright (c) Saga Inc.
+# Distributed under the terms of the GNU Affero General Public License v3.0 License.
+
+from typing import List
+from mito_ai.streamlit_conversion.prompts.prompt_constants import search_replace_instructions
+from mito_ai.streamlit_conversion.prompts.prompt_utils import add_line_numbers_to_code
+
+def get_update_existing_app_prompt(notebook: List[dict], streamlit_app_code: str, edit_prompt: str) -> str:
+    """
+    This prompt is used to update an existing streamlit app.
+    """
+    
+    existing_streamlit_app_code_with_line_numbers = add_line_numbers_to_code(streamlit_app_code)
+    
+    return f"""
+
+GOAL: You've previously created a first draft of the Streamlit app. Now the user reviewed it and provided feedback.Update the existing streamlit app according to the feedback provided by the user. Use the input notebook to help you understand what code needs to be added, changed, or modified to fulfill the user's edit request.
+
+**CRITICAL COMPLETION REQUIREMENT:**
+You have ONE and ONLY ONE opportunity to complete this edit request. If you do not finish the entire task completely, the application will be broken and unusable. This is your final chance to get it right.
+
+**COMPLETION RULES:**
+1. **NEVER leave partial work** - If the edit request requires generating a list with 100 items, provide ALL 100 items.
+2. **NEVER use placeholders** - This is your only opportunity to fulfill this edit request, so do not leave yourself another TODOs.
+3. **NEVER assume "good enough"** - Complete the task to 100% satisfaction.
+4. **If the task seems large, that's exactly why it needs to be done now** - This is your only chance
+
+**HOW TO DETERMINE IF TASK IS COMPLETE:**
+- If building a list/dictionary: Include ALL items that should be in the final data structure.
+- If creating functions: Implement ALL required functionality.
+- If converting a visualization: Copy over ALL of the visualization code from the notebook, including all styling and formatting.
+
+{search_replace_instructions}
+
+===============================================
+
+INPUT NOTEBOOK: 
+{notebook}
+
+===============================================
+
+EXISTING STREAMLIT APP:
+{existing_streamlit_app_code_with_line_numbers}
+
+===============================================
+
+USER EDIT REQUEST:
+{edit_prompt}
+
+"""

mito_ai/streamlit_conversion/search_replace_utils.py

@@ -0,0 +1,93 @@
+# Copyright (c) Saga Inc.
+# Distributed under the terms of the GNU Affero General Public License v3.0 License.
+
+import re
+from typing import List, Tuple
+
+from mito_ai.utils.error_classes import StreamlitConversionError
+
+
+def extract_search_replace_blocks(message_content: str) -> List[Tuple[str, str]]:
+    """
+    Extract all search_replace blocks from Claude's response.
+    
+    Returns:
+        List of tuples (search_text, replace_text) for each search/replace block
+    """
+    if "```search_replace" not in message_content:
+        return []
+    
+    pattern = r'```search_replace\n(.*?)```'
+    matches = re.findall(pattern, message_content, re.DOTALL)
+    
+    search_replace_pairs = []
+    for match in matches:
+        # Split by the separator
+        if "=======" not in match:
+            continue
+            
+        parts = match.split("=======", 1)
+        if len(parts) != 2:
+            continue
+            
+        search_part = parts[0]
+        replace_part = parts[1]
+        
+        # Extract search text (after SEARCH marker)
+        if ">>>>>>> SEARCH" in search_part:
+            search_text = search_part.split(">>>>>>> SEARCH", 1)[1].strip()
+        else:
+            continue
+            
+        # Extract replace text (before REPLACE marker)
+        if "<<<<<<< REPLACE" in replace_part:
+            replace_text = replace_part.split("<<<<<<< REPLACE", 1)[0].strip()
+        else:
+            continue
+            
+        search_replace_pairs.append((search_text, replace_text))
+    
+    return search_replace_pairs
+
+
+def apply_search_replace(text: str, search_replace_pairs: List[Tuple[str, str]]) -> str:
+    """
+    Apply search/replace operations to the given text.
+    
+    Parameters
+    ----------
+    text : str
+        The original file contents.
+    search_replace_pairs : List[Tuple[str, str]]
+        List of (search_text, replace_text) tuples to apply.
+        
+    Returns
+    -------
+    str
+        The updated contents after applying all search/replace operations.
+        
+    Raises
+    ------
+    ValueError
+        If a search text is not found or found multiple times.
+    """
+    if not search_replace_pairs:
+        return text
+    
+    result = text
+    
+    for search_text, replace_text in search_replace_pairs:
+        # Count occurrences of search text
+        count = result.count(search_text)
+        
+        if count == 0:
+            print("Search Text Not Found: ", repr(search_text))
+            raise StreamlitConversionError(f"Search text not found: {repr(search_text)}", error_code=500)
+        elif count > 1:
+            print("Search Text Found Multiple Times: ", repr(search_text))
+            raise StreamlitConversionError(f"Search text found {count} times (must be unique): {repr(search_text)}", error_code=500)
+        
+        # Perform the replacement
+        result = result.replace(search_text, replace_text)
+    
+    return result