File size: 9,436 Bytes
7c43635 aac085d 1dbebbe 90ffc09 1dbebbe 90ffc09 1dbebbe aac085d 1dbebbe c43efb6 1dbebbe aac085d c43efb6 1dbebbe c43efb6 1dbebbe c43efb6 1dbebbe aac085d 1dbebbe 72de955 e7fa4b7 72de955 e7fa4b7 72de955 90ffc09 aac085d 1dbebbe 72de955 1dbebbe 90ffc09 908d9fd aac085d 908d9fd 1dbebbe 72de955 1dbebbe aac085d 908d9fd 6449b98 aac085d 908d9fd 1dbebbe 72de955 1dbebbe aac085d 908d9fd aac085d 72de955 aac085d 90ffc09 aac085d 72de955 aac085d 90ffc09 7c43635 9912372 7c43635 1dbebbe aac085d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 |
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
from process_flow_generator import generate_process_flow_diagram
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'
with gr.Blocks(
title="Advanced Graph Generator",
css="""
.gradio-container {
font-family: 'Inter', sans-serif !important;
}
.gr-tab-item {
padding: 10px 20px;
font-size: 1.1em;
font-weight: bold;
}
.gr-button {
border-radius: 8px;
box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
background-color: #4CAF50;
color: white;
padding: 10px 20px;
font-size: 1.1em;
}
.gr-button:hover {
background-color: #45a049;
}
.gr-textbox {
border-radius: 8px;
padding: 10px;
}
/* Dark mode styles, adjust if needed */
.gradio-container.dark {
--tw-bg-opacity: 1;
background-color: rgb(24 24 27 / var(--tw-bg-opacity));
color: #d4d4d8; /* text-zinc-300 */
}
.gradio-container.dark .gr-textbox {
background-color: rgb(39 39 42 / var(--tw-bg-opacity));
color: #d4d4d8;
border-color: #52525b; /* border-zinc-600 */
}
.gradio-container.dark .gr-tab-item {
color: #d4d4d8;
}
.gradio-container.dark .gr-tab-item.selected {
background-color: rgb(39 39 42 / var(--tw-bg-opacity));
color: #fff;
}
"""
) as demo:
gr.Markdown(
"""
# Graphify: Generate concept maps, synoptic charts and radial diagrams from JSON!
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.
"""
)
# 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>")
with gr.Tabs():
with gr.TabItem("Concept Map"):
json_input_cm = gr.Textbox(
value=CONCEPT_MAP_JSON,
placeholder="Paste JSON following the documented format",
label="Structured JSON Input",
lines=25
)
output_cm = gr.Image(
label="Generated Graph",
type="filepath",
show_download_button=True
)
submit_btn_cm = gr.Button("Submit")
submit_btn_cm.click(
fn=generate_concept_map,
inputs=[json_input_cm, current_base_color], # Pass the State's value
outputs=output_cm
)
gr.Markdown("<br>")
gr.Markdown("## Example Concept Maps")
with gr.Row():
gr.Image(value="./images/cm1.png", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
gr.Image(value="./images/cm2.png", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
with gr.TabItem("Synoptic Chart"):
json_input_sc = gr.Textbox(
value=SYNOPTIC_CHART_JSON,
placeholder="Paste JSON following the documented format",
label="Structured JSON Input",
lines=25
)
output_sc = gr.Image(
label="Generated Graph",
type="filepath",
show_download_button=True
)
submit_btn_sc = gr.Button("Submit")
submit_btn_sc.click(
fn=generate_synoptic_chart,
inputs=[json_input_sc, current_base_color], # Pass the State's value
outputs=output_sc
)
gr.Markdown("<br>")
gr.Markdown("## Example Synoptic Charts")
with gr.Row():
gr.Image(value="./images/sc1.png", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
gr.Image(value="./images/sc2.png", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
with gr.TabItem("Radial Diagram"):
json_input_rd = gr.Textbox(
value=RADIAL_DIAGRAM_JSON,
placeholder="Paste JSON following the documented format",
label="Structured JSON Input",
lines=25
)
output_rd = gr.Image(
label="Generated Graph",
type="filepath",
show_download_button=True
)
submit_btn_rd = gr.Button("Submit")
submit_btn_rd.click(
fn=generate_radial_diagram,
inputs=[json_input_rd, current_base_color], # Pass the State's value
outputs=output_rd
)
gr.Markdown("<br>")
gr.Markdown("## Example Radial Diagrams")
with gr.Row():
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,
placeholder="Paste JSON following the documented format",
label="Structured JSON Input",
lines=25
)
output_pf = gr.Image(
label="Generated Graph",
type="filepath",
show_download_button=True
)
submit_btn_pf = gr.Button("Submit")
submit_btn_pf.click(
fn=generate_process_flow_diagram,
inputs=[json_input_pf, current_base_color], # Pass the State's value
outputs=output_pf
)
gr.Markdown("<br>")
gr.Markdown("## Example Process Flow Diagrams")
with gr.Row():
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,
placeholder="Paste JSON following the documented format",
label="Structured JSON Input",
lines=25
)
output_wbs = gr.Image(
label="Generated Graph",
type="filepath",
show_download_button=True
)
submit_btn_wbs = gr.Button("Submit")
submit_btn_wbs.click(
fn=generate_wbs_diagram,
inputs=[json_input_wbs, current_base_color], # Pass the State's value
outputs=output_wbs
)
gr.Markdown("<br>")
gr.Markdown("## Example WBS Diagrams")
with gr.Row():
gr.Image(value="./images/wbs1.png", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
gr.Image(value="./images/wbs2.png", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
demo.launch(
mcp_server=True,
share=False,
server_port=7860,
server_name="0.0.0.0"
)
|