Update app.py

app.py

@@ -50,9 +50,13 @@ def main():
                 st.code(diagram_code, language='mermaid')
 
                 # Display Mermaid diagram using iframe (mermaid.live)
-                diagram_url = f"https://mermaid-js.github.io/mermaid-live-editor/#/edit/{diagram_code}"
+                encoded_code = diagram_code.replace("\n", "%0A").replace(" ", "%20")  # Encode the Mermaid code
+                diagram_url = f"https://mermaid-js.github.io/mermaid-live-editor/#/edit/{encoded_code}"
                 st.markdown(f'<iframe src="{diagram_url}" width="800" height="600"></iframe>', unsafe_allow_html=True)
 
+                # Provide a direct link to download the diagram as PNG (this depends on mermaid's features)
+                st.markdown(f"View and save the diagram using this [Mermaid Live Editor link]({diagram_url})")
+
             else:
                 st.error("Failed to generate diagram code.")
         else: