Update app.py

app.py

@@ -48,7 +48,6 @@ for directory in [OUTPUTS_DIR, QR_CODES_DIR, TEMP_DIR]:
 
 class EnhancedURLProcessor:
     """Advanced URL processing with complete content extraction"""
-
     def __init__(self):
         self.session = requests.Session()
         self.timeout = 15  # Extended timeout for larger content
@@ -115,6 +114,7 @@ class EnhancedURLProcessor:
                 encoding = chardet.detect(response.content)['encoding'] or 'utf-8'
             else:
                 encoding = response.encoding
+
             # Decode content with fallback
             try:
                 raw_content = response.content.decode(encoding, errors='replace')
@@ -178,7 +178,6 @@ class EnhancedURLProcessor:
 
 class EnhancedFileProcessor:
     """Advanced file processing with complete content extraction"""
-
     def __init__(self, max_file_size: int = 5 * 1024 * 1024 * 1024):  # 5GB default
         self.max_file_size = max_file_size
         self.supported_extensions = {
@@ -227,10 +226,8 @@ class EnhancedFileProcessor:
         try:
             file_stat = os.stat(file.name)
             file_size = file_stat.st_size
-
             # Initialize content storage
             content_parts = []
-
             # Process file in chunks for large files
             chunk_size = 10 * 1024 * 1024  # 10MB chunks
             with open(file.name, 'rb') as f:
@@ -238,7 +235,6 @@ class EnhancedFileProcessor:
                     chunk = f.read(chunk_size)
                     if not chunk:
                         break
-
                     # Detect encoding for each chunk
                     encoding = chardet.detect(chunk)['encoding'] or 'utf-8'
                     try:
@@ -247,10 +243,8 @@ class EnhancedFileProcessor:
                     except (UnicodeDecodeError, LookupError):
                         decoded_chunk = chunk.decode('utf-8', errors='replace')
                         content_parts.append(decoded_chunk)
-
             # Combine all chunks
             complete_content = ''.join(content_parts)
-
             # Check if the content is valid JSON regardless of file extension
             try:
                 if mimetypes.guess_type(file.name)[0] == 'application/json' or file.name.lower().endswith('.json'):
@@ -504,15 +498,14 @@ def generate_qr_codes(data: Union[str, Dict, List], combined: bool = True) -> Li
 
 def create_qr_sequence_visualizer(output_gallery):
     """Add QR sequence visualization capabilities to the application"""
-    
     # Create a new tab for the QR code sequence visualization
     with gr.Tab("🔄 QR Sequence Visualizer"):
         gr.Markdown("""
         ## QR Code Sequence Visualizer
-        
+
         Arrange and visualize your QR code sequences. Enable or disable individual QR codes to see how they connect.
         """)
-        
+
         # Inputs for the visualizer
         with gr.Row():
             qr_input = gr.File(
@@ -520,43 +513,50 @@ def create_qr_sequence_visualizer(output_gallery):
                 file_types=["image/png", "image/jpeg"],
                 file_count="multiple"
             )
-            
+
             with gr.Column():
                 visualize_btn = gr.Button("🔄 Generate Visualization", variant="primary")
                 reset_btn = gr.Button("🗑️ Reset", variant="secondary")
-        
-        # Container for QR code toggles
+                # Container for QR code toggles
         qr_toggles_container = gr.HTML(label="QR Code Controls")
-        
+
         # Output visualization
         with gr.Row():
             qr_visualization = gr.Image(label="QR Code Sequence Map", height=600)
             qr_preview = gr.Gallery(label="Selected QR Codes", columns=2, height=600)
-        
+
         # Status output
         visualization_status = gr.Textbox(label="Visualization Status", interactive=False)
-        
+
         # Function to process uploaded QR codes
         def process_qr_codes(files):
             if not files:
                 return "Please upload QR code images.", None, None, "⚠️ No QR codes uploaded"
-            
+
             try:
                 # Load QR codes and extract metadata
                 qr_data = []
                 qr_paths = []
-                
+
                 for file in files:
                     try:
                         img = Image.open(file.name)
-                        
+
                         # Try to decode QR code
                         try:
                             detector = qrcode.QRCodeDetector()
                             data, bbox, _ = detector.detectAndDecode(np.array(img))
                             if data:
-                                qr_json = json.loads(data)
-                                qr_data.append(qr_json)
+                                try:
+                                    qr_json = json.loads(data)
+                                    qr_data.append(qr_json)
+                                    qr_paths.append(file.name)
+                                except json.JSONDecodeError:
+                                    logger.warning(f"Could not decode JSON from QR: {data}")
+                                    qr_data.append({"data": data}) # Store raw data if JSON fails
+                                    qr_paths.append(file.name)
+                            else:
+                                qr_data.append({"data": "Empty QR"})
                                 qr_paths.append(file.name)
                         except Exception as e:
                             logger.warning(f"Could not decode QR: {e}")
@@ -569,10 +569,10 @@ def create_qr_sequence_visualizer(output_gallery):
                             qr_paths.append(file.name)
                     except Exception as e:
                         logger.error(f"Error processing QR image {file.name}: {e}")
-                
+
                 if not qr_data:
                     return "No valid QR codes found.", None, None, "❌ Failed to process QR codes"
-                
+
                 # Sort by chunk_index if available
                 try:
                     sorted_data = sorted(zip(qr_data, qr_paths), key=lambda x: x[0].get("chunk_index", 0))
@@ -580,36 +580,32 @@ def create_qr_sequence_visualizer(output_gallery):
                     qr_paths = [d[1] for d in sorted_data]
                 except Exception as e:
                     logger.error(f"Error sorting QR data: {e}")
-                
+
                 # Generate toggle controls HTML
                 toggle_html = '<div style="max-height: 500px; overflow-y: auto; padding: 10px;">'
                 toggle_html += '<h3>Enable/Disable QR Codes:</h3>'
                 for i, path in enumerate(qr_paths):
-                    toggle_html += f'<div><input type="checkbox" id="qr_toggle_{i}" checked> <label for="qr_toggle_{i}">{path}</label></div>'
+                    toggle_html += f'<div><input type="checkbox" id="qr_toggle_{i}" checked> <label for="qr_toggle_{i}">{os.path.basename(path)}</label></div>'
                 toggle_html += '</div>'
-                
+
                 # Update the toggles container
                 qr_toggles_container.update(value=toggle_html)
-                
-                # Create initial visualization
-                initial_visualization = create_composite(qr_paths)
+
+                # Create initial visualization (replace with actual visualization logic)
+                initial_visualization = "Visualization will appear here." # Replace with your composite image generation
                 qr_visualization.update(value=initial_visualization)
-                
+
                 return "QR codes processed successfully.", qr_paths, qr_data, "✅ Visualization ready!"
             except Exception as e:
                 logger.error(f"Error processing QR codes: {e}")
                 return "An error occurred while processing QR codes.", None, None, "❌ Error"
 
-        # Function to generate visualization
+        # Function to generate visualization (replace with actual logic)
         def generate_visualization(qr_paths):
             enabled_indices = [i for i in range(len(qr_paths))]  # Start with all enabled
-            composite_image = create_composite(enabled_indices)
+            composite_image = "Updated visualization will appear here." # Replace with your composite image generation based on enabled_indices
             qr_visualization.update(value=composite_image)
 
-
-
-        # Later in your code, use it in the button click event
-        visualize_btn.click(visualize_qr_codes, inputs=output_gallery, outputs=None)
         # Event handlers
         visualize_btn.click(process_qr_codes, inputs=qr_input, outputs=[visualization_status, qr_visualization, qr_preview])
         reset_btn.click(lambda: (None, None, None, "⚠️ Visualization reset."), outputs=[visualization_status, qr_visualization, qr_preview])
@@ -617,12 +613,11 @@ def create_qr_sequence_visualizer(output_gallery):
 # 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)
+    # This function currently receives the output gallery content (list of file paths)
+    # You might need to adapt this based on how you want to visualize.
+    # For now, let's just log the paths.
+    logger.info(f"Visualizing QR codes: {qr_paths}")
+    return "Visualization placeholder" # Replace with actual visualization logic
 
 def create_modern_interface():
     """Create a modern and visually appealing Gradio interface"""
@@ -703,7 +698,6 @@ def create_modern_interface():
     with gr.Blocks(css=css, title="Advanced Data Processor & QR Generator") as interface:
         gr.Markdown("""
         # 🌐 Advanced Data Processing & QR Code Generator
-
         Transform your data into beautifully designed, sequenced QR codes with our cutting-edge processor.
         """)
         with gr.Tab("📝 URL Processing"):
@@ -852,7 +846,7 @@ def create_modern_interface():
             outputs=[output_json, output_gallery, output_text]
         )
 
-        # Add the visualization button
+        # Add the visualization button and its click event within the interface scope
         visualize_btn = gr.Button("🔍 Visualize QR Codes")
         visualize_btn.click(visualize_qr_codes, inputs=output_gallery, outputs=None)
 
@@ -864,15 +858,13 @@ def create_modern_interface():
                 - **Smart JSON Handling**: Processes any size JSON with automatic chunking, either via direct input or file upload.
                 - **Sequential QR Codes**: Maintains data integrity across multiple codes
                 - **Modern Design**: Clean, responsive interface with visual feedback
-
-                ### 💡 Tips
+        ### 💡 Tips
                 1. **URLs**: Enter multiple URLs separated by commas or newlines
                 2. **Files**: Upload any type of file. The processor will attempt to handle supported text-based files, archives (.zip, .tar, .gz), and JSON files.
-                3. **JSON**: Use the example button to see the expected format or upload a .json file. The system will also try to detect JSON content in other file types. 
+                3. **JSON**: Use the example button to see the expected format or upload a .json file. The system will also try to detect JSON content in other file types.
                 4. **QR Codes**: Choose whether to combine data into sequential codes
                 5. **Processing**: Monitor the status for real-time feedback
-
-                ### 🎨 Output
+        ### 🎨 Output
                 - Generated QR codes are saved in the `output/qr_codes` directory
                 - Each QR code contains metadata for proper sequencing
                 - Hover over QR codes in the gallery to see details
@@ -888,6 +880,10 @@ def main():
         # Create and launch interface
         interface = create_modern_interface()
 
+        # Add the QR sequence visualizer tab
+        with interface:
+            create_qr_sequence_visualizer(None) # output_gallery might not be relevant here
+
         # Launch with configuration
         interface.launch(
             share=False,