Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 
-# Import generator functions and sample JSONs
 from concept_map_generator import generate_concept_map
 from synoptic_chart_generator import generate_synoptic_chart
 from radial_diagram_generator import generate_radial_diagram
@@ -10,8 +9,7 @@ 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 (no longer tied to a picker)
-    DEFAULT_BASE_COLOR = '#19191a' # This is now the fixed base color for all diagrams
+    DEFAULT_BASE_COLOR = '#19191a' 
 
     with gr.Blocks(
         title="Advanced Graph Generator",
@@ -61,16 +59,24 @@ if __name__ == "__main__":
     ) as demo:
         gr.Markdown(
             """
-            # Graphify: Generate concept maps, synoptic charts and radial diagrams from JSON!
+            # Graphify: generate diagrams from JSON super fast and easy ⚡!
             Choose a graph type and provide your JSON data to generate a visual representation.
             All graphs maintain a consistent, elegant style with rounded boxes,
             a dark-to-light color gradient, and a clean white background.
             """
         )
 
-        # 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.Row(variant="panel"):
+            gr.Markdown("### Output Format")
+            output_format_radio = gr.Radio(
+                choices=["png", "svg"],
+                value="png",
+                label="Select Output File Format",
+                interactive=True,
+                elem_id="output-format-selector"
+            )
+
+        gr.Markdown("<br>") 
 
         with gr.Tabs():
             with gr.TabItem("Concept Map"):
@@ -87,10 +93,9 @@ 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], 
+                    inputs=[json_input_cm, output_format_radio], 
                     outputs=output_cm
                 )
                 gr.Markdown("<br>") 
@@ -113,10 +118,9 @@ 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], 
+                    inputs=[json_input_sc, output_format_radio], 
                     outputs=output_sc
                 )
                 gr.Markdown("<br>")
@@ -139,10 +143,9 @@ 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], 
+                    inputs=[json_input_rd, output_format_radio], 
                     outputs=output_rd
                 )
                 gr.Markdown("<br>")
@@ -151,7 +154,7 @@ if __name__ == "__main__":
                     gr.Image(value="./images/rd1.png", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
                     gr.Image(value="./images/rd2.png", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
 
-            # New Tab for Process Flow Diagram
+
             with gr.TabItem("Process Flow"):
                 json_input_pf = gr.Textbox(
                     value=PROCESS_FLOW_JSON,
@@ -166,10 +169,9 @@ 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], 
+                    inputs=[json_input_pf, output_format_radio], 
                     outputs=output_pf
                 )
                 gr.Markdown("<br>")
@@ -178,7 +180,6 @@ if __name__ == "__main__":
                     gr.Image(value="./images/pf1.png", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
                     gr.Image(value="./images/pf2.png", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
 
-            # New Tab for WBS Diagram
             with gr.TabItem("WBS Diagram"):
                 json_input_wbs = gr.Textbox(
                     value=WBS_DIAGRAM_JSON,
@@ -193,10 +194,9 @@ 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], 
+                    inputs=[json_input_wbs, output_format_radio], 
                     outputs=output_wbs
                 )
                 gr.Markdown("<br>")