Update app.py

app.py

@@ -3,7 +3,6 @@ import qrcode
 from PIL import Image
 from io import BytesIO
 
-
 # Function to generate QR code
 def generate_qr_code(data):
     qr = qrcode.QRCode(
@@ -14,11 +13,10 @@ def generate_qr_code(data):
     )
     qr.add_data(data)
     qr.make(fit=True)
-
+    
     img = qr.make_image(fill_color="black", back_color="white")
     return img
 
-
 # Streamlit app
 st.title('QR Code Generator')
 st.write('Enter text or a URL to generate a QR code.')
@@ -29,14 +27,17 @@ user_input = st.text_input("Enter text or URL:")
 if user_input:
     # Generate QR code
     img = generate_qr_code(user_input)
-
-    # Display QR code
-    st.image(img, caption='Generated QR Code')
-
-    # Option to download the QR code
+    
+    # Convert the PIL image to a bytes-like object
     buffer = BytesIO()
     img.save(buffer, format="PNG")
     buffer.seek(0)
+    img_bytes = buffer.getvalue()
+    
+    # Display QR code
+    st.image(img_bytes, caption='Generated QR Code')
+
+    # Option to download the QR code
     st.download_button(
         label="Download QR Code",
         data=buffer,