Update app.py

app.py

@@ -160,70 +160,190 @@ Return:
 """
     return call_llm(model, prompt)
 
+# Validation function for Apex Trigger correction
+def validate_apex_trigger(validation_model, original_code, corrected_code):
+    if not validation_model or not original_code.strip() or not corrected_code.strip():
+        return "Please provide all required inputs for validation."
+    
+    prompt = f"""
+I need you to validate and review the following Apex Trigger code migration from CloudCraze to B2B Lightning Experience.
+
+ORIGINAL CODE:
+```apex
+{original_code}
+```
+
+CORRECTED CODE:
+```apex
+{corrected_code}
+```
+
+Please review the corrected code and provide:
+1. Is the correction accurate and complete? Highlight any missed issues.
+2. Are there any potential runtime errors or performance concerns?
+3. Does it follow Salesforce best practices for B2B Lightning Experience?
+4. Would you recommend any additional improvements?
+5. Rate the quality of the correction on a scale of 1-10 with explanation.
+
+Be thorough and detailed in your assessment.
+"""
+    return call_llm(validation_model, prompt)
+
+# Validation function for object conversion
+def validate_cc_object_conversion(validation_model, original_object, converted_object):
+    if not validation_model or not original_object.strip() or not converted_object.strip():
+        return "Please provide all required inputs for validation."
+    
+    prompt = f"""
+I need you to validate and review the following CloudCraze Object conversion to B2B Lightning Experience format.
+
+ORIGINAL CLOUDCRAZE OBJECT:
+```
+{original_object}
+```
+
+CONVERTED B2B LEX OBJECT:
+```
+{converted_object}
+```
+
+Please review the conversion and provide:
+1. Is the object mapping correct and complete? Identify any missed fields or relationships.
+2. Are there any potential data migration issues or concerns?
+3. Does the converted object follow B2B Lightning Experience best practices?
+4. Would you recommend any additional improvements or optimizations?
+5. Rate the quality of the conversion on a scale of 1-10 with explanation.
+
+Be thorough and detailed in your assessment.
+"""
+    return call_llm(validation_model, prompt)
+
 # Gradio App
 def main():
     with gr.Blocks(title="Salesforce B2B Commerce Migration Assistant") as app:
         gr.Markdown("# Salesforce B2B Commerce Migration Assistant")
         gr.Markdown("This tool helps migrate CloudCraze code to B2B Lightning Experience.")
 
-        # Create a dropdown with model type labels
-        model_dropdown = gr.Dropdown(
-            choices=all_models,
-            value=anthropic_models[0],  # Default to Claude 3.7 Sonnet
-            label="Select AI Model"
-        )
+        # Create model dropdowns
+        with gr.Row():
+            with gr.Column():
+                gr.Markdown("### Primary Model")
+                primary_model_dropdown = gr.Dropdown(
+                    choices=all_models,
+                    value=anthropic_models[0],  # Default to Claude 3.7 Sonnet
+                    label="Select Primary AI Model for Conversion"
+                )
+            
+            with gr.Column():
+                gr.Markdown("### Validation Model")
+                validation_model_dropdown = gr.Dropdown(
+                    choices=all_models,
+                    value=anthropic_models[1],  # Default to Claude 3 Haiku
+                    label="Select Validation AI Model for Review",
+                    info="This model will validate and review the output from the primary model"
+                )
 
         with gr.Tab("Apex Trigger Correction"):
             gr.Markdown("### Apex Trigger Correction")
             gr.Markdown("Paste your Apex Trigger code below:")
 
             trigger_input = gr.Textbox(
-                lines=15,
+                lines=12,
                 placeholder="Paste Apex Trigger code here...",
                 label="Apex Trigger Code"
             )
             trigger_output = gr.Textbox(
-                lines=20,
+                lines=15,
                 label="Corrected Apex Trigger",
                 interactive=True
             )
             trigger_button = gr.Button("Correct Apex Trigger")
             trigger_button.click(
                 fn=correct_apex_trigger,
-                inputs=[model_dropdown, trigger_input],
+                inputs=[primary_model_dropdown, trigger_input],
                 outputs=trigger_output,
                 show_progress=True
             )
-            trigger_clear = gr.Button("Clear")
-            trigger_clear.click(lambda: "", [], trigger_output)
+            
+            gr.Markdown("### Validation Results")
+            trigger_validation_output = gr.Textbox(
+                lines=15,
+                label="Validation Assessment",
+                placeholder="Validation results will appear here after clicking 'Validate Correction'",
+                interactive=True
+            )
+            validate_trigger_button = gr.Button("Validate Correction")
+            validate_trigger_button.click(
+                fn=validate_apex_trigger,
+                inputs=[validation_model_dropdown, trigger_input, trigger_output],
+                outputs=trigger_validation_output,
+                show_progress=True
+            )
+            
+            with gr.Row():
+                trigger_clear = gr.Button("Clear Input")
+                trigger_clear.click(lambda: "", [], trigger_input)
+                
+                results_clear = gr.Button("Clear Results")
+                results_clear.click(
+                    lambda: ["", ""],
+                    [],
+                    [trigger_output, trigger_validation_output]
+                )
 
         with gr.Tab("CloudCraze Object Conversion"):
             gr.Markdown("### CloudCraze Object Conversion")
             gr.Markdown("Paste your CloudCraze Object code below:")
 
             object_input = gr.Textbox(
-                lines=15,
+                lines=12,
                 placeholder="Paste CC object code here...",
                 label="CloudCraze Object Code"
             )
             object_output = gr.Textbox(
-                lines=20,
+                lines=15,
                 label="Converted B2B LEx Object",
                 interactive=True
             )
             object_button = gr.Button("Convert Object")
             object_button.click(
                 fn=convert_cc_object,
-                inputs=[model_dropdown, object_input],
+                inputs=[primary_model_dropdown, object_input],
                 outputs=object_output,
                 show_progress=True
             )
-            object_clear = gr.Button("Clear")
-            object_clear.click(lambda: "", [], object_output)
+            
+            gr.Markdown("### Validation Results")
+            object_validation_output = gr.Textbox(
+                lines=15,
+                label="Validation Assessment",
+                placeholder="Validation results will appear here after clicking 'Validate Conversion'",
+                interactive=True
+            )
+            validate_object_button = gr.Button("Validate Conversion")
+            validate_object_button.click(
+                fn=validate_cc_object_conversion,
+                inputs=[validation_model_dropdown, object_input, object_output],
+                outputs=object_validation_output,
+                show_progress=True
+            )
+            
+            with gr.Row():
+                object_clear = gr.Button("Clear Input")
+                object_clear.click(lambda: "", [], object_input)
+                
+                object_results_clear = gr.Button("Clear Results")
+                object_results_clear.click(
+                    lambda: ["", ""],
+                    [],
+                    [object_output, object_validation_output]
+                )
 
         gr.Markdown("### About This Tool")
         gr.Markdown(
             """
+- **Primary Model**: Performs the initial code conversion or correction.
+- **Validation Model**: Reviews and validates the output from the primary model, identifying potential issues or improvements.
 - **Trigger Correction**: Fixes Apex Triggers for B2B LEx compatibility.
 - **Object Conversion**: Maps and converts CloudCraze object definitions to B2B LEx.
 - **Model Selection**: Choose from Together AI models or Anthropic's Claude models.