Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import os
 import time
 import requests
+import json
+import plotly.express as px
 
 # Model options for dropdown with both Together AI and Anthropic models
 together_models = [
@@ -18,7 +20,19 @@ anthropic_models = [
 
 all_models = together_models + anthropic_models
 
-# Helper to fetch API keys from environment
+VALIDATION_SCHEMA = {
+    "quality_rating": "int (1–10)",
+    "accuracy": "float (0.0–1.0)",
+    "completeness": "float (0.0–1.0)",
+    "best_practices_alignment": "float (0.0–1.0)",
+    "explanations": {
+        "quality_rating": "string",
+        "accuracy": "string",
+        "completeness": "string",
+        "best_practices_alignment": "string"
+    }
+}
+
 def get_api_key(provider):
     if provider == "together":
         api_key = os.getenv("TOGETHER_API_KEY")
@@ -33,7 +47,6 @@ def get_api_key(provider):
     else:
         raise ValueError(f"Unknown provider: {provider}")
 
-# Determine the provider based on model name
 def get_provider(model):
     if model in together_models:
         return "together"
@@ -42,14 +55,12 @@ def get_provider(model):
     else:
         raise ValueError(f"Unknown model: {model}")
 
-# Call Together AI via REST API
 def call_together_api(model, prompt, temperature=0.7, max_tokens=1500):
     api_key = get_api_key("together")
     system_message = (
         "You are a Salesforce development expert specializing in B2B Commerce migrations,"
         " CloudCraze to B2B Lightning Experience conversions, and Apex code optimization."
     )
-    start = time.time()
     try:
         headers = {
             "Authorization": f"Bearer {api_key}",
@@ -78,14 +89,12 @@ def call_together_api(model, prompt, temperature=0.7, max_tokens=1500):
     except Exception as e:
         return f"Error calling Together AI API: {e}"
 
-# Call Anthropic API
 def call_anthropic_api(model, prompt, temperature=0.7, max_tokens=1500):
     api_key = get_api_key("anthropic")
     system_message = (
         "You are a Salesforce development expert specializing in B2B Commerce migrations,"
         " CloudCraze to B2B Lightning Experience conversions, and Apex code optimization."
     )
-    start = time.time()
     try:
         headers = {
             "x-api-key": api_key,
@@ -114,7 +123,6 @@ def call_anthropic_api(model, prompt, temperature=0.7, max_tokens=1500):
     except Exception as e:
         return f"Error calling Anthropic API: {e}"
 
-# Unified function to call the appropriate API based on the model
 def call_llm(model, prompt, temperature=0.7, max_tokens=1500):
     provider = get_provider(model)
     if provider == "together":
@@ -124,7 +132,6 @@ def call_llm(model, prompt, temperature=0.7, max_tokens=1500):
     else:
         return f"Error: Unknown provider for model {model}"
 
-# Build prompt for trigger correction
 def correct_apex_trigger(model, trigger_code):
     if not trigger_code.strip():
         return "Please provide Apex Trigger code to correct."
@@ -133,226 +140,4 @@ Please analyze and correct the following Apex Trigger code for migration from Cl
 Identify any issues, optimize it according to best practices, and provide the corrected code.
 
 ```apex
-{trigger_code}
-```
-
-Please return the corrected code with explanations of changes.
-"""
-    return call_llm(model, prompt)
-
-# Build prompt for object conversion
-def convert_cc_object(model, cc_object_code):
-    if not cc_object_code.strip():
-        return "Please provide CloudCraze Object code to convert."
-    prompt = f"""
-Please convert the following CloudCraze Object code to B2B Lightning Experience format.
-Identify the corresponding B2B LEx system object, map fields, and provide the full definition.
-
-```
-{cc_object_code}
-```
-
-Return:
-1. Equivalent B2B LEx object name
-2. Field mappings
-3. Full implementation code
-4. Additional steps if needed
-"""
-    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 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=12,
-                placeholder="Paste Apex Trigger code here...",
-                label="Apex Trigger Code"
-            )
-            trigger_output = gr.Textbox(
-                lines=15,
-                label="Corrected Apex Trigger",
-                interactive=True
-            )
-            trigger_button = gr.Button("Correct Apex Trigger")
-            trigger_button.click(
-                fn=correct_apex_trigger,
-                inputs=[primary_model_dropdown, trigger_input],
-                outputs=trigger_output,
-                show_progress=True
-            )
-            
-            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=12,
-                placeholder="Paste CC object code here...",
-                label="CloudCraze Object Code"
-            )
-            object_output = gr.Textbox(
-                lines=15,
-                label="Converted B2B LEx Object",
-                interactive=True
-            )
-            object_button = gr.Button("Convert Object")
-            object_button.click(
-                fn=convert_cc_object,
-                inputs=[primary_model_dropdown, object_input],
-                outputs=object_output,
-                show_progress=True
-            )
-            
-            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.
-
-Always review AI-generated code before production use.
-            """
-        )
-
-    app.launch()
-
-if __name__ == "__main__":
-    main()
+{trigger_code}