Update app.py

app.py

@@ -385,6 +385,95 @@ def get_theme_styles(theme_choice: str) -> Tuple[str, str, str, str]:
     theme = themes.get(theme_choice, themes["Light"])
     return (theme["explanation"], theme["code"], theme["explanation"], theme["code"])
 
+def trigger_correction_wrapper(model, code):
+    """Wrapper for trigger correction with proper error handling."""
+    try:
+        if not model or not code.strip():
+            return "Please select a model and provide code.", "", "Please provide valid inputs."
+        
+        # Create dummy progress function since Gradio progress doesn't work in lambda
+        def dummy_progress(value, desc=""):
+            pass
+            
+        return correct_apex_trigger(model, code, progress=dummy_progress)
+    except Exception as e:
+        logger.error(f"Trigger correction error: {str(e)}")
+        error_msg = f"Error processing request: {str(e)}"
+        return error_msg, "", error_msg
+
+def object_conversion_wrapper(model, code):
+    """Wrapper for object conversion with proper error handling."""
+    try:
+        if not model or not code.strip():
+            return "Please select a model and provide code.", "", "Please provide valid inputs."
+            
+        def dummy_progress(value, desc=""):
+            pass
+            
+        return convert_cc_object(model, code, progress=dummy_progress)
+    except Exception as e:
+        logger.error(f"Object conversion error: {str(e)}")
+        error_msg = f"Error processing request: {str(e)}"
+        return error_msg, "", error_msg
+
+def validate_and_chart_trigger(model, original, corrected):
+    """Wrapper for trigger validation with error handling."""
+    try:
+        if not model or not original.strip() or not corrected.strip():
+            return "Please provide all required inputs for validation.", None
+            
+        validation_text = validate_apex_trigger(model, original, corrected)
+        metrics = extract_validation_metrics(validation_text)
+        chart = create_enhanced_radar_chart(metrics) if metrics else None
+        return validation_text, chart
+    except Exception as e:
+        logger.error(f"Trigger validation error: {str(e)}")
+        return f"Validation error: {str(e)}", None
+
+def validate_and_chart_object(model, original, converted):
+    """Wrapper for object validation with error handling."""
+    try:
+        if not model or not original.strip() or not converted.strip():
+            return "Please provide all required inputs for validation.", None
+            
+        validation_text = validate_cc_object_conversion(model, original, converted)
+        metrics = extract_validation_metrics(validation_text)
+        chart = create_enhanced_radar_chart(metrics) if metrics else None
+        return validation_text, chart
+    except Exception as e:
+        logger.error(f"Object validation error: {str(e)}")
+        return f"Validation error: {str(e)}", None
+
+# In the Apex Trigger tab section:
+trigger_button.click(
+    fn=trigger_correction_wrapper,
+    inputs=[primary_model_dropdown, trigger_input],
+    outputs=[trigger_full_response, trigger_code_output, trigger_explanation],
+    show_progress=True
+)
+
+validate_trigger_button.click(
+    fn=validate_and_chart_trigger,
+    inputs=[validation_model_dropdown, trigger_input, trigger_code_output],
+    outputs=[trigger_validation_output, trigger_chart],
+    show_progress=True
+)
+
+# In the CloudCraze Object tab section:
+object_button.click(
+    fn=object_conversion_wrapper,
+    inputs=[primary_model_dropdown, object_input],
+    outputs=[object_full_response, object_code_output, object_explanation],
+    show_progress=True
+)
+
+validate_object_button.click(
+    fn=validate_and_chart_object,
+    inputs=[validation_model_dropdown, object_input, object_code_output],
+    outputs=[object_validation_output, object_chart],
+    show_progress=True
+)
+
 def main():
     """Main application entry point."""
     with gr.Blocks(