ZahirJS commited on
Commit
69acbf0
·
verified ·
1 Parent(s): 66e66ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -1
app.py CHANGED
@@ -7,8 +7,9 @@ from process_flow_generator import generate_process_flow_diagram
7
  from wbs_diagram_generator import generate_wbs_diagram
8
  from timeline_generator import generate_timeline_diagram
9
  from binary_tree_generator import generate_binary_tree_diagram
 
10
 
11
- from sample_data import CONCEPT_MAP_JSON, SYNOPTIC_CHART_JSON, RADIAL_DIAGRAM_JSON, PROCESS_FLOW_JSON, WBS_DIAGRAM_JSON, TIMELINE_JSON, BINARY_TREE_JSON
12
 
13
  if __name__ == "__main__":
14
  DEFAULT_BASE_COLOR = '#19191a'
@@ -294,6 +295,36 @@ if __name__ == "__main__":
294
  with gr.Row(elem_classes=["example-images"]):
295
  gr.Image(value="./images/bt1.svg", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
296
  gr.Image(value="./images/bt2.svg", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
297
 
298
  demo.launch(
299
  mcp_server=True,
 
7
  from wbs_diagram_generator import generate_wbs_diagram
8
  from timeline_generator import generate_timeline_diagram
9
  from binary_tree_generator import generate_binary_tree_diagram
10
+ from network_graph_generator import generate_network_graph
11
 
12
+ from sample_data import CONCEPT_MAP_JSON, SYNOPTIC_CHART_JSON, RADIAL_DIAGRAM_JSON, PROCESS_FLOW_JSON, WBS_DIAGRAM_JSON, TIMELINE_JSON, BINARY_TREE_JSON, NETWORK_GRAPH_JSON
13
 
14
  if __name__ == "__main__":
15
  DEFAULT_BASE_COLOR = '#19191a'
 
295
  with gr.Row(elem_classes=["example-images"]):
296
  gr.Image(value="./images/bt1.svg", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
297
  gr.Image(value="./images/bt2.svg", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
298
+
299
+ with gr.TabItem("Network Graph"):
300
+ with gr.Row():
301
+ with gr.Column(scale=1):
302
+ json_input_ng = gr.Textbox(
303
+ value=NETWORK_GRAPH_JSON,
304
+ placeholder="Paste JSON following the documented format",
305
+ label="JSON Input",
306
+ lines=20
307
+ )
308
+ submit_btn_ng = gr.Button("Generate Network Graph", variant="primary")
309
+
310
+ with gr.Column(scale=2):
311
+ output_ng = gr.Image(
312
+ label="Generated Diagram",
313
+ type="filepath",
314
+ show_download_button=True,
315
+ height=500
316
+ )
317
+
318
+ submit_btn_ng.click(
319
+ fn=generate_network_graph,
320
+ inputs=[json_input_ng, output_format_radio],
321
+ outputs=output_ng
322
+ )
323
+
324
+ gr.Markdown("## Examples")
325
+ #with gr.Row(elem_classes=["example-images"]):
326
+ #gr.Image(value="./images/ng1.svg", label="Sample 1", show_label=True, interactive=False, height="auto", width="100%")
327
+ #gr.Image(value="./images/ng2.svg", label="Sample 2", show_label=True, interactive=False, height="auto", width="100%")
328
 
329
  demo.launch(
330
  mcp_server=True,