Update app.py

app.py

@@ -25,7 +25,7 @@ def get_diagram_code(prompt, diagram_type, api_key):
 
 # Streamlit App UI
 def main():
-    st.title("Generate Diagrams using GPT-4 and Mermaid.js")
+    st.title("Generate Diagrams using GPT-4 and Mermaid AI")
 
     # User input for OpenAI API key
     api_key = st.text_input("Enter your OpenAI API Key:", type="password")
@@ -49,13 +49,12 @@ def main():
                 # Render Mermaid code to Streamlit
                 st.code(diagram_code, language='mermaid')
 
-                # Display Mermaid diagram using iframe (mermaid.live)
+                # Prepare Mermaid code for the Mermaid AI platform
                 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)
+                mermaid_ai_url = f"https://www.mermaidchart.com/app/projects/1179e788-52af-40a6-a788-6f15474a9c04/diagrams/f787a6a7-4922-4063-ad40-29bdec71a7ab/version/v0.1/ai?mermaidCode={encoded_code}"
 
-                # 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})")
+                # Provide a link to the Mermaid AI platform
+                st.markdown(f"Click the link to generate and save the diagram: [Mermaid AI - Generate Diagram]({mermaid_ai_url})")
 
             else:
                 st.error("Failed to generate diagram code.")