Update app.py

app.py

@@ -32,7 +32,7 @@ def generate_qrcode(link):
     qr.make(fit=True)
     qr_image = qr.make_image(fill_color="black", back_color="white")
 
-    small_qr_image = qr_image.resize((60, 60), Image.Resampling.LANCZOS)
+    small_qr_image = qr_image.resize((512, 512), Image.Resampling.LANCZOS)
 
     buffer = BytesIO()
     small_qr_image.save(buffer, format="PNG")
@@ -162,12 +162,12 @@ def main():
         ["Accueil", "Transformations d'image", "Cropping", "Rotation", "Floutage", "Contours", "Génération de Code-barres et QR Code", "Détection Faciale"],
         format_func=lambda x: {
             "Accueil": "🏠 Accueil",
-            "Transformations d'image": "🖼️ Transformations",
+            "Transformations d'image": "🖼️ Niveaux de couleurs",
             "Cropping": "✂️ Cropping",
             "Rotation": "🔄 Rotation",
             "Floutage": "🌫️ Floutage",
             "Contours": "🔍 Contours",
-            "Génération de Code-barres et QR Code": "📇 Codes",
+            "Génération de Code-barres et QR Code": "📇 Codes numériques",
             "Détection Faciale": "🙂 Détection Faciale"
         }.get(x, x)
     )
@@ -180,8 +180,7 @@ def main():
             """
             <div class="stMarkdown">
                 <h2>ADS VISOR est une application innovante pour analyser, transformer et explorer vos images. 🖼️✨</h2>
-                <p>Elle a été concu par un groupe de trois étudiants dans le contexte du contrôle continu de Computer Vision.</p>
-                <p>Que vous soyez un professionnel ou un passionné, découvrez un large éventail de fonctionnalités interactives !</p>
+                <p>Que vous soyez un professionnel ou un passionné, découvrez un large éventail de fonctionnalités interactives !</p>
                 <ul>
                     <li><b>Transformations d'image :</b> Couleurs, niveaux de gris, etc.</li>
                     <li><b>Découpage & Rotation :</b> Ajustez vos images à la perfection.</li>
@@ -213,30 +212,86 @@ def main():
         ])
 
         with tab1:
-            st.image(cv2.cvtColor(image_np, cv2.COLOR_BGR2GRAY), caption="Image en Niveaux de Gris", use_container_width=True)
+            gray_image = cv2.cvtColor(image_np, cv2.COLOR_BGR2GRAY)
+            st.image(gray_image, caption="Image en Niveaux de Gris", use_container_width=True)
+            st.download_button(
+                label="Télécharger l'image en gris",
+                data=image_to_bytes(gray_image),
+                file_name="image_gris.png",
+                mime="image/png"
+            )
         with tab2:
             red = image_np.copy()
             red[:, :, 1:] = 0
             st.image(red, caption="Image Rouge", use_container_width=True)
+            st.download_button(
+                label="Télécharger l'image rouge",
+                data=image_to_bytes(red),
+                file_name="image_rouge.png",
+                mime="image/png"
+            )
         with tab3:
             green = image_np.copy()
             green[:, :, [0, 2]] = 0
             st.image(green, caption="Image Verte", use_container_width=True)
+            st.download_button(
+                label="Télécharger l'image verte",
+                data=image_to_bytes(green),
+                file_name="image_verte.png",
+                mime="image/png"
+            )
         with tab4:
             yellow = image_np.copy()
             yellow[:, :, 0] = 0
             st.image(yellow, caption="Image Jaune", use_container_width=True)
+            st.download_button(
+                label="Télécharger l'image jaune",
+                data=image_to_bytes(yellow),
+                file_name="image_jaune.png",
+                mime="image/png"
+            )
 
     elif menu_option == "Cropping":
         st.subheader("Cropping")
 
-        x1 = st.number_input("x1", 0, image_np.shape[1] - 1, step=1)
-        y1 = st.number_input("y1", 0, image_np.shape[0] - 1, step=1)
-        x2 = st.number_input("x2", 1, image_np.shape[1], step=1)
-        y2 = st.number_input("y2", 1, image_np.shape[0], step=1)
+        # Explications pour les paramètres
+        st.markdown(
+            """
+            ### Légende : Fonctionnement du recadrage
+            - **`x1` (Coordonnée x du coin supérieur gauche)** : Position horizontale du début du rectangle (0 ≤ `x1` < largeur-1).
+            - **`y1` (Coordonnée y du coin supérieur gauche)** : Position verticale du début du rectangle (0 ≤ `y1` < hauteur-1).
+            - **`x2` (Coordonnée x du coin inférieur droit)** : Position horizontale de la fin du rectangle (1 ≤ `x2` ≤ largeur).
+            - **`y2` (Coordonnée y du coin inférieur droit)** : Position verticale de la fin du rectangle (1 ≤ `y2` ≤ hauteur).
+            """
+        )
 
-        cropped = image_np[int(y1):int(y2), int(x1):int(x2)]
-        st.image(cropped, caption="Image Croppée", use_container_width=True)
+        # Entrée des coordonnées
+        try:
+            x1 = st.number_input("x1 (Coordonnée x du coin supérieur gauche)", 0, image_np.shape[1] - 1, step=1)
+            y1 = st.number_input("y1 (Coordonnée y du coin supérieur gauche)", 0, image_np.shape[0] - 1, step=1)
+            x2 = st.number_input("x2 (Coordonnée x du coin inférieur droit)", 1, image_np.shape[1], step=1)
+            y2 = st.number_input("y2 (Coordonnée y du coin inférieur droit)", 1, image_np.shape[0], step=1)
+
+            # Validation des coordonnées
+            if x1 >= x2:
+                st.error("`x2` doit être strictement supérieur à `x1`. Veuillez corriger vos entrées.")
+            elif y1 >= y2:
+                st.error("`y2` doit être strictement supérieur à `y1`. Veuillez corriger vos entrées.")
+            else:
+                # Recadrage de l'image
+                cropped = image_np[int(y1):int(y2), int(x1):int(x2)]
+                st.image(cropped, caption="Image Croppée", use_container_width=True)
+                
+                # Bouton de téléchargement
+                st.download_button(
+                    label="Télécharger l'image croppée",
+                    data=image_to_bytes(cropped),
+                    file_name="image_cropped.png",
+                    mime="image/png"
+                )
+
+        except Exception as e:
+            st.error(f"Une erreur est survenue : {e}")
 
     elif menu_option == "Rotation":
         st.subheader("Rotation")
@@ -247,6 +302,14 @@ def main():
         rotation_matrix = cv2.getRotationMatrix2D((cols / 2, rows / 2), angle, 1)
         rotated = cv2.warpAffine(image_np, rotation_matrix, (cols, rows))
         st.image(rotated, caption=f"Image Rotée de {angle} degrés", use_container_width=True)
+        
+        # Bouton de téléchargement
+        st.download_button(
+            label="Télécharger l'image rotée",
+            data=image_to_bytes(rotated),
+            file_name=f"image_rotated_{angle}.png",
+            mime="image/png"
+        )
 
     elif menu_option == "Floutage":
         st.subheader("Floutage")
@@ -254,6 +317,14 @@ def main():
         blur_level = st.slider("Niveau de flou (k)", min_value=1, max_value=51, step=2, value=15)
         blurred = cv2.GaussianBlur(image_np, (blur_level, blur_level), 0)
         st.image(blurred, caption=f"Image Floutée (k={blur_level})", use_container_width=True)
+        
+        # Bouton de téléchargement
+        st.download_button(
+            label="Télécharger l'image floutée",
+            data=image_to_bytes(blurred),
+            file_name=f"image_blurred_k{blur_level}.png",
+            mime="image/png"
+        )
 
     elif menu_option == "Contours":
         st.subheader("Contours")
@@ -261,18 +332,61 @@ def main():
         gray = cv2.cvtColor(image_np, cv2.COLOR_BGR2GRAY)
         edges = cv2.Canny(gray, 100, 200)
         st.image(edges, caption="Contours de l'Image", use_container_width=True)
-
+        # Bouton de téléchargement
+        st.download_button(
+            label="Télécharger l'image avec contours",
+            data=image_to_bytes(edges),
+            file_name=f"image_edges.png",
+            mime="image/png"
+        )
+        
+        
     elif menu_option == "Génération de Code-barres et QR Code":
         st.subheader("Génération de Code-barres et QR Code")
-        link = st.text_input("Entre un lien ou un texte pour générer le code-barres et le QR code")
-        if link:
-            barcode_image = generate_barcode(link)
-            st.image(barcode_image, caption="Code-barres généré", use_container_width=True)
-
-            qrcode_image = generate_qrcode(link)
-            st.image(qrcode_image, caption="QR Code généré", use_container_width=True)
-        else:
-            st.error("Veuillez entrer un lien valide pour générer les codes.")
+
+        # Entrée utilisateur pour le lien ou le texte
+        link = st.text_input("Entre un lien ou un texte pour générer les codes")
+        
+        # Ajout du bouton pour confirmer la génération
+        if st.button("Générer"):
+            if link:
+                try:
+                    # Génération du code-barres
+                    barcode_image = generate_barcode(link)
+                    st.image(barcode_image, caption="Code-barres généré", use_container_width=True)
+                    
+                    barcode_buffer = BytesIO()
+                    barcode_image.save(barcode_buffer, format="PNG")
+                    barcode_buffer.seek(0)
+                    st.download_button(
+                        label="Télécharger le Code-barres",
+                        data=barcode_buffer,
+                        file_name="barcode.png",
+                        mime="image/png"
+                    )
+
+                    # Génération du QR Code
+                    qrcode_image = generate_qrcode(link)
+                    st.image(qrcode_image, caption="QR Code généré", use_container_width=True)
+
+                    # Bouton de téléchargement pour le code-barres
+                    
+
+                    # Bouton de téléchargement pour le QR code
+                    qrcode_buffer = BytesIO()
+                    qrcode_image.save(qrcode_buffer, format="PNG")
+                    qrcode_buffer.seek(0)
+                    st.download_button(
+                        label="Télécharger le QR Code",
+                        data=qrcode_buffer,
+                        file_name="qrcode.png",
+                        mime="image/png"
+                    )
+
+                except Exception as e:
+                    st.error(f"Une erreur est survenue : {e}")
+            else:
+                st.error("Veuillez entrer un lien ou un texte valide pour générer les codes.")
 
     elif menu_option == "Détection Faciale":
         st.subheader("Détection Faciale")