Update app.py

app.py

@@ -308,8 +308,11 @@ class EnhancedFileProcessor:
                         if file_info.file_size > 0 and not file_info.filename.endswith('/'):
                             extracted_path = extract_to / file_info.filename
                             if extracted_path.suffix.lower() in self.supported_extensions:
-                                with open(extracted_path, 'rb') as f:
-                                    dataset.extend(self._process_single_file(f))
+                                try:
+                                    with open(extracted_path, 'rb') as f:
+                                        dataset.extend(self._process_single_file(f))
+                                except Exception as e:
+                                    logger.error(f"Error processing extracted file {extracted_path}: {e}")
             # Handle TAR archives
             elif archive_path.lower().endswith(('.tar', '.tar.gz', '.tgz')):
                 try:
@@ -317,10 +320,13 @@ class EnhancedFileProcessor:
                         for member in tar_ref.getmembers():
                             if member.isfile():
                                 extracted_path = extract_to / member.name
-                                tar_ref.extract(member, path=extract_to)
-                                if extracted_path.suffix.lower() in self.supported_extensions:
-                                    with open(extracted_path, 'rb') as f:
-                                        dataset.extend(self._process_single_file(f))
+                                try:
+                                    tar_ref.extract(member, path=extract_to)
+                                    if extracted_path.suffix.lower() in self.supported_extensions:
+                                        with open(extracted_path, 'rb') as f:
+                                            dataset.extend(self._process_single_file(f))
+                                except Exception as e:
+                                    logger.error(f"Error extracting or processing TAR member {member.name}: {e}")
                 except tarfile.TarError as e:
                     logger.error(f"Error processing TAR archive: {e}")
             # Handle GZIP archives (single file)
@@ -606,7 +612,7 @@ def create_qr_sequence_visualizer(output_gallery):
             qr_visualization.update(value=composite_image)
 
         # Event handlers
-        visualize_btn.click(process_qr_codes, inputs=qr_input, outputs=[visualization_status, qr_visualization, qr_preview])
+        visualize_btn.click(process_qr_codes, inputs=qr_input, outputs=[visualization_status, qr_paths, qr_preview])
         reset_btn.click(lambda: (None, None, None, "⚠️ Visualization reset."), outputs=[visualization_status, qr_visualization, qr_preview])
 
 # Integrate the visualizer into the main application
@@ -779,120 +785,120 @@ def create_modern_interface():
             return ""
 
         def process_inputs(urls, files, text, combine):
-            """Process all inputs and generate QR codes"""
-            try:
-                results = []
-                url_processor = EnhancedURLProcessor()
-                file_processor = EnhancedFileProcessor()
+"""Process all inputs and generate QR codes"""
+try:
+results = []
+url_processor = EnhancedURLProcessor()
+file_processor = EnhancedFileProcessor()
 
-                # Process JSON input
-                if text and text.strip():
-                    try:
-                        json_data = json.loads(text)
-                        if isinstance(json_data, list):
-                            results.extend(json_data)
-                        else:
-                            results.append(json_data)
-                    except json.JSONDecodeError as e:
-                        return None, [], f"❌ Invalid JSON format: {str(e)}"
-
-                # Process URLs
-                if urls and urls.strip():
-                    url_list = re.split(r'[,\n]', urls)
-                    url_list = [url.strip() for url in url_list if url.strip()]
-                    for url in url_list:
-                        validation = url_processor.validate_url(url)
-                        if validation['is_valid']:
-                            content = url_processor.fetch_content(url)
-                            if content:
-                                results.append({
-                                    'source': 'url',
-                                    'url': url,
-                                    'content': content,
-                                    'timestamp': datetime.now().isoformat()
-                                })
-
-                # Process files
-                if files:
-                    for file in files:
-                        file_results = file_processor.process_file(file)
-                        if file_results:
-                            results.extend(file_results)
-
-                # Generate QR codes
-                if results:
-                    qr_paths = generate_qr_codes(results, combine)
-                    if qr_paths:
-                        return (
-                            results,
-                            [str(path) for path in qr_paths],
-                            f"✅ Successfully processed {len(results)} items and generated {len(qr_paths)} QR codes!"
-                        )
-                    else:
-                        return None, [], "❌ Failed to generate QR codes"
-                else:
-                    return None, [], "⚠️ No valid content to process"
-            except Exception as e:
-                logger.error(f"Processing error: {e}")
-                return None, [], f"❌ Error: {str(e)}"
-
-        # Set up event handlers
-        example_btn.click(load_example, outputs=[text_input])
-        clear_btn.click(clear_input, outputs=[text_input])
-        process_btn.click(
-            process_inputs,
-            inputs=[url_input, file_input, text_input, combine_data],
-            outputs=[output_json, output_gallery, output_text]
-        )
 
