Update app.py

app.py

@@ -342,11 +342,14 @@ class EnhancedFileProcessor:
             logger.error(f"Archive processing error: {e}")
         return dataset
 
-    def chunk_data(self, data: Union[Dict, List], max_size: int = 2953) -> List[Dict]:
+    def chunk_data(self, data: Union[Dict, List, str], max_size: int = 2953) -> List[Dict]:
         """Enhanced data chunking with sequence metadata"""
         try:
-            # Convert data to JSON string
-            json_str = json.dumps(data, ensure_ascii=False)
+            if not isinstance(data, str):
+                # Convert data to JSON string
+                json_str = json.dumps(data, ensure_ascii=False)
+            else:
+                json_str = data
             total_length = len(json_str)
 
             # Calculate overhead for metadata
@@ -407,7 +410,7 @@ def generate_stylish_qr(data: Union[str, Dict],
     try:
         qr = qrcode.QRCode(
             version=None,
-            error_correction=qrcode.constants.ERROR_CORRECT_L,
+            error_correction=qrcode.constants.ERROR_CORRECT_H,
             box_size=size,
             border=border
         )
@@ -457,7 +460,7 @@ def generate_qr_codes(data: Union[str, Dict, List], combined: bool = True) -> Li
             for i, chunk in enumerate(chunks):
                 filename = f'combined_qr_{int(time.time())}_{i+1}_of_{len(chunks)}.png'
                 qr_path = generate_stylish_qr(
-                    data=chunk,
+                    data=chunk['data'], # Use the 'data' part of the chunk
                     filename=filename,
                     fill_color="#1a365d",  # Deep blue
                     back_color="#ffffff"
@@ -472,7 +475,7 @@ def generate_qr_codes(data: Union[str, Dict, List], combined: bool = True) -> Li
                     for chunk_idx, chunk in enumerate(chunks):
                         filename = f'item_{idx+1}_chunk_{chunk_idx+1}_of_{len(chunks)}_{int(time.time())}.png'
                         qr_path = generate_stylish_qr(
-                            data=chunk,
+                            data=chunk['data'], # Use the 'data' part of the chunk
                             filename=filename,
                             fill_color="#1a365d",  # Deep blue
                             back_color="#ffffff"
@@ -484,7 +487,7 @@ def generate_qr_codes(data: Union[str, Dict, List], combined: bool = True) -> Li
                 for i, chunk in enumerate(chunks):
                     filename = f'single_qr_{i+1}_of_{len(chunks)}_{int(time.time())}.png'
                     qr_path = generate_stylish_qr(
-                        data=chunk,
+                        data=chunk['data'], # Use the 'data' part of the chunk
                         filename=filename,
                         fill_color="#1a365d",  # Deep blue
                         back_color="#ffffff"
@@ -804,14 +807,19 @@ def create_modern_interface():
                     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()
-                                })
+                            content_data = url_processor.fetch_content(url)
+                            if content_data and 'content' in content_data:
+                                # Chunk the content of each URL
+                                chunks = file_processor.chunk_data(content_data['content'])
+                                for i, chunk in enumerate(chunks):
+                                    results.append({
+                                        'source': 'url',
+                                        'url': url,
+                                        'chunk_index': i + 1,
+                                        'total_chunks': len(chunks),
+                                        'content': chunk['data'], # Store the chunked data
+                                        'timestamp': datetime.now().isoformat()
+                                    })
 
                 # Process files
                 if files: