Update process_flow_generator.py

process_flow_generator.py

@@ -3,7 +3,7 @@ import json
 from tempfile import NamedTemporaryFile
 import os
 
-def generate_process_flow_diagram(json_input: str) -> str: 
+def generate_process_flow_diagram(json_input: str, output_format: str) -> str:
     """
     Generates a Process Flow Diagram (Flowchart) from JSON input.
 
@@ -181,10 +181,9 @@ def generate_process_flow_diagram(json_input: str) -> str:
                 fontsize='10'
             )
         
-        # Save to temporary file
-        with NamedTemporaryFile(delete=False, suffix='.png') as tmp:
-            dot.render(tmp.name, format='png', cleanup=True)
-            return tmp.name + '.png'
+        with NamedTemporaryFile(delete=False, suffix=f'.{output_format}') as tmp:
+            dot.render(tmp.name, format=output_format, cleanup=True)
+            return f"{tmp.name}.{output_format}"
 
     except json.JSONDecodeError:
         return "Error: Invalid JSON format"