Update app.py

app.py

@@ -1,7 +1,8 @@
 import qrcode
 from qrcode.constants import ERROR_CORRECT_L
 import streamlit as st
-from datetime import datetime
+import io
+from PIL import Image
 from urllib.parse import urlencode
 
 def generate_contact_qr(contact_info):
@@ -135,7 +136,9 @@ def main():
         }
         if st.button("Generate QR Code"):
             qr_code = generate_contact_qr(contact_info)
-            st.image(qr_code, use_column_width=True)
+            buf = io.BytesIO()
+            qr_code.save(buf, format="PNG")
+            st.image(buf.getvalue(), use_container_width=True)
     elif tab == "Email":
         st.subheader("Email QR Code")
         email_info = {
@@ -145,7 +148,9 @@ def main():
         }
         if st.button("Generate QR Code"):
             qr_code = generate_email_qr(email_info)
-            st.image(qr_code, use_column_width=True)
+            buf = io.BytesIO()
+            qr_code.save(buf, format="PNG")
+            st.image(buf.getvalue(), use_container_width=True)
     elif tab == "SMS":
         st.subheader("SMS QR Code")
         sms_info = {
@@ -154,20 +159,26 @@ def main():
         }
         if st.button("Generate QR Code"):
             qr_code = generate_sms_qr(sms_info)
-            st.image(qr_code, use_column_width=True)
+            buf = io.BytesIO()
+            qr_code.save(buf, format="PNG")
+            st.image(buf.getvalue(), use_container_width=True)
     elif tab == "Location":
         st.subheader("Location QR Code")
         latitude = st.number_input("Latitude", min_value=-90.0, max_value=90.0, step=0.000001)
         longitude = st.number_input("Longitude", min_value=-180.0, max_value=180.0, step=0.000001)
         if st.button("Generate QR Code"):
             qr_code = generate_location_qr(latitude, longitude)
-            st.image(qr_code, use_column_width=True)
+            buf = io.BytesIO()
+            qr_code.save(buf, format="PNG")
+            st.image(buf.getvalue(), use_container_width=True)
     elif tab == "Link":
         st.subheader("Link QR Code")
         url = st.text_input("URL")
         if st.button("Generate QR Code"):
             qr_code = generate_link_qr(url)
-            st.image(qr_code, use_column_width=True)
+            buf = io.BytesIO()
+            qr_code.save(buf, format="PNG")
+            st.image(buf.getvalue(), use_container_width=True)
 
 if __name__ == "__main__":
     main()