Update app.py

app.py

@@ -497,6 +497,118 @@ def generate_qr_codes(data: Union[str, Dict, List], combined: bool = True) -> Li
         logger.error(f"QR code generation error: {e}")
         return []
 
+def create_qr_visualizer(qr_paths, metadata=None):
+    """Create an interactive visualization of sequenced QR codes"""
+    if not qr_paths:
+        return None
+    
+    # Extract metadata from QR codes if not provided
+    if metadata is None:
+        metadata = []
+        for path in qr_paths:
+            try:
+                img = Image.open(path)
+                qr = qrcode.QRCode()
+                data = qrcode.image.pil.PilImage.get_qr_data(img)
+                if data:
+                    metadata.append(json.loads(data))
+                else:
+                    # If can't extract, add placeholder
+                    metadata.append({"chunk_index": len(metadata), "total_chunks": len(qr_paths)})
+            except Exception as e:
+                logger.error(f"Error extracting QR metadata: {e}")
+                metadata.append({"chunk_index": len(metadata), "total_chunks": len(qr_paths)})
+    
+    # Compute optimal grid size
+    total_codes = len(qr_paths)
+    grid_size = math.ceil(math.sqrt(total_codes))
+    
+    # Create a composite image with placeholders for disabled QR codes
+    def create_composite(enabled_indices):
+        # Size calculations for the grid
+        qr_size = 200  # Size of each QR code in pixels
+        padding = 20   # Padding between QR codes
+        
+        # Create grid for visualization
+        grid_width = grid_size * (qr_size + padding) + padding
+        grid_height = grid_size * (qr_size + padding) + padding
+        
+        # Create a white background image
+        composite = Image.new('RGBA', (grid_width, grid_height), (255, 255, 255, 255))
+        draw = ImageDraw.Draw(composite)
+        
+        # Load and place QR codes on the grid
+        for i, path in enumerate(qr_paths):
+            # Calculate grid position
+            row = i // grid_size
+            col = i % grid_size
+            
+            # Calculate pixel position
+            x = col * (qr_size + padding) + padding
+            y = row * (qr_size + padding) + padding
+            
+            if i in enabled_indices:
+                try:
+                    # Load and resize QR code
+                    qr_img = Image.open(path)
+                    qr_img = qr_img.resize((qr_size, qr_size), Image.Resampling.LANCZOS)
+                    
+                    # Extract metadata for this QR
+                    meta = metadata[i] if i < len(metadata) else {}
+                    chunk_index = meta.get("chunk_index", i)
+                    total_chunks = meta.get("total_chunks", len(qr_paths))
+                    
+                    # Add visual indicator for sequence position
+                    sequence_indicator = Image.new('RGBA', (qr_size, 30), (26, 54, 93, 200))  # Dark blue
+                    draw_indicator = ImageDraw.Draw(sequence_indicator)
+                    draw_indicator.text((10, 5), f"#{chunk_index+1} of {total_chunks}", fill=(255, 255, 255))
+                    
+                    # Combine QR with indicator
+                    qr_with_indicator = Image.new('RGBA', (qr_size, qr_size + 30))
+                    qr_with_indicator.paste(qr_img, (0, 0))
+                    qr_with_indicator.paste(sequence_indicator, (0, qr_size), sequence_indicator)
+                    
+                    # Paste onto composite
+                    composite.paste(qr_with_indicator, (x, y))
+                    
+                    # Draw connection lines based on sequence
+                    if i >  0:
+                        prev_x = (col - 1) * (qr_size + padding) + padding if col > 0 else x
+                        prev_y = (row * (qr_size + padding)) + padding
+                        draw.line([(prev_x + qr_size // 2, prev_y + qr_size), (x + qr_size // 2, y)], fill=(0, 0, 0, 255), width=2)
+
+        return composite
+
+    # Create a toggleable interface for enabling/disabling QR codes
+    enabled_indices = list(range(total_codes))  # Start with all enabled
+    def toggle_qr(index):
+        if index in enabled_indices:
+            enabled_indices.remove(index)
+        else:
+            enabled_indices.append(index)
+        return create_composite(enabled_indices)
+
+    # Create the initial composite image
+    initial_composite = create_composite(enabled_indices)
+
+    # Display the composite image
+    plt.figure(figsize=(10, 10))
+    plt.imshow(initial_composite)
+    plt.axis('off')
+    plt.show()
+
+    return toggle_qr
+
+# Integrate the visualizer into the main application
+def visualize_qr_codes(qr_paths):
+    """Visualize the generated QR codes with enable/disable functionality"""
+    toggle_function = create_qr_visualizer(qr_paths)
+    return toggle_function
+
+# Add a button in the Gradio interface to trigger visualization
+visualize_btn = gr.Button("🔍 Visualize QR Codes")
+visualize_btn.click(visualize_qr_codes, inputs=output_gallery, outputs=None)
+
 def create_modern_interface():
     """Create a modern and visually appealing Gradio interface"""