Update app.py

app.py

@@ -40,7 +40,11 @@ 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 record and future editing
+    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)