-        # 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)
 
-        # Add helpful documentation
-        gr.Markdown("""
-        ### 🚀 Features
-                - **Complete URL Scraping**: Extracts every character from web pages
-                - **Advanced File Processing**: Full content extraction from various text-based files and common archives. Supports flexible JSON handling.
-                - **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
-                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.
-                4. **QR Codes**: Choose whether to combine data into sequential codes
-                5. **Processing**: Monitor the status for real-time feedback
-        ### 🎨 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
-                """)
-    return interface
+# Process JSON input
+            if text and text.strip():
+                try:
+                    json_data = json.loads(text)
+                    if isinstance(json_data, list):
+                        results.extend(json_data)
+                    else:
+                        results.append(json_data)
+                except json.JSONDecodeError as e:
+                    return None, [], f"❌ Invalid JSON format: {str(e)}"
+
+            # Process URLs
+            if urls and urls.strip():
+                url_list = re.split(r'[,\n]', urls)
+                url_list = [url.strip() for url in url_list if url.strip()]
+                for url in url_list:
+                    validation = url_processor.validate_url(url)
+                    if validation['is_valid']:
+                        content = url_processor.fetch_content(url)
+                        if content:
+                            results.append({
+                                'source': 'url',
+                                'url': url,
+                                'content': content,
+                                'timestamp': datetime.now().isoformat()
+                            })
 
+            # Process files
+            if files:
+                for file in files:
+                    file_results = file_processor.process_file(file)
+                    if file_results:
+                        results.extend(file_results)
+
+            # Generate QR codes
+            if results:
+                qr_paths = generate_qr_codes(results, combine)
+                if qr_paths:
+                    return (
+                        results,
+                        [str(path) for path in qr_paths],
+                        f"✅ Successfully processed {len(results)} items and generated {len(qr_paths)} QR codes!"
+                    )
+                else:
+                    return None, [], "❌ Failed to generate QR codes"
+            else:
+                return None, [], "⚠️ No valid content to process"
+        except Exception as e:
+            logger.error(f"Processing error: {e}")
+            return None, [], f"❌ Error: {str(e)}"
+
+    # Set up event handlers
+    example_btn.click(load_example, outputs=[text_input])
+    clear_btn.click(clear_input, outputs=[text_input])
+    process_btn.click(
+        process_inputs,
+        inputs=[url_input, file_input, text_input, combine_data],
+        outputs=[output_json, output_gallery, output_text]
+    )
+
+    # 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)
+
+    # Add helpful documentation
+    gr.Markdown("""
+    ### 🚀 Features
+            - **Complete URL Scraping**: Extracts every character from web pages
+            - **Advanced File Processing**: Full content extraction from various text-based files and common archives. Supports flexible JSON handling.
+            - **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
+            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.
+            4. **QR Codes**: Choose whether to combine data into sequential codes
+            5. **Processing**: Monitor the status for real-time feedback
+    ### 🎨 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
+            """)
+return interface
 def main():
-    """Initialize and launch the application"""
-    try:
-        # Configure system settings
-        mimetypes.init()
-
-        # 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,
-            debug=False,
-            show_error=True,
-            show_api=False
-        )
-    except Exception as e:
-        logger.error(f"Application startup error: {e}")
-        raise
-
-if __name__ == "__main__":
-    main()
+"""Initialize and launch the application"""
+try:
+# Configure system settings
+mimetypes.init()
+
+# 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,
+        debug=False,
+        show_error=True,
+        show_api=False
+    )
+except Exception as e:
+    logger.error(f"Application startup error: {e}")
+    raise
+if name == "main":
+main()