Update app.py

app.py

@@ -355,21 +355,10 @@ def create_radar_chart(metrics):
     
     return fig
 
-def toggle_dark_mode(dark_mode, elements):
-    """Toggle between light and dark mode for interface elements"""
-    if dark_mode:
-        style = "background-color: #2e2e2e; color: #ffffff;"
-        code_style = "background-color: #1e1e1e; color: #e0e0e0; font-family: 'Courier New', monospace;"
-    else:
-        style = "background-color: #ffffff; color: #333333;"
-        code_style = "background-color: #f5f5f5; color: #333333; font-family: 'Courier New', monospace;"
-    
-    return [style for _ in range(len(elements) // 2)] + [code_style for _ in range(len(elements) // 2)]
+# Theme functionality has been moved to inline function in main()
 
 def main():
     with gr.Blocks(title="Salesforce B2B Commerce Migration Assistant", theme=gr.themes.Soft(primary_hue="blue")) as app:
-        # Initialize dark mode state
-        dark_mode = gr.State(False)
         
         gr.Markdown("# Salesforce B2B Commerce Migration Assistant")
         gr.Markdown("This tool helps migrate CloudCraze code to B2B Lightning Experience.")
@@ -585,18 +574,38 @@ def main():
 
         # UI Preferences
         with gr.Accordion("UI Preferences", open=False):
-            dark_mode_toggle = gr.Checkbox(label="Dark Mode", value=False)
+            theme_radio = gr.Radio(
+                label="Theme", 
+                choices=["Light", "Dark"], 
+                value="Light"
+            )
             
-            # Create a list of UI elements to update
-            ui_elements = [
-                trigger_explanation, trigger_code_output,
-                object_explanation, object_code_output
-            ]
+            # Simple theme change function that applies CSS to individual components
+            def change_theme(theme_choice):
+                if theme_choice == "Dark":
+                    explanation_style = "background-color: #2e2e2e; color: #ffffff;"
+                    code_style = "background-color: #1e1e1e; color: #e0e0e0; font-family: 'Courier New', monospace;"
+                else:  # Light
+                    explanation_style = "background-color: #ffffff; color: #333333;"
+                    code_style = "background-color: #f5f5f5; color: #333333; font-family: 'Courier New', monospace;"
+                
+                return (
+                    explanation_style,  # trigger_explanation
+                    code_style,         # trigger_code_output
+                    explanation_style,  # object_explanation
+                    code_style          # object_code_output
+                )
             
-            dark_mode_toggle.change(
-                fn=toggle_dark_mode,
-                inputs=[dark_mode_toggle, gr.State(ui_elements)],
-                outputs=ui_elements
+            # Connect the theme radio to the change_theme function
+            theme_radio.change(
+                fn=change_theme,
+                inputs=[theme_radio],
+                outputs=[
+                    trigger_explanation, 
+                    trigger_code_output,
+                    object_explanation,
+                    object_code_output
+                ]
             )
 
         gr.Markdown("### About This Tool")