Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 import openai
 import urllib.parse
+import cairosvg
+import io
 
 # Function to call OpenAI GPT model for prompt processing
 def get_diagram_code(prompt, diagram_type, api_key):
@@ -24,6 +26,18 @@ def get_diagram_code(prompt, diagram_type, api_key):
         st.error(f"Error: {e}")
         return None
 
+# Function to convert SVG to PNG
+def convert_svg_to_png(svg_file):
+    try:
+        # Read the uploaded SVG file and convert it to PNG using cairosvg
+        png_output = io.BytesIO()
+        cairosvg.svg2png(file_obj=svg_file, write_to=png_output)
+        png_output.seek(0)
+        return png_output
+    except Exception as e:
+        st.error(f"Error during SVG to PNG conversion: {e}")
+        return None
+
 # Streamlit App UI
 def main():
     st.title("Generate Diagrams using GPT-4 and MermaidFlow")
@@ -60,12 +74,33 @@ def main():
                 st.markdown(f'<iframe src="{mermaidflow_url}" width="100%" height="600px"></iframe>', unsafe_allow_html=True)
 
                 # Provide a direct link to MermaidFlow editor for convenience
-                st.markdown(f"Copy and paste the Generated Mermaid Code in the Live Editor and download the generated diagram by clicking on Export as SVG format to get high resolution generated image.")
+                st.markdown(f"Copy and paste the Generated Mermaid Code in the Live Editor and download the generated diagram by clicking on Export as SVG format to get a high-resolution generated image.")
 
             else:
                 st.error("Failed to generate diagram code.")
         else:
             st.error("Please enter a prompt.")
 
+    # Add SVG file upload and conversion section
+    st.subheader("SVG to PNG Converter")
+    svg_file = st.file_uploader("Upload an SVG file to convert to PNG", type="svg")
+
+    if svg_file:
+        st.write("SVG file uploaded successfully!")
+        png_file = convert_svg_to_png(svg_file)
+
+        if png_file:
+            st.image(png_file, caption="Converted PNG Image", use_column_width=True)
+
+            # Provide a download link for the PNG file
+            st.download_button(
+                label="Download PNG",
+                data=png_file,
+                file_name="converted_image.png",
+                mime="image/png"
+            )
+        else:
+            st.error("Error in converting the SVG file to PNG.")
+
 if __name__ == "__main__":
     main()