Update app.py

app.py

@@ -3,9 +3,10 @@ import io
 import time
 import requests
 import base64
+import threading
 from flask import Flask, request, jsonify, render_template_string
 from PIL import Image, ImageOps
-from huggingface_hub import HfApi
+from huggingface_hub import HfApi, CommitOperationAdd
 
 app = Flask(__name__)
 
@@ -17,18 +18,66 @@ XOR_KEY = os.environ.get("XOR_KEY", "badal_master_key") # Tumhari Secret Chabi
 
 Image.MAX_IMAGE_PIXELS = None
 
+# --- 📁 LOCAL STORAGE SETUP (Queue ke liye) ---
+PENDING_DIR = "pending_uploads"
+os.makedirs(PENDING_DIR, exist_ok=True)
+
+# --- 🔄 BACKGROUND BATCH UPLOADER (Har 45 seconds) ---
+def background_uploader():
+    api = HfApi()
+    while True:
+        time.sleep(45) # ⏱️ 45 Second Timer
+        try:
+            files_to_upload = os.listdir(PENDING_DIR)
+            if not files_to_upload:
+                continue
+                
+            print(f"🚀 [BATCH UPLOAD] {len(files_to_upload)} files ko ek saath upload kar rahe hain...")
+            
+            operations = []
+            uploaded_files = []
+            
+            for filename in files_to_upload:
+                file_path = os.path.join(PENDING_DIR, filename)
+                operations.append(
+                    CommitOperationAdd(
+                        path_in_repo=f"posters/{filename}",
+                        path_or_fileobj=file_path
+                    )
+                )
+                uploaded_files.append(file_path)
+            
+            # Single Commit on Hugging Face
+            api.create_commit(
+                repo_id=HF_REPO,
+                repo_type="dataset",
+                token=HF_TOKEN,
+                operations=operations,
+                commit_message=f"Batch upload of {len(operations)} images by OptiPix"
+            )
+            print("✅ [BATCH UPLOAD] Success! Local files delete kar rahe hain...")
+            
+            for f in uploaded_files:
+                try:
+                    os.remove(f)
+                except:
+                    pass
+        except Exception as e:
+            print(f"❌ [BATCH UPLOAD] Error: {e}")
+
+# App start hote hi background thread on
+threading.Thread(target=background_uploader, daemon=True).start()
+
 # --- 🔐 XOR ENCRYPTION ENGINE ---
 def xor_encrypt_decrypt_bytes(data: bytes, key: str) -> bytes:
     """Image ke binary data ko corrupt/decrypt karta hai"""
     key_bytes = key.encode()
     key_len = len(key_bytes)
-    # Python me bytes ko XOR karna
     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))
-    # Padding hata dete hain taaki URL clean dikhe
     return base64.urlsafe_b64encode(xored.encode()).decode().rstrip("=")
 
 # --- 🧠 SUPER-OPTIMIZER LOGIC ---
@@ -79,12 +128,10 @@ def handle_upload():
         return jsonify({"success": False, "error": "HF Secrets missing!"}), 500
 
     try:
-        # 1. Check if Direct File Upload
         if 'file' in request.files and request.files['file'].filename != '':
             file = request.files['file']
             image_content = file.read()
         
-        # 2. Check if URL Upload
         elif request.args.get('url') or request.form.get('url'):
             img_url = request.args.get('url') or request.form.get('url')
             resp = requests.get(img_url, timeout=20)
@@ -95,36 +142,28 @@ def handle_upload():
         if not image_content:
             return jsonify({"success": False, "error": "No file or URL provided"}), 400
         
-        # 3. Optimize Image
+        # Optimize & Encrypt
         optimized_data = dynamic_optimize(image_content, comp_level)
-        
-        # 4. 🔐 ENCRYPT THE IMAGE DATA (Magic Happens Here!)
         encrypted_image_data = xor_encrypt_decrypt_bytes(optimized_data, XOR_KEY)
         
