- Fix code_modifier

app.py

@@ -10,7 +10,7 @@ from typing import Dict, Any, Tuple
 from Project import update_system_prompts
 import pandas as pd
 from io import StringIO
-from step_handlers import setup_all_handlers
+from event_handlers import setup_all_handlers
 
 def create_ui_component(config: UIConfig, prompt: str = None) -> Any:
     if config.component_type == UIComponentType.TEXTBOX:
@@ -49,66 +49,50 @@ def create_ui_component(config: UIConfig, prompt: str = None) -> Any:
 def create_section_components(prompt_config: PromptConfig) -> Dict[str, Any]:
     components = {}
     
+    editor_key = next((k for k in prompt_config.ui.keys() if k.endswith('_prompt_editor')), None)
+    if editor_key:
+        # Format the prompt
+        lines = prompt_config.prompt.split('\n')
+        if lines:
+            min_indent = min((len(line) - len(line.lstrip()) 
+                           for line in lines if line.strip()), 
+                           default=0)
+            formatted_lines = [
+                line[min_indent:] if line.strip() else ''
+                for line in lines
+            ]
+            formatted_prompt = '\n'.join(formatted_lines)
+            formatted_prompt = f'"""\n{formatted_prompt}\n"""'
+        else:
+            formatted_prompt = '""""""'
+        
+        # Create prompt editor in dropdown
+        with gr.Accordion(prompt_config.ui[editor_key].label, open=False):
+            components[editor_key] = create_ui_component(
+                prompt_config.ui[editor_key],
+                formatted_prompt
+            )
+    
+    # Create output components in two columns
     with gr.Row():
-        with gr.Column(scale=1):
-            editor_key = next((k for k in prompt_config.ui.keys() if k.endswith('_prompt_editor')), None)
-            if editor_key:
-                lines = prompt_config.prompt.split('\n')
-                if lines:
-                    min_indent = min((len(line) - len(line.lstrip()) 
-                                   for line in lines if line.strip()), 
-                                   default=0)
-                    formatted_lines = [
-                        line[min_indent:] if line.strip() else ''
-                        for line in lines
-                    ]
-                    formatted_prompt = '\n'.join(formatted_lines)
-                    formatted_prompt = f'"""\n{formatted_prompt}\n"""'
-                else:
-                    formatted_prompt = '""""""'
-                
-                components[editor_key] = create_ui_component(
-                    prompt_config.ui[editor_key],
-                    formatted_prompt
-                )
+        text_key = next((k for k in prompt_config.ui.keys() if k.endswith('_text')), None)
+        markdown_key = next((k for k in prompt_config.ui.keys() if k.endswith('_markdown')), None)
         
-        with gr.Column(scale=1):
-            text_key = next((k for k in prompt_config.ui.keys() if k.endswith('_text')), None)
-            markdown_key = next((k for k in prompt_config.ui.keys() if k.endswith('_markdown')), None)
-            
-            if text_key and markdown_key:
-                display_mode_key = f"display_mode_{text_key.replace('_text', '')}"
-                components[display_mode_key] = create_ui_component(
-                    UIConfig(
-                        component_type=UIComponentType.RADIO,
-                        label="Select Mode",
-                        elem_classes=["radio-group-horizontal"]
-                    )
+        if text_key and markdown_key:
+            # Left column for textbox
+            with gr.Column(scale=1):
+                components[text_key] = create_ui_component(
+                    prompt_config.ui[text_key]
                 )
-
-                with gr.Group():
-                    components[text_key] = create_ui_component(
-                        prompt_config.ui[text_key]
-                    )
-                    components[markdown_key] = create_ui_component(
-                        prompt_config.ui[markdown_key]
-                    )
-
-                components[display_mode_key].change(
-                    fn=update_display_mode,
-                    inputs=[
-                        components[display_mode_key],
-                        components[text_key]
-                    ],
-                    outputs=[
-                        components[text_key],
-                        components[markdown_key]
-                    ]
+            # Right column for markdown
+            with gr.Column(scale=1):
+                components[markdown_key] = create_ui_component(
+                    prompt_config.ui[markdown_key]
                 )
-            else:
-                for key, config in prompt_config.ui.items():
-                    if not key.endswith('_prompt_editor'):
-                        components[key] = create_ui_component(config)
+        else:
+            for key, config in prompt_config.ui.items():
+                if not key.endswith('_prompt_editor'):
+                    components[key] = create_ui_component(config)
     
     return components
 

code_modifier.py

@@ -101,28 +101,32 @@ def analyze_config_structure(config_content: str, project_content: str, current_
     3. DO NOT suggest changes based on previous versions
     4. ONLY compare against the current files provided
     5. IGNORE any remembered patterns or previous suggestions
+    6. The config file (page_prompts_config.py) is the source of truth for naming conventions
+    7. Component names MUST match their corresponding config names EXACTLY
     
     Your task is to analyze and focus on the configuration structure and validate the relationships between components.
-    
 
     1. OUTPUT AND INPUT VALIDATION RULES:
        - Not all outputs/inputs require UI component mappings
        - Special outputs like 'quotation_cost' can be internal values used as inputs and outputs in other steps
-       - Internal values can be passed between steps without UI components , so just IGNORE them
+       - Internal values can be passed between steps without UI components , so just DO NOT suggest any changes for them.
        - Examples of internal values:
-         * quotation_cost: Generated as output, used as input in other steps
-         * project_detail: Used as input without UI mapping
-         * gathered_project_input: Internal state passed between steps 
-         * reviewed_project_input: Internal state passed between steps  
+         * quotation_cost
+         * project_detail
+         * gathered_project_input
+         * reviewed_project_input
+         * client_follow_up_questions 
+         * follow_up_questions
+         * client_followup_question_sample_answers
 
     2. OUTPUT TO UI COMPONENT MAPPING RULES:
        - Most outputs in config map to specific UI components
        - Standard Output Pattern:
-         * Base output name (e.g., 'generated_prd') MUST map to TWO components:
+         * Base output name MUST map to TWO components:
            - base_name_text: For textbox display
            - base_name_markdown: For markdown display
        - Special Case - Mandays Pattern:
-         * Mandays outputs (e.g., 'generated_plan_test_mandays') MUST map to ONE component:
+         * Mandays outputs MUST map to ONE component:
            - base_name_dataframe: For dataframe display
            
     3. INPUT TO UI COMPONENT MAPPING RULES:
@@ -131,12 +135,12 @@ def analyze_config_structure(config_content: str, project_content: str, current_
 
     4. NAMING CONVENTION (Source of Truth: Config File)
        MUST CHECK:
-       - Prompt names (e.g., "generate_dev_components" --> "generate_development_components")
-       - Component names (e.g., "generated_Tech_SOW_text" --> "generated_technical_sow_text")
+       - Prompt names 
+       - Component names 
        - Input/Output references
        
        WHERE TO CHECK:
-       - step_handlers.py: Function names, component references
+       - event_handlers.py: Function names, component references
        - Project.py: Method names, mappings
        - all_components dictionary: Component keys
        
@@ -146,6 +150,33 @@ def analyze_config_structure(config_content: str, project_content: str, current_
          * Outputs to both _text and _markdown (or _dataframe for mandays)
        - all_components dictionary keys must exactly match config naming
        - Function parameters should align with Project.py signatures
+    
+    6. PROMPTCONFIG CHANGES:
+       A. REMOVED PROMPTCONFIGS:
+          - When a PromptConfig is removed from config:
+            * ALL references to that prompt must be removed from Project.py
+            * ALL UI components associated with that prompt must be removed from event_handlers.py
+            * ALL method signatures and return values must be updated accordingly
+            * ALL dependent code that referenced the removed components must be updated
+          - Check for:
+            * Orphaned UI components
+            * Unused function parameters
+            * Outdated return values
+            * Broken event handler mappings
+
+       B. NEW PROMPTCONFIGS:
+          - When a new PromptConfig is added to config:
+            * Add corresponding execute_prompt() call in Project.py
+            * Add UI component references in event_handlers.py
+            * Add new step button handler if step is specified
+            * Add necessary input/output mappings
+          - Check for:
+            * Required method signatures
+            * Input/output component creation
+            * Event handler connections
+            * Step button integration
+            * State management updates
+
 
     Context Files:
     1. Current Config:
@@ -154,9 +185,31 @@ def analyze_config_structure(config_content: str, project_content: str, current_
     2. Project.py:
     {project_content}
 
-    3. step_handlers.py:
+    3. event_handlers.py:
     {current_handlers}
-
+    
+    CRITICAL ANALYSIS RULES:
+    1. RENAMED PROMPTS:
+       - First check if a component missing from config exists with a similar name
+       - Compare component names using similarity matching (e.g., generate_Tech_SOW vs generate_Technical_SOW)
+       - If a similar name exists in config, treat it as a rename operation
+       - Add it to the 'renamed_prompts' section with old and new names
+       - Do NOT mark it as a removed prompt
+       
+    2. REMOVED PROMPTS:
+       - Only mark a prompt as removed if NO similar name exists in config
+       - If unsure, prefer rename over removal
+       - For confirmed removals:
+         * List ALL code locations that need cleanup
+         * Include exact file paths and line numbers if possible
+         * Specify what code needs to be removed
+
+    3. NAMING UPDATES:
+       - When a component name changes in config:
+         * Update ALL references to match exactly
+         * Maintain consistent casing and formatting
+         * Update both component names and method references
+    
     Return a JSON analysis with this EXACT structure as shown as an example below:
     {{
         "component_mappings": {{
@@ -177,12 +230,25 @@ def analyze_config_structure(config_content: str, project_content: str, current_
                     "is_valid": false,
                     "issues": ["detailed_issue_description"]
                 }}
+            ],
+            "buttons": [
+                {{
+                    "button_id": "button_name",
+                    "type": "step|action|recalculate|upload",
+                    "handler": "handler_function_name",
+                    "inputs": ["input1", "input2"],
+                    "outputs": ["output1", "output2"],
+                    "state_updates": ["state1", "state2"],
+                    "is_valid": false,
+                    "issues": ["detailed_issue_description"]
+                }}
             ]
         }},
-        "step_handlers": {{
-            "steps": [
+       "event_handlers": {{
+            "handlers": [
                 {{
-                    "step": "step_number",
+                    "button_id": "button_name",
+                    "type": "step|recalculate|upload",
                     "input_mappings": {{
                         "is_valid": false,
                         "issues": ["detailed_issue_description"]
@@ -190,6 +256,88 @@ def analyze_config_structure(config_content: str, project_content: str, current_
                     "output_mappings": {{
                         "is_valid": false,
                         "issues": ["detailed_issue_description"]
+                    }},
+                    "state_mappings": {{
+                        "is_valid": false,
+                        "issues": ["detailed_issue_description"]
+                    }}
+                }}
+            ]
+        }},
+        "prompt_changes": {{
+            "removed_prompts": [
+                {{
+                    "prompt_name": "name_of_removed_prompt",
+                    "affected_components": ["component1", "component2"],
+                    "affected_handlers": ["handler1", "handler2"],
+                    "cleanup_required": {{
+                        "Project.py": [
+                            {{
+                                "type": "method_removal|signature_update",
+                                "location": "exact_location",
+                                "code_to_remove": "code_snippet",
+                                "reason": "detailed_explanation"
+                            }}
+                        ],
+                        "event_handlers.py": [
+                            {{
+                                "type": "component_removal|handler_update",
+                                "location": "exact_location",
+                                "code_to_remove": "code_snippet",
+                                "reason": "detailed_explanation"
+                            }}
+                        ]
+                    }}
+                }}
+            ],
+              "renamed_prompts": [
+                {{
+                    "old_name": "old_prompt_name",
+                    "new_name": "new_prompt_name",
+                    "affected_files": {{
+                        "Project.py": [
+                            {{
+                                "type": "method_rename",
+                                "location": "method references",
+                                "old_code": "old_code_snippet",
+                                "new_code": "new_code_snippet",
+                                "reason": "Prompt name updated in config"
+                            }}
+                        ],
+                        "event_handlers.py": [
+                            {{
+                                "type": "component_rename",
+                                "location": "component references",
+                                "old_code": "old_code_snippet",
+                                "new_code": "new_code_snippet",
+                                "reason": "Prompt name updated in config"
+                            }}
+                        ]
+                    }}
+                }}
+            ]
+              "new_prompts": [
+                {{
+                    "prompt_name": "name_of_new_prompt",
+                    "required_components": ["component1", "component2"],
+                    "required_handlers": ["handler1", "handler2"],
+                    "additions_required": {{
+                        "Project.py": [
+                            {{
+                                "type": "method_addition|signature_update",
+                                "location": "exact_location",
+                                "code_to_add": "code_snippet",
+                                "reason": "detailed_explanation"
+                            }}
+                        ],
+                        "event_handlers.py": [
+                            {{
+                                "type": "component_addition|handler_creation",
+                                "location": "exact_location",
+                                "code_to_add": "code_snippet",
+                                "reason": "detailed_explanation"
+                            }}
+                        ]
                     }}
                 }}
             ]
@@ -202,10 +350,10 @@ def analyze_config_structure(config_content: str, project_content: str, current_
                     "reason": "detailed_explanation"
                 }}
             ],
-            "step_handlers.py": [
+            "event_handlers.py": [
                 {{
-                    "step": "step_number",
-                    "type": "component|input|output",
+                    "button_id": "button_name",
+                    "type": "step|recalculate|upload",
                     "current": "current_code",
                     "required": "required_code",
                     "reason": "detailed_explanation"
@@ -231,7 +379,7 @@ def analyze_config_structure(config_content: str, project_content: str, current_
         print("Raw response:", result)
         return {
             "component_mappings": {"outputs": [], "inputs": []},
-            "step_handlers": {"steps": []},
+            "event_handlers": {"steps": []},
             "required_updates": {}
         }
 
@@ -244,38 +392,86 @@ def generate_code_updates(analysis_result: Dict) -> Dict:
     Analysis Result:
     {json.dumps(analysis_result, indent=2)}
 
-    Generate a JSON response with this EXACT structure as shown as an example below:
+    CRITICAL RULES:
+    1. REMOVED PROMPTS:
+       When a prompt is removed from config:
+       - REMOVE all related component references
+       - REMOVE all related handler code
+       - DO NOT suggest adding new components
+       - DO NOT suggest new mappings
+       
+    2. NEW PROMPTS:
+       When a new prompt is added to config:
+       - ADD corresponding execute_prompt() method in Project.py
+       - ADD UI component references in event_handlers.py
+       - ADD new step button handler if step is specified
+       - ADD necessary input/output mappings
+       - FOLLOW exact naming from config
+       - ENSURE all required components are created
+       
+    3. Processing Order:
+       a) Process removals FIRST (prompt_changes.removed_prompts)
+       b) Then process additions (prompt_changes.new_prompts)
+       c) Finally process other updates
+
+    Generate a JSON response with this EXACT structure:
     {{
         "Project.py": {{
-            "updates": []
-        }},
-        "step_handlers.py": {{
             "updates": [
                 {{
-                    "type": "output",  # Must be one of: component, input, output
-                    "location": "step_handlers.py - Step 3 Output Mapping",
-                    "explanation": "Mismatch in prompt name; 'generate_Tech_SOW' should be 'generate_Technical_SOW'",
-                    "old_code": "output_name = 'generate_Tech_SOW'",
-                    "new_code": "output_name = 'generate_Technical_SOW'"
-                }},
+                    "type": "removal|addition|update",  # Must specify type
+                    "location": "exact_location",
+                    "explanation": "Removing/Adding/Updating for prompt 'prompt_name'",
+                    "old_code": "existing code to remove or update",
+                    "new_code": "new code to add or update with",
+                    "template": {{  # Only for additions
+                        "method_signature": "def method_name(self, ...)",
+                        "docstring": "\"\"\"Method documentation\"\"\"",
+                        "implementation": "method implementation",
+                        "return_statement": "return statement"
+                    }}
+                }}
+            ]
+        }},
+        "event_handlers.py": {{
+            "updates": [
                 {{
-                    "type": "component",
-                    "location": "step_handlers.py - Step 3 Component Names",
-                    "explanation": "Component names should reflect the correct prompt name",
-                    "old_code": "generated_Tech_SOW_text, generated_Tech_SOW_markdown",
-                    "new_code": "generated_Technical_SOW_text, generated_Technical_SOW_markdown"
+                    "type": "removal|addition|update",
+                    "location": "event_handlers.py - Component References",
+                    "explanation": "Adding/Removing components for prompt 'prompt_name'",
+                    "old_code": "code to remove or update",
+                    "new_code": "new code or empty string for removals",
+                    "components": [  # Only for additions
+                        {{
+                            "name": "component_name",
+                            "type": "text|markdown|dataframe",
+                            "initial_value": "initial value"
+                        }}
+                    ],
+                    "handler": {{  # Only for additions requiring handlers
+                        "name": "handler_name",
+                        "inputs": ["input1", "input2"],
+                        "outputs": ["output1", "output2"],
+                        "implementation": "handler implementation"
+                    }}
                 }}
             ]
         }}
     }}
 
     Requirements:
-    1. Each update MUST include 'type' field with one of: component, input, output
-    2. Maintain exact function signatures
-    3. Preserve existing code structure
-    4. Only generate the specific changes needed
-    5. Include clear comments explaining changes
-    6. Each file must have an 'updates' array, even if empty
+    1. Process changes in correct order (removals -> additions -> updates)
+    2. Each update MUST specify correct type (removal|addition|update)
+    3. For removals, new_code should be empty string
+    4. For additions:
+       - Include complete method templates
+       - Include all required components
+       - Include handler implementations if needed
+       - Follow exact naming from config
+    5. Maintain exact function signatures
+    6. Preserve existing code structure
+    7. Include clear comments explaining changes
+    8. Each file must have an 'updates' array, even if empty
 
     IMPORTANT: Return ONLY the JSON object, without any markdown formatting or explanation.
     """
@@ -285,19 +481,41 @@ def generate_code_updates(analysis_result: Dict) -> Dict:
         cleaned_result = clean_ai_response(result)
         parsed_result = json.loads(cleaned_result)
         
-        # Ensure correct structure
-        if "Project.py" not in parsed_result:
-            parsed_result["Project.py"] = {"updates": []}
-        if "step_handlers.py" not in parsed_result:
-            parsed_result["step_handlers.py"] = {"updates": []}
+        # Ensure correct structure and validate updates
+        for file_name in ["Project.py", "event_handlers.py"]:
+            if file_name not in parsed_result:
+                parsed_result[file_name] = {"updates": []}
             
-        # Ensure updates array exists and has required fields
-        for file_updates in parsed_result.values():
-            if "updates" not in file_updates:
-                file_updates["updates"] = []
-            for update in file_updates["updates"]:
+            # Ensure updates array exists
+            if "updates" not in parsed_result[file_name]:
+                parsed_result[file_name]["updates"] = []
+                
+            # Validate each update
+            for update in parsed_result[file_name]["updates"]:
                 if "type" not in update:
-                    update["type"] = "component"  # Default type if missing
+                    update["type"] = "update"  # Default type if missing
+                if update["type"] == "removal" and update.get("new_code"):
+                    update["new_code"] = ""  # Ensure removals have empty new_code
+                if update["type"] == "addition":
+                    # Ensure template exists for Project.py additions
+                    if file_name == "Project.py" and "template" not in update:
+                        update["template"] = {
+                            "method_signature": "",
+                            "docstring": "",
+                            "implementation": "",
+                            "return_statement": ""
+                        }
+                    # Ensure components and handler exist for event_handlers.py additions
+                    if file_name == "event_handlers.py":
+                        if "components" not in update:
+                            update["components"] = []
+                        if "handler" not in update:
+                            update["handler"] = {
+                                "name": "",
+                                "inputs": [],
+                                "outputs": [],
+                                "implementation": ""
+                            }
                     
         return parsed_result
         
@@ -306,75 +524,132 @@ def generate_code_updates(analysis_result: Dict) -> Dict:
         print("Raw response:", result)
         return {
             "Project.py": {"updates": []},
-            "step_handlers.py": {"updates": []}
+            "event_handlers.py": {"updates": []}
         }
 
-def apply_project_updates(updates: List[Dict]) -> bool:
-    """Apply updates to Project.py"""
+def apply_updates_with_ai(file_path: str, updates: List[Dict]) -> bool:
+    """Third AI: Intelligently applies code updates while preserving structure"""
     try:
-        with open('Project.py', 'r') as f:
-            content = f.read()
+        with open(file_path, 'r') as f:
+            current_code = f.read()
             
-        for update in updates:
-            if update['type'] == 'method':
-                # Add new methods before the last class method
-                class_end = content.rfind('\n\n')
-                if class_end == -1:
-                    return False
-                content = content[:class_end] + '\n' + update['new_code'] + content[class_end:]
-                
-            elif update['type'] == 'mapping':
-                # Update INPUT_MAPPINGS
-                input_mappings_start = content.find('INPUT_MAPPINGS = {')
-                if input_mappings_start != -1:
-                    input_mappings_end = content.find('}', input_mappings_start)
-                    if input_mappings_end != -1:
-                        # Replace old mapping with new one
-                        content = content[:input_mappings_start] + update['new_code'] + content[input_mappings_end + 1:]
-        
-        with open('Project.py', 'w') as f:
-            f.write(content)
+        # Store original empty lines and their positions
+        original_lines = current_code.splitlines(keepends=True)
+        empty_line_indices = [i for i, line in enumerate(original_lines) if not line.strip()]
+        trailing_spaces = [len(line) - len(line.rstrip()) for line in original_lines]
             
-        return True
+        update_prompt = f"""
+        You are an expert code updater. Your task is to apply updates to the code while perfectly preserving its structure and formatting.
         
-    except Exception as e:
-        print(f"Error applying Project.py updates: {str(e)}")
-        return False
-
-def apply_handler_updates(updates: List[Dict]) -> bool:
-    """Apply updates to step_handlers.py"""
-    try:
-        with open('step_handlers.py', 'r') as f:
-            content = f.read()
-            
+        CRITICAL RULES:
+        1. DO NOT modify any line breaks or indentation
+        2. DO NOT add or remove any lines
+        3. DO NOT change any code structure or formatting
+        4. ONLY replace the exact old_code with new_code
+        5. Return the COMPLETE file with ALL original formatting preserved
+        6. Keep ALL whitespace, commas, and brackets exactly as they are
+        7. Return ONLY the code, no explanations or markdown
+        8. Preserve ALL empty lines in their exact positions
+        9. Keep ALL trailing spaces at the end of each line
+        
+        Current Code:
+        {current_code}
+        
+        Required Updates:
+        {json.dumps(updates, indent=2)}
+        
+        IMPORTANT: 
+        - Empty lines are at these positions: {empty_line_indices}
+        - Each line must maintain its exact trailing spaces
+        - Return the complete file with perfect structure preservation
+        """
+        
+        # Get AI-generated update
+        result = call_o1_mini(update_prompt)
+        updated_code = clean_code_response(result)
+        
+        # Split into lines while preserving endings
+        updated_lines = updated_code.splitlines(keepends=True)
+        
+        # Handle line structure preservation
+        updated_code = preserve_line_structure(original_lines, updated_lines, empty_line_indices, trailing_spaces)
+        
+        # Verify updates were applied using flexible pattern matching
         for update in updates:
-            if not update.get('type') or not update.get('old_code') or not update.get('new_code'):
-                print(f"Skipping invalid update: {update}")
+            old_code = update.get('old_code', '').strip()
+            new_code = update.get('new_code', '').strip()
+            
+            # Skip empty updates
+            if not old_code and not new_code:
                 continue
                 
-            if update['type'] == 'component':
-                # Update component references
-                content = content.replace(update['old_code'], update['new_code'])
-            elif update['type'] == 'input':
-                # Update input parameters
-                pattern = rf"inputs=\[([^\]]*)\]"
-                replacement = f"inputs=[{update['new_code']}]"
-                content = re.sub(pattern, replacement, content)
-            elif update['type'] == 'output':
-                # Update output parameters
-                old_code = update['old_code'].strip()
-                new_code = update['new_code'].strip()
-                content = content.replace(old_code, new_code)
-                    
-        with open('step_handlers.py', 'w') as f:
-            f.write(content)
+            # For renames, extract the core identifiers
+            if update.get('type') == 'rename':
+                old_identifier = extract_identifier(old_code)
+                new_identifier = extract_identifier(new_code)
+                
+                if old_identifier in updated_code:
+                    print(f"Error: Old identifier '{old_identifier}' still present")
+                    return False
+                if new_identifier not in updated_code:
+                    print(f"Error: New identifier '{new_identifier}' not found")
+                    return False
+            else:
+                # For other updates, check if the change was applied
+                if old_code and old_code in updated_code:
+                    print(f"Error: Old code still present")
+                    return False
+                if new_code and new_code not in updated_code:
+                    print(f"Error: New code not found")
+                    return False
+        
+        # Write the updated code
+        with open(file_path, 'w') as f:
+            f.write(updated_code)
             
         return True
         
     except Exception as e:
-        print(f"Error applying handler updates: {str(e)}")
+        print(f"Error applying AI updates: {str(e)}")
         return False
 
+def clean_code_response(result: str) -> str:
+    """Clean up the AI response code"""
+    if '```python' in result:
+        result = result.split('```python', 1)[1]
+    if '```' in result:
+        result = result.split('```', 1)[0]
+    return result.strip()
+
+def preserve_line_structure(original_lines: List[str], updated_lines: List[str], 
+                          empty_indices: List[int], trailing_spaces: List[int]) -> str:
+    """Preserve the original code structure"""
+    if len(original_lines) != len(updated_lines):
+        fixed_lines = []
+        for i in range(len(original_lines)):
+            if i in empty_indices:
+                fixed_lines.append('\n')
+            else:
+                line = updated_lines[i].rstrip() if i < len(updated_lines) else ''
+                line = line + ' ' * trailing_spaces[i]
+                fixed_lines.append(line + '\n')
+        return ''.join(fixed_lines)
+    return ''.join(updated_lines)
+
+def extract_identifier(code: str) -> str:
+    """Extract the core identifier from a code snippet"""
+    # Remove common code patterns
+    code = code.replace('self.', '')
+    code = code.replace('execute_prompt(', '')
+    code = code.replace('generated_', '')
+    code = code.replace('all_components[', '')
+    code = code.replace(']', '')
+    code = code.replace('"', '')
+    code = code.replace("'", '')
+    
+    # Return the cleaned identifier
+    return code.strip()
+
 def extract_config_without_prompts() -> str:
     """Extract config file content without prompts"""
     try:
@@ -428,7 +703,7 @@ def main():
             
         with open('Project.py', 'r') as f:
             project_content = f.read()
-        with open('step_handlers.py', 'r') as f:
+        with open('event_handlers.py', 'r') as f:
             current_handlers = f.read()
 
         # First AI: Analyze current structure
@@ -443,19 +718,25 @@ def main():
         print("\nConfiguration Analysis:")
         print("1. Output Mappings:", len(analysis_result['component_mappings']['outputs']))
         print("2. Input Mappings:", len(analysis_result['component_mappings']['inputs']))
-        print("3. Step Handlers:", len(analysis_result['step_handlers']['steps']))
+        print("3. Event Handlers:", len(analysis_result['event_handlers']['handlers']))
 
         if analysis_result.get('required_updates'):
             print("\nRequired Updates:")
             for file, updates in analysis_result['required_updates'].items():
                 print(f"\n{file}:")
                 for update in updates:
-                    print(f"- {update['type']}: {update['reason']}")
+                    # Safely access update information with fallbacks
+                    update_type = update.get('type', 'Unknown')
+                    explanation = update.get('reason', update.get('explanation', 'No explanation provided'))
+                    print(f"- {update_type}: {explanation}")
 
         if analysis_result.get('validation_issues'):
             print("\nValidation Issues:")
             for issue in analysis_result['validation_issues']:
-                print(f"- {issue['type']} ({issue['severity']}): {issue['description']}")
+                severity = issue.get('severity', 'unknown')
+                description = issue.get('description', 'No description provided')
+                issue_type = issue.get('type', 'Unknown')
+                print(f"- {issue_type} ({severity}): {description}")
 
         # Get user confirmation
         confirm = input("\nProceed with generating code updates? (y/n): ")
@@ -472,12 +753,12 @@ def main():
         for file, updates in code_updates.items():
             print(f"\n{file}:")
             for update in updates['updates']:
-                print(f"- Location: {update['location']}")
-                print(f"  Explanation: {update['explanation']}")
+                print(f"- Location: {update.get('location', 'Unknown location')}")
+                print(f"  Explanation: {update.get('explanation', 'No explanation provided')}")
                 print("  Old code:")
-                print(f"```\n{update['old_code']}\n```")
+                print(f"```\n{update.get('old_code', 'No old code provided')}\n```")
                 print("  New code:")
-                print(f"```\n{update['new_code']}\n```")
+                print(f"```\n{update.get('new_code', 'No new code provided')}\n```")
 
         # Final confirmation
         confirm = input("\nApply these code updates? (y/n): ")
@@ -488,9 +769,9 @@ def main():
         # Apply updates
         success = True
         if 'Project.py' in code_updates:
-            success &= apply_project_updates(code_updates['Project.py']['updates'])
-        if 'step_handlers.py' in code_updates:
-            success &= apply_handler_updates(code_updates['step_handlers.py']['updates'])
+            success &= apply_updates_with_ai('Project.py' ,code_updates['Project.py']['updates'])
+        if 'event_handlers.py' in code_updates:
+            success &= apply_updates_with_ai('event_handlers.py' , code_updates['event_handlers.py']['updates'])
 
         if success:
             print("Successfully updated all files")
@@ -503,4 +784,4 @@ def main():
         print("Traceback:", traceback.format_exc())
 
 if __name__ == "__main__":
-    main()
+    main()

code_updater.py

@@ -95,9 +95,9 @@ You are an expert software engineer performing a comprehensive analysis. You mus
 
 4. Prompt Name Validation : 
    - Prompt names in config are the source of truth 
-   - Any references to these names in Project.py and step_handlers.py must match EXACTLY
+   - Any references to these names in Project.py and event_handlers.py must match EXACTLY
    - Check for:
-         * Function names in step_handlers.py
+         * Function names in event_handlers.py
          * Component references in all_components dictionary
          * Input/output mappings in step handlers
          * Method names in Project.py
@@ -120,7 +120,7 @@ Context Files:
 2. Project.py (SOURCE OF TRUTH FOR FUNCTION PARAMETERS):
 {project_content}
 
-3. step_handlers.py (NEEDS VALIDATION):
+3. event_handlers.py (NEEDS VALIDATION):
 {current_handlers}
 
 Analysis Requirements:
@@ -238,9 +238,9 @@ def analyze_prompts_and_generate_handlers() -> Dict[int, str]:
             project_content = f.read()
         print("Project.py content loaded")
             
-        with open('step_handlers.py', 'r') as f:
+        with open('event_handlers.py', 'r') as f:
             current_handlers = f.read()
-        print("step_handlers.py content loaded")
+        print("event_handlers.py content loaded")
         
         # Do one analysis for all files
         print("\nAnalyzing all files...")
@@ -309,7 +309,7 @@ def main():
         if update.lower() == 'y':
             try:
                 # Read current content
-                with open('step_handlers.py', 'r') as f:
+                with open('event_handlers.py', 'r') as f:
                     content = f.read()
                 
                 # Find and replace the specific step handler
@@ -324,13 +324,13 @@ def main():
                 updated_content = content[:match.start()] + code.strip() + content[match.end():]
                 
                 # Write back to file
-                with open('step_handlers.py', 'w') as f:
+                with open('event_handlers.py', 'w') as f:
                     f.write(updated_content)
                     
-                print(f"Successfully updated Step {step} handler in step_handlers.py")
+                print(f"Successfully updated Step {step} handler in event_handlers.py")
                 
                 # Verify the write
-                with open('step_handlers.py', 'r') as f:
+                with open('event_handlers.py', 'r') as f:
                     new_content = f.read()
                 if code.strip() in new_content:
                     print("Verified: Update successful")

step_handlers.py → event_handlers.py

@@ -1,81 +1,81 @@
-from Project import *
-from google_drive import *
+from Project import * 
+from google_drive import * 
 
-def setup_all_handlers(step_buttons, all_components, page_progress_update, quotation_cost , page_recalc_btn , page_upload_btn , drive_folder_name):
-    """Set up all step handlers with the provided UI components"""
-    step_buttons['Step 1 : Scope & Components'].click(
-        fn=state.quotation_project.generate_prd_and_components,
-        inputs=[],
-        outputs=[
-            all_components['generate_prd']['generated_prd_text'],
-            all_components['generate_prd']['generated_prd_markdown'],
-            all_components['generate_plan_test_components']['generated_plan_test_components_text'],
-            all_components['generate_plan_test_components']['generated_plan_test_components_markdown'],
-            all_components['generate_dev_components']['generated_dev_components_text'],
-            all_components['generate_dev_components']['generated_dev_components_markdown'],
-            page_progress_update
-        ],
-    )
-
-    step_buttons['Step 2 : Planning & Testing Mandays'].click(
-        fn=state.quotation_project.generate_mandays_and_quotation,
-        inputs=[
-            all_components['generate_plan_test_components']['generated_plan_test_components_text'],
-            all_components['generate_dev_components']['generated_dev_components_text']
-        ],
-        outputs=[
-            all_components['generate_plan_test_mandays']['generated_plan_test_mandays_dataframe'],
-            all_components['generate_dev_mandays']['generated_dev_mandays_dataframe'],
-            page_progress_update,
-            quotation_cost
-        ],
-    )
+def setup_all_handlers(step_buttons, all_components, page_progress_update, quotation_cost , page_recalc_btn , page_upload_btn , drive_folder_name): 
+    """Set up all step handlers with the provided UI components""" 
+    step_buttons['Step 1 : Scope & Components'].click( 
+        fn=state.quotation_project.generate_prd_and_components, 
+        inputs=[], 
+        outputs=[ 
+            all_components['generate_prd']['generated_prd_text'], 
+            all_components['generate_prd']['generated_prd_markdown'], 
+            all_components['generate_plan_test_components']['generated_plan_test_components_text'], 
+            all_components['generate_plan_test_components']['generated_plan_test_components_markdown'], 
+            all_components['generate_dev_components']['generated_dev_components_text'], 
+            all_components['generate_dev_components']['generated_dev_components_markdown'], 
+            page_progress_update 
+        ], 
+    ) 
 
-    step_buttons['Step 3 : SOW Doc'].click(
-    fn=state.quotation_project.generate_sow,
-    inputs=[
-        all_components['generate_prd']['generated_prd_text'],
-        all_components['generate_plan_test_components']['generated_plan_test_components_text'],
-        all_components['generate_dev_components']['generated_dev_components_text'],
-        quotation_cost
-    ],
-    outputs=[
-        all_components['generate_BD_SOW']['generated_BD_SOW_text'],
-        all_components['generate_BD_SOW']['generated_BD_SOW_markdown'],
-        all_components['generate_Tech_SOW']['generated_Tech_SOW_text'],
-        all_components['generate_Tech_SOW']['generated_Tech_SOW_markdown'],
-        page_progress_update
-    ],
-)
-    
-    page_recalc_btn.click(
-        fn=state.quotation_project.recalculate_cost,
-        inputs=[
-            all_components['generate_plan_test_mandays']['generated_plan_test_mandays_dataframe'],
-            all_components['generate_dev_mandays']['generated_dev_mandays_dataframe'],
+    step_buttons['Step 2 : Planning & Testing Mandays'].click( 
+        fn=state.quotation_project.generate_mandays_and_quotation, 
+        inputs=[ 
+            all_components['generate_plan_test_components']['generated_plan_test_components_text'], 
+            all_components['generate_dev_components']['generated_dev_components_text'] 
         ], 
-        outputs=[
-            quotation_cost,
-            page_progress_update
-        ]
-    )
-    
-    page_upload_btn.click(
-        fn=upload_to_gdrive,
-        inputs=[
-            drive_folder_name,
-            all_components['generate_prd']['generated_prd_markdown'],
-            all_components['generate_plan_test_components']['generated_plan_test_components_markdown'],
-            all_components['generate_dev_components']['generated_dev_components_markdown'],
-            all_components['generate_plan_test_mandays']['generated_plan_test_mandays_dataframe'],
-            all_components['generate_dev_mandays']['generated_dev_mandays_dataframe'],
-            quotation_cost,
-            all_components['generate_BD_SOW']['generated_BD_SOW_markdown'],
-            all_components['generate_Tech_SOW']['generated_Tech_SOW_markdown'],
+        outputs=[ 
+            all_components['generate_plan_test_mandays']['generated_plan_test_mandays_dataframe'], 
+            all_components['generate_dev_mandays']['generated_dev_mandays_dataframe'], 
+            page_progress_update, 
+            quotation_cost 
         ], 
-        outputs=[
-            page_progress_update
-        ]
-    )
-    
-    
+    ) 
+
+    step_buttons['Step 3 : SOW Doc'].click( 
+    fn=state.quotation_project.generate_sow, 
+    inputs=[ 
+        all_components['generate_prd']['generated_prd_text'], 
+        all_components['generate_plan_test_components']['generated_plan_test_components_text'], 
+        all_components['generate_dev_components']['generated_dev_components_text'], 
+        quotation_cost 
+    ], 
+    outputs=[ 
+        all_components['generate_BD_SOW']['generated_BD_SOW_text'], 
+        all_components['generate_BD_SOW']['generated_BD_SOW_markdown'], 
+        all_components['generate_Tech_SOW']['generated_Tech_SOW_text'], 
+        all_components['generate_Tech_SOW']['generated_Tech_SOW_markdown'], 
+        page_progress_update 
+    ], 
+) 
+
+    page_recalc_btn.click( 
+        fn=state.quotation_project.recalculate_cost, 
+        inputs=[ 
+            all_components['generate_plan_test_mandays']['generated_plan_test_mandays_dataframe'], 
+            all_components['generate_dev_mandays']['generated_dev_mandays_dataframe'], 
+        ],  
+        outputs=[ 
+            quotation_cost, 
+            page_progress_update 
+        ] 
+    ) 
+
+    page_upload_btn.click( 
+        fn=upload_to_gdrive, 
+        inputs=[ 
+            drive_folder_name, 
+            all_components['generate_prd']['generated_prd_markdown'], 
+            all_components['generate_plan_test_components']['generated_plan_test_components_markdown'], 
+            all_components['generate_dev_components']['generated_dev_components_markdown'], 
+            all_components['generate_plan_test_mandays']['generated_plan_test_mandays_dataframe'], 
+            all_components['generate_dev_mandays']['generated_dev_mandays_dataframe'], 
+            quotation_cost, 
+            all_components['generate_BD_SOW']['generated_BD_SOW_markdown'], 
+            all_components['generate_Tech_SOW']['generated_Tech_SOW_markdown'], 
+        ],  
+        outputs=[ 
+            page_progress_update 
+        ] 
+    ) 
+
+

page_prompts_config.py

@@ -134,7 +134,6 @@ PROMPTS = {
         Rewrite this for clarity while keeping all specific details, metrics, and constraints.
                     Do not include context or assumptions beyond the input provided.
                     Structure the document to ensure clarity and logical flow.
-                    Respond only in English.
         """,
         inputs=["project_detail"],
         outputs=["generated_prd"],
@@ -157,7 +156,7 @@ PROMPTS = {
             "generated_prd_markdown": UIConfig(
                 component_type=UIComponentType.MARKDOWN,
                 label="Requirements(PRD) Output",
-                visible=False,
+                visible=True,
                 show_copy_button=True
             )
         }
@@ -288,7 +287,7 @@ PROMPTS = {
             "generated_plan_test_components_markdown": UIConfig(
                 component_type=UIComponentType.MARKDOWN,
                 label="Plan & Test Components Output",
-                visible=False,
+                visible=True,
                 show_copy_button=True
             )
         }
@@ -340,7 +339,7 @@ PROMPTS = {
         step="Step 1 : Scope & Components",
         ui={
             "dev_component_prompt_editor": UIConfig(
-                component_type=UIComponentType.TEXTBOX,
+                component_type=UIComponentType.TEXTBOX, 
                 label="Development System Prompt",
                 lines=20,
                 interactive=True
@@ -354,7 +353,7 @@ PROMPTS = {
             "generated_dev_components_markdown": UIConfig(
                 component_type=UIComponentType.MARKDOWN,
                 label="Developemnt Components Output",
-                visible=False,
+                visible=True,
                 show_copy_button=True
             )
         }
@@ -392,7 +391,7 @@ PROMPTS = {
         inputs=["generated_plan_test_components"],
         outputs=["generated_plan_test_mandays"],
         model=ModelType.O1_MINI,
-        description="Generate planning and testing mandays from plan_test components table",
+        description="Generate planning and testing mandays",
         step="Step 2 : Planning & Testing Mandays",
         ui={
             "plan_test_mandays_prompt_editor": UIConfig(
@@ -451,7 +450,7 @@ PROMPTS = {
         inputs=["generated_dev_components"],
         outputs=["generated_dev_mandays"],
         model=ModelType.O1_MINI,
-        description="Generate development mandays from dev components table",
+        description="Generate development mandays",
         step="Step 2 : Planning & Testing Mandays",
         ui={
             "dev_mandays_prompt_editor": UIConfig(
@@ -664,7 +663,7 @@ PROMPTS = {
             "generated_BD_SOW_markdown": UIConfig(
                 component_type=UIComponentType.MARKDOWN,
                 label="BD SOW Doc",
-                visible=False,
+                visible=True,
                 show_copy_button=True
             )
         }
@@ -739,7 +738,7 @@ PROMPTS = {
             "generated_Tech_SOW_markdown": UIConfig(
                 component_type=UIComponentType.MARKDOWN,
                 label="Technical SOW Doc",
-                visible=False,
+                visible=True,
                 show_copy_button=True
             )
         }