Update app.py

app.py

@@ -387,6 +387,7 @@ 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"])
 
+# Wrapper functions with proper error handling
 def trigger_correction_wrapper(model, code):
     """Wrapper for trigger correction with proper error handling."""
     try:
@@ -446,36 +447,6 @@ def validate_and_chart_object(model, original, converted):
         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(
@@ -578,15 +549,9 @@ def main():
             
             validate_trigger_button = gr.Button("🔍 Validate Correction", variant="secondary", size="lg")
             
-            # Wire up functionality
-            def validate_and_chart_trigger(model, original, corrected):
-                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
-            
+            # Wire up functionality - INSIDE the main function where UI elements are defined
             trigger_button.click(
-                fn=lambda m, c: correct_apex_trigger(m, c),
+                fn=trigger_correction_wrapper,
                 inputs=[primary_model_dropdown, trigger_input],
                 outputs=[trigger_full_response, trigger_code_output, trigger_explanation],
                 show_progress=True
@@ -674,14 +639,9 @@ def main():
             
             validate_object_button = gr.Button("🔍 Validate Conversion", variant="secondary", size="lg")
             
-            def validate_and_chart_object(model, original, converted):
-                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
-            
+            # Wire up functionality - INSIDE the main function where UI elements are defined
             object_button.click(
-                fn=lambda m, c: convert_cc_object(m, c),
+                fn=object_conversion_wrapper,
                 inputs=[primary_model_dropdown, object_input],
                 outputs=[object_full_response, object_code_output, object_explanation],
                 show_progress=True
@@ -719,7 +679,17 @@ def main():
                 value="Light"
             )
             
-            
+            # Comment out theme functionality for now to avoid conflicts
+            # theme_radio.change(
+            #     fn=get_theme_styles,
+            #     inputs=[theme_radio],
+            #     outputs=[
+            #         trigger_explanation, 
+            #         trigger_code_output,
+            #         object_explanation,
+            #         object_code_output
+            #     ]
+            # )
 
         gr.Markdown("### 📚 About This Tool")
         gr.Markdown(