Update app.py

app.py

@@ -22,11 +22,11 @@ Image.MAX_IMAGE_PIXELS = None
 PENDING_DIR = "pending_uploads"
 os.makedirs(PENDING_DIR, exist_ok=True)
 
-# --- 🔄 BACKGROUND BATCH UPLOADER (Har 45 seconds) ---
+# --- 🔄 BACKGROUND BATCH UPLOADER (Har 30 seconds) ---
 def background_uploader():
     api = HfApi()
     while True:
-        time.sleep(30) # ⏱️ 30 Second Timer
+        time.sleep(30) # ⏱️ 30 Second Timer (Tune bola tha)
         try:
             files_to_upload = os.listdir(PENDING_DIR)
             if not files_to_upload:
@@ -39,9 +39,18 @@ def background_uploader():
             
             for filename in files_to_upload:
                 file_path = os.path.join(PENDING_DIR, filename)
+                
+                # 🗂️ 10k LIMIT BYPASS: Smart Folder System (Max ~9000 safe capacity)
+                # Naam (img_177237_3a7b9c.webp) me se '3a7' nikalega. Isse 4096 alag folders banenge!
+                try:
+                    sub_folder = filename.split('_')[2][:3] 
+                except:
+                    sub_folder = "misc"
+                    
                 operations.append(
                     CommitOperationAdd(
-                        path_in_repo=f"posters/{filename}",
+                        # Ab file 'posters/3a7/img_xxx.webp' me jayegi
+                        path_in_repo=f"posters/{sub_folder}/{filename}",
                         path_or_fileobj=file_path
                     )
                 )
@@ -53,7 +62,7 @@ def background_uploader():
                 repo_type="dataset",
                 token=HF_TOKEN,
                 operations=operations,
-                commit_message=f"Batch upload of {len(operations)} images by OptiPix"
+                commit_message=f"Batch upload of {len(operations)} images by OptiPix (Sharded)"
             )
             print("✅ [BATCH UPLOAD] Success! Local files delete kar rahe hain...")
             
@@ -75,9 +84,9 @@ def xor_encrypt_decrypt_bytes(data: bytes, key: str) -> bytes:
     key_len = len(key_bytes)
     return bytes([b ^ key_bytes[i % key_len] for i, b in enumerate(data)])
 
-def encrypt_filename(filename: str, key: str) -> str:
-    """Filename ko encrypt karke base64 URL-safe banata hai"""
-    xored = "".join(chr(ord(c) ^ ord(key[i % len(key)])) for i, c in enumerate(filename))
+def encrypt_text(text: str, key: str) -> str:
+    """Ab ye sidha target URL ko encrypt karega"""
+    xored = "".join(chr(ord(c) ^ ord(key[i % len(key)])) for i, c in enumerate(text))
     return base64.urlsafe_b64encode(xored.encode()).decode().rstrip("=")
 
 # --- 🧠 SUPER-OPTIMIZER LOGIC ---
@@ -142,24 +151,32 @@ def handle_upload():
         if not image_content:
             return jsonify({"success": False, "error": "No file or URL provided"}), 400
         
-        # Optimize & Encrypt
+        # Optimize & Encrypt Data
         optimized_data = dynamic_optimize(image_content, comp_level)
         encrypted_image_data = xor_encrypt_decrypt_bytes(optimized_data, XOR_KEY)
         
-        # Generate Filenames
-        real_filename = f"img_{int(time.time())}_{os.urandom(3).hex()}.webp"
-        encrypted_filename = encrypt_filename(real_filename, XOR_KEY)
+        # Generate Unique Filename
+        hex_random = os.urandom(3).hex()
+        real_filename = f"img_{int(time.time())}_{hex_random}.webp"
+        
+        # 🗂️ Calculate Sub-folder based on logic
+        sub_folder = hex_random[:3]
+        
+        # 🌐 ASLI JADU: Poora Target URL banao
+        full_hf_url = f"https://huggingface.co/datasets/{HF_REPO}/resolve/main/posters/{sub_folder}/{real_filename}"
+        
+        # 🔐 URL ko encrypt karo taaki worker ko de sake
+        encrypted_url_payload = encrypt_text(full_hf_url, XOR_KEY)
         
-        # Save to Local Queue instead of Direct Upload
+        # Save to Local Queue
         local_filepath = os.path.join(PENDING_DIR, real_filename)
         with open(local_filepath, "wb") as f:
             f.write(encrypted_image_data)
         
         # Create Final Worker URL
         worker_base = WORKER_URL if WORKER_URL.endswith('/') else WORKER_URL + '/'
-        final_cf_url = f"{worker_base}{encrypted_filename}"
+        final_cf_url = f"{worker_base}{encrypted_url_payload}"
         
-        # Exact Same JSON Response!
         return jsonify({
             "success": True, 
             "url": final_cf_url,
@@ -240,7 +257,7 @@ HTML_UI = """
             </div>
             
             <div class="notice-text">
-                ⏳ Note: The image URL will be live in max 45 seconds.
+                ⏳ Note: The image URL will be live in max 30 seconds.
             </div>
             
             <img id="resImg" class="preview-img" src="" alt="Preview will load soon...">
@@ -275,13 +292,12 @@ HTML_UI = """
                     document.getElementById('oldSize').innerText = data.original_size_kb;
                     document.getElementById('newSize').innerText = data.encrypted_size_kb;
                     
-                    // Pehle image src hatayenge taaki broken image na dikhe timer ke dauran
                     document.getElementById('resImg').src = "";
                     
-                    // 45 seconds baad automatically image reload karwane ka try karenge preview ke liye
+                    // 30 seconds wait timeout (plus 1s buffer)
                     setTimeout(() => {
                         document.getElementById('resImg').src = data.url;
-                    }, 46000);
+                    }, 31000);
 
                     resDiv.style.display = "block";
                 } else {