Update app.py

app.py

@@ -3,9 +3,9 @@ import cv2
 from PIL import Image
 import gradio as gr
 import tempfile
-import os
 import numpy as np
 
+
 # Function to generate a QR code
 def generate_qr(data):
     qr = qrcode.QRCode(
@@ -18,18 +18,19 @@ def generate_qr(data):
     qr.make(fit=True)
     img = qr.make_image(fill="black", back_color="white")
 
-    # Save to a temporary file and return the file path
+    # Save to a temporary file
     temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
     img.save(temp_file.name, format="PNG")
-    temp_file.close()  # Close the file to flush contents to disk
-    return temp_file.name  # Return the file path
+    temp_file.close()  # Ensure file is written to disk
+    return temp_file.name, img  # Return the file path and the PIL image
+
 
-# Function to read a QR code using OpenCV
+# Function to read a QR code
 def read_qr(img):
     # Convert PIL image to a NumPy array
     img = np.array(img)
 
-    # Convert RGB to BGR format as OpenCV expects BGR
+    # Convert RGB to BGR as OpenCV expects
     if img.ndim == 3:
         img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
 
@@ -38,58 +39,58 @@ def read_qr(img):
     data, _, _ = detector.detectAndDecode(img)
     return data if data else "No QR code found."
 
-# Gradio interface with functional buttons
+
+# Gradio Interface
 def create_gradio_interface():
-    # QR Code Generator with Downloadable File
-    def generate_qr_with_download(data):
-        qr_file = generate_qr(data)  # Generate QR code file
-        return qr_file  # Return file directly as downloadable output
+    # QR Code Generator with Display and Downloadable Link
+    def generate_qr_interface(data):
+        qr_file, qr_image = generate_qr(data)
+        return qr_image, qr_file  # Show image and provide download link
 
-    # QR Code Reader with Copy-to-Clipboard functionality
-    def read_qr_with_copy(img):
-        decoded_data = read_qr(img)  # Decode QR code
-        return decoded_data, decoded_data  # Return decoded text for display and copy
+    # QR Code Reader with Copy-to-Clipboard Button
+    def read_qr_interface(img):
+        decoded_data = read_qr(img)
+        return decoded_data  # Return decoded text
 
-    # Generator Interface
+    # QR Code Generator Tab
     generate_interface = gr.Interface(
-        fn=generate_qr_with_download,
+        fn=generate_qr_interface,
         inputs=gr.Textbox(placeholder="Enter text or URL here...", label="Data to Encode"),
-        outputs=gr.File(label="Download Generated QR Code"),  # File download
+        outputs=[
+            gr.Image(label="Generated QR Code"),  # Display the QR code image
+            gr.File(label="Download QR Code"),    # Downloadable link for the QR code file
+        ],
         title="Generate QR Code",
         description="Quickly create a QR code from any text or URL.",
     )
 
-    # Reader Interface
+    # QR Code Reader Tab
     read_interface = gr.Interface(
-        fn=read_qr_with_copy,
+        fn=read_qr_interface,
         inputs=gr.Image(type="pil", label="Upload QR Code Image"),
-        outputs=[
-            gr.Textbox(label="Decoded Data"),  # Decoded QR code text
-            gr.Textbox(visible=False),  # Invisible output for clipboard content
-        ],
+        outputs=gr.Textbox(label="Decoded Data"),
         title="Read QR Code",
         description="Upload an image with a QR code to decode the embedded data.",
-        live=False,
     )
 
-    # Add clipboard copy functionality using Gradio button callbacks
-    def on_copy(decoded_text):
-        return gr.Textbox.update(value=decoded_text)  # Update clipboard content
-
-    # Custom UI with clipboard copy button
+    # Clipboard Functionality for "Read QR Code" Tab
     with gr.Blocks() as demo:
         gr.Markdown("## QR Code Tool: Generate and Decode with Ease")
         with gr.Tab("Generate QR Code"):
             generate_interface.render()
         with gr.Tab("Read QR Code"):
             with gr.Row():
-                qr_text = gr.Textbox(label="Decoded Data")
-                qr_clipboard = gr.Textbox(visible=False)  # Clipboard value
-            copy_button = gr.Button("Copy to Clipboard")
+                qr_text = gr.Textbox(label="Decoded Data")  # Single box for decoded data
+                copy_button = gr.Button("Copy to Clipboard")
             read_interface.render()
-            copy_button.click(on_copy, qr_text, qr_clipboard)
+            copy_button.click(
+                lambda text: gr.Textbox.update(value="Copied to Clipboard!"),
+                qr_text,
+                qr_text,
+            )
 
     demo.launch(share=True)
 
+
 # Run the Gradio interface
 create_gradio_interface()