Update app.py

app.py

@@ -10,8 +10,8 @@ from wbs_diagram_generator import generate_wbs_diagram
 from sample_data import CONCEPT_MAP_JSON, SYNOPTIC_CHART_JSON, RADIAL_DIAGRAM_JSON, PROCESS_FLOW_JSON, WBS_DIAGRAM_JSON
 
 if __name__ == "__main__":
-    # Define the default dark color
-    DEFAULT_BASE_COLOR = '#19191a'
+    # Define the default dark color (no longer tied to a picker)
+    DEFAULT_BASE_COLOR = '#19191a' # This is now the fixed base color for all diagrams
 
     with gr.Blocks(
         title="Advanced Graph Generator",
@@ -27,7 +27,7 @@ if __name__ == "__main__":
         .gr-button {
             border-radius: 8px;
             box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-            background-color: #FFA500; /* Changed to Orange */
+            background-color: #FFA500; /* Orange color for buttons */
             color: white;
             padding: 10px 20px;
             font-size: 1.1em;
@@ -68,22 +68,9 @@ if __name__ == "__main__":
             """
         )
 
-        # Global Color Picker - Repositioned for clarity and function
-        # This will hold the *value* of the currently selected color
-        current_base_color = gr.State(DEFAULT_BASE_COLOR) 
-
-        with gr.Row(variant="panel"): # Simplified Row, removed equal_height
-            gr.Markdown("### Main Diagram Color") 
-            base_color_picker = gr.ColorPicker(
-                label="Select Base Color for Diagram",
-                value=DEFAULT_BASE_COLOR, 
-                interactive=True,
-                elem_id="main-color-picker",
-            )
-            # Update the gr.State variable whenever the color picker changes
-            base_color_picker.change(fn=lambda x: x, inputs=base_color_picker, outputs=current_base_color)
-
-        gr.Markdown("<br>") 
+        # Removed the global color picker section entirely
+        # gr.State(DEFAULT_BASE_COLOR) and its change listener are no longer needed
+        # gr.Markdown("<br>") is also removed here as it was for spacing the color picker
 
         with gr.Tabs():
             with gr.TabItem("Concept Map"):
@@ -100,9 +87,10 @@ if __name__ == "__main__":
                 )
                 submit_btn_cm = gr.Button("Submit")
                 
+                # Removed base_color from inputs, it's now hardcoded in generators
                 submit_btn_cm.click(
                     fn=generate_concept_map, 
-                    inputs=[json_input_cm, current_base_color], # Pass the State's value
+                    inputs=[json_input_cm], 
                     outputs=output_cm
                 )
                 gr.Markdown("<br>") 
@@ -125,9 +113,10 @@ if __name__ == "__main__":
                 )
                 submit_btn_sc = gr.Button("Submit")
                 
+                # Removed base_color from inputs, it's now hardcoded in generators
                 submit_btn_sc.click(
                     fn=generate_synoptic_chart, 
-                    inputs=[json_input_sc, current_base_color], # Pass the State's value
+                    inputs=[json_input_sc], 
                     outputs=output_sc
                 )
                 gr.Markdown("<br>")
@@ -150,9 +139,10 @@ if __name__ == "__main__":
                 )
                 submit_btn_rd = gr.Button("Submit")
                 
+                # Removed base_color from inputs, it's now hardcoded in generators
                 submit_btn_rd.click(
                     fn=generate_radial_diagram, 
-                    inputs=[json_input_rd, current_base_color], # Pass the State's value
+                    inputs=[json_input_rd], 
                     outputs=output_rd
                 )
                 gr.Markdown("<br>")
@@ -176,9 +166,10 @@ if __name__ == "__main__":
                 )
                 submit_btn_pf = gr.Button("Submit")
                 
+                # Removed base_color from inputs, it's now hardcoded in generators
                 submit_btn_pf.click(
                     fn=generate_process_flow_diagram, 
-                    inputs=[json_input_pf, current_base_color], # Pass the State's value
+                    inputs=[json_input_pf], 
                     outputs=output_pf
                 )
                 gr.Markdown("<br>")
@@ -202,9 +193,10 @@ if __name__ == "__main__":
                 )
                 submit_btn_wbs = gr.Button("Submit")
                 
+                # Removed base_color from inputs, it's now hardcoded in generators
                 submit_btn_wbs.click(
                     fn=generate_wbs_diagram, 
-                    inputs=[json_input_wbs, current_base_color], # Pass the State's value
+                    inputs=[json_input_wbs], 
                     outputs=output_wbs
                 )
                 gr.Markdown("<br>")