Update app.py

app.py

@@ -3,6 +3,7 @@ from groq import Groq
 import os
 import schemdraw
 import schemdraw.elements as elm
+import matplotlib.pyplot as plt  # Ensure matplotlib is imported
 
 # Set your Groq API Key
 os.environ["GROQ_API_KEY"] = "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"
@@ -50,6 +51,8 @@ elif option == "Generate Circuit Diagram":
                     d += elm.Capacitor().label("C1").down()
                     d += elm.Line().left()
                     d += elm.Line().up()
+
+                    # To display the diagram correctly in Streamlit, use st.pyplot
                     st.pyplot(d.fig)
                     st.success("Circuit Diagram Generated!")
             except Exception as e: