Update app2.py

app2.py

@@ -4,33 +4,20 @@ import re
 import logging
 import mimetypes
 import time
-import zbar
 from PIL import Image
 from pyzbar.pyzbar import decode
 import io
-from selenium.webdriver.common.by import By
-import concurrent.futures
-import string
 import zipfile
 import tempfile
 from datetime import datetime
 from typing import List, Dict, Optional, Union
 from pathlib import Path
-from urllib.parse import urlparse
 import requests
 import validators
 import gradio as gr
 from bs4 import BeautifulSoup
 from fake_useragent import UserAgent
-from ratelimit import limits, sleep_and_retry
 from cleantext import clean
-import qrcode
-from PIL import Image
-import nest_asyncio
-nest_asyncio.apply()
-import aiohttp
-
-
 
 # Setup logging
 logging.basicConfig(
@@ -271,50 +258,68 @@ class FileProcessor:
             logger.error(f"File processing error: {e}")
             return []
 
-def clean_json(data: Union[str, Dict]) -> Optional[Dict]:
-    """Clean and validate JSON data"""
-    try:
-        if isinstance(data, str):
-            data = data.strip()
-            data = json.loads(data)
-        
-        cleaned = json.loads(json.dumps(data))
-        return cleaned
-    except json.JSONDecodeError as e:
-        logger.error(f"JSON cleaning error: {e}")
-        return None
-    except Exception as e:
-        logger.error(f"Unexpected error while cleaning JSON: {e}")
-        return None
-
-def generate_qr_code(data: Union[str, Dict], combined: bool = True) -> List[str]:
-    """Generate QR code(s) from data"""
-    try:
-        output_dir = Path('output/qr_codes')
-        output_dir.mkdir(parents=True, exist_ok=True)
-        
-        if combined:
-            cleaned_data = clean_json(data)
-            if cleaned_data:
-                qr = qrcode.QRCode(
-                    version=None,
-                    error_correction=qrcode.constants.ERROR_CORRECT_L,
-                    box_size=10,
-                    border=4,
-                )
-                json_str = json.dumps(cleaned_data, ensure_ascii=False)
-                qr.add_data(json_str)
-                qr.make(fit=True)
-                
-                img = qr.make_image(fill_color="black", back_color="white")
-                output_path = output_dir / f'combined_qr_{int(time.time())}.png'
-                img.save(str(output_path))
-                return [str(output_path)]
-        else:
-            if isinstance(data, list):
-                paths = []
-                for idx, item in enumerate(data):
-                    cleaned_item = clean_json(item)
+    def clean_json(data: Union[str, Dict]) -> Optional[Dict]:
+        """Clean and validate JSON data"""
+        try:
+            if isinstance(data, str):
+                data = data.strip()
+                data = json.loads(data)
+            
+            cleaned = json.loads(json.dumps(data))
+            return cleaned
+        except json.JSONDecodeError as e:
+            logger.error(f"JSON cleaning error: {e}")
+            return None
+        except Exception as e:
+            logger.error(f"Unexpected error while cleaning JSON: {e}")
+            return None
+    
+    def generate_qr_code(data: Union[str, Dict], combined: bool = True) -> List[str]:
+        """Generate QR code(s) from data"""
+        try:
+            output_dir = Path('output/qr_codes')
+            output_dir.mkdir(parents=True, exist_ok=True)
+            
+            if combined:
+                cleaned_data = clean_json(data)
+                if cleaned_data:
+                    qr = qrcode.QRCode(
+                        version=None,
+                        error_correction=qrcode.constants.ERROR_CORRECT_L,
+                        box_size=10,
+                        border=4,
+                    )
+                    json_str = json.dumps(cleaned_data, ensure_ascii=False)
+                    qr.add_data(json_str)
+                    qr.make(fit=True)
+                    
+                    img = qr.make_image(fill_color="black", back_color="white")
+                    output_path = output_dir / f'combined_qr_{int(time.time())}.png'
+                    img.save(str(output_path))
+                    return [str(output_path)]
+            else:
+                if isinstance(data, list):
+                    paths = []
+                    for idx, item in enumerate(data):
+                        cleaned_item = clean_json(item)
+                        if cleaned_item:
+                            qr = qrcode.QRCode(
+                                version=None,
+                                error_correction=qrcode.constants.ERROR_CORRECT_L,
+                                box_size=10,
+                                border=4,
+                            )
+                            json_str = json.dumps(cleaned_item, ensure_ascii=False)
+                            qr.add_data(json_str)
+                            qr.make(fit=True)
+                            
+                            img = qr.make_image(fill_color="black", back_color="white")
+                            output_path = output_dir / f'item_{idx}_qr_{int(time.time())}.png'
+                            img.save(str(output_path))
+                            paths.append(str(output_path))
+                    return paths
+                else:
+                    cleaned_item = clean_json(data)
                     if cleaned_item:
                         qr = qrcode.QRCode(
                             version=None,
@@ -327,44 +332,26 @@ def generate_qr_code(data: Union[str, Dict], combined: bool = True) -> List[str]
                         qr.make(fit=True)
                         
                         img = qr.make_image(fill_color="black", back_color="white")
-                        output_path = output_dir / f'item_{idx}_qr_{int(time.time())}.png'
+                        output_path = output_dir / f'single_qr_{int(time.time())}.png'
                         img.save(str(output_path))
-                        paths.append(str(output_path))
-                return paths
-            else:
-                cleaned_item = clean_json(data)
-                if cleaned_item:
-                    qr = qrcode.QRCode(
-                        version=None,
-                        error_correction=qrcode.constants.ERROR_CORRECT_L,
-                        box_size=10,
-                        border=4,
-                    )
-                    json_str = json.dumps(cleaned_item, ensure_ascii=False)
-                    qr.add_data(json_str)
-                    qr.make(fit=True)
-                    
-                    img = qr.make_image(fill_color="black", back_color="white")
-                    output_path = output_dir / f'single_qr_{int(time.time())}.png'
-                    img.save(str(output_path))
-                    return [str(output_path)]
-        
+                        return [str(output_path)]
+            
+                return []
+        except Exception as e:
+            logger.error(f"QR generation error: {e}")
             return []
-    except Exception as e:
-        logger.error(f"QR generation error: {e}")
-        return []
-
-def decode_qr(image):
-    decoded_objects = decode(image)
-    results = []
-    for obj in decoded_objects:
-        results.append(obj.data.decode('utf-8'))
-    return results
-        
-        raise ValueError("Unable to decode QR code")
-    except Exception as e:
-        logger.error(f"QR decoding error: {e}")
-        return None, None  # Return None for both data and resolution in case of error
+
+    def decode_qr(image):
+        decoded_objects = decode(image)
+        results = []
+        for obj in decoded_objects:
+            results.append(obj.data.decode('utf-8'))
+        return results
+            
+            raise ValueError("Unable to decode QR code")
+        except Exception as e:
+            logger.error(f"QR decoding error: {e}")
+            return None, None  # Return None for both data and resolution in case of error
 
 def datachat_trained(data_input: str, query: str) -> str:
     """Handle trained data interaction logic"""