Update app.py

app.py

@@ -40,18 +40,20 @@ Generate the correct code, ensure it is free from construct error and this kind
     svg_code = completion.choices[0].message.content.strip()
     full_svg_code = f"<svg xmlns='http://www.w3.org/2000/svg' width='100%' height='100%' viewBox='0 0 1000 400'>{svg_code}</svg>"
     
-    # Save the SVG code to a file for download
-    with open("diagram.svg", "w") as f:
-        f.write(full_svg_code)
+    # Convert the SVG code to PNG using CairoSVG
+    png_filename = "diagram.png"
+    cairosvg.svg2png(bytestring=full_svg_code.encode('utf-8'), write_to=png_filename)
     
-    # Return: HTML for display, file path, and store the full code in state for editing in Tab 2.
-    return full_svg_code, "diagram.svg", full_svg_code
+    # Return the PNG for display, the PNG file for download, and the SVG code for editing later.
+    return png_filename, png_filename, full_svg_code
 
 def update_svg(modified_svg_code):
     # Save the modified SVG code to the file
     with open("diagram.svg", "w") as f:
         f.write(modified_svg_code)
-    return modified_svg_code, "diagram.svg"
+    png_filename = "diagram.png"
+    cairosvg.svg2png(bytestring=modified_svg_code.encode('utf-8'), write_to=png_filename)
+    return png_filename, png_filename
 
 with gr.Blocks(css="""