-        # 5. Generate Real Filename & Encrypt it
+        # Generate Filenames
         real_filename = f"img_{int(time.time())}_{os.urandom(3).hex()}.webp"
         encrypted_filename = encrypt_filename(real_filename, XOR_KEY)
         
-        path_in_repo = f"posters/{real_filename}"
+        # Save to Local Queue instead of Direct Upload
+        local_filepath = os.path.join(PENDING_DIR, real_filename)
+        with open(local_filepath, "wb") as f:
+            f.write(encrypted_image_data)
         
-        # 6. Upload ENCRYPTED GARBAGE to Hugging Face
-        api = HfApi()
-        api.upload_file(
-            path_or_fileobj=encrypted_image_data, # Uploading corrupted bytes
-            path_in_repo=path_in_repo,
-            repo_id=HF_REPO,
-            repo_type="dataset",
-            token=HF_TOKEN
-        )
-        
-        # 7. Create Final Worker URL for User
+        # Create Final Worker URL
         worker_base = WORKER_URL if WORKER_URL.endswith('/') else WORKER_URL + '/'
         final_cf_url = f"{worker_base}{encrypted_filename}"
         
+        # Exact Same JSON Response!
         return jsonify({
             "success": True, 
-            "url": final_cf_url,  # Cloudflare Worker URL with Encrypted Name
-            "real_hf_filename": real_filename, # (Just for debugging, DB me mat dalna)
+            "url": final_cf_url,
+            "real_hf_filename": real_filename, 
             "level_used": comp_level,
             "original_size_kb": round(len(image_content)/1024, 2),
             "encrypted_size_kb": round(len(encrypted_image_data)/1024, 2)
@@ -158,6 +197,7 @@ HTML_UI = """
         .url-box { background: #0f172a; padding: 12px; border: 1px solid #059669; border-radius: 6px; margin: 10px 0; font-family: monospace; font-size: 13px; color: #34d399; overflow-x: auto; white-space: nowrap; }
         .stats { display: flex; justify-content: space-between; font-size: 13px; margin-top: 15px; background: rgba(0,0,0,0.2); padding: 8px 12px; border-radius: 6px; }
         .preview-img { max-width: 100%; border-radius: 8px; margin-top: 15px; border: 1px solid #334155; display: block; }
+        .notice-text { color: #fbbf24; font-size: 13.5px; margin-top: 15px; font-weight: 600; text-align: center; background: rgba(251, 191, 36, 0.1); padding: 10px; border-radius: 6px; border: 1px solid rgba(251, 191, 36, 0.3); }
         @keyframes fadeIn { from { opacity: 0; transform: translateY(10px); } to { opacity: 1; transform: translateY(0); } }
         .copy-btn { background: #334155; color: #fff; padding: 5px 10px; font-size: 11px; border-radius: 4px; border: none; cursor: pointer; float: right; margin-top: -35px; margin-right: 5px; }
     </style>
@@ -198,7 +238,12 @@ HTML_UI = """
                 <span>Original: <span id="oldSize">--</span> KB</span>
                 <span style="color: #34d399;">Optimized: <span id="newSize">--</span> KB</span>
             </div>
-            <img id="resImg" class="preview-img" src="" alt="Preview">
+            
+            <div class="notice-text">
+                ⏳ Note: The image URL will be live in max 45 seconds.
+            </div>
+            
+            <img id="resImg" class="preview-img" src="" alt="Preview will load soon...">
         </div>
     </div>
 
@@ -229,7 +274,15 @@ HTML_UI = """
                     document.getElementById('resLink').innerText = data.url;
                     document.getElementById('oldSize').innerText = data.original_size_kb;
                     document.getElementById('newSize').innerText = data.encrypted_size_kb;
-                    document.getElementById('resImg').src = data.url;
+                    
+                    // 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
+                    setTimeout(() => {
+                        document.getElementById('resImg').src = data.url;
+                    }, 46000);
+
                     resDiv.style.display = "block";
                 } else {
                     alert("Processing Error: " + data.error);
@@ -256,4 +309,4 @@ def index():
     return render_template_string(HTML_UI)
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860)