Commit

src/core/workers.py

@@ -593,6 +593,33 @@ class PostProcessorWorker:
                         os.remove(downloaded_part_file_path)
                     except OSError: pass
                 return 0, 1, filename_to_save_in_main_path, was_original_name_kept_flag, FILE_DOWNLOAD_STATUS_SKIPPED, None
+            
+            if (self.compress_images and downloaded_part_file_path and
+                    is_image(api_original_filename) and
+                    os.path.getsize(downloaded_part_file_path) > 1.5 * 1024 * 1024):
+                
+                self.logger(f"   🔄 Compressing '{api_original_filename}' to WebP...")
+                try:
+                    with Image.open(downloaded_part_file_path) as img:
+                        # Convert to RGB to avoid issues with paletted images or alpha channels in WebP
+                        if img.mode not in ('RGB', 'RGBA'):
+                            img = img.convert('RGBA')
+                        
+                        # Use an in-memory buffer to save the compressed image
+                        output_buffer = BytesIO()
+                        img.save(output_buffer, format='WebP', quality=85)
+                        
+                        # This buffer now holds the compressed data
+                        data_to_write_io = output_buffer
+                        
+                        # Update the filename to use the .webp extension
+                        base, _ = os.path.splitext(filename_to_save_in_main_path)
+                        filename_to_save_in_main_path = f"{base}.webp"
+                        self.logger(f"   ✅ Compression successful. New size: {len(data_to_write_io.getvalue()) / (1024*1024):.2f} MB")
+
+                except Exception as e_compress:
+                    self.logger(f"   ⚠️ Failed to compress '{api_original_filename}': {e_compress}. Saving original file instead.")
+                    data_to_write_io = None # Ensure we fall back to saving the original
 
             effective_save_folder = target_folder_path
             base_name, extension = os.path.splitext(filename_to_save_in_main_path)
@@ -610,15 +637,17 @@ class PostProcessorWorker:
 
             try:
                 if data_to_write_io:
+                    # Write the compressed data from the in-memory buffer
                     with open(final_save_path, 'wb') as f_out:
-                        time.sleep(0.05)
                         f_out.write(data_to_write_io.getvalue())
+                    # Clean up the original downloaded part file
                     if downloaded_part_file_path and os.path.exists(downloaded_part_file_path):
                         try:
                             os.remove(downloaded_part_file_path)
                         except OSError as e_rem:
                             self.logger(f"  -> Failed to remove .part after compression: {e_rem}")
                 else:
+                    # No compression was done, just rename the original file
                     if downloaded_part_file_path and os.path.exists(downloaded_part_file_path):
                         time.sleep(0.1)
                         os.rename(downloaded_part_file_path, final_save_path)
@@ -691,7 +720,6 @@ class PostProcessorWorker:
             else:
                 return 0, 1, filename_to_save_in_main_path, was_original_name_kept_flag, FILE_DOWNLOAD_STATUS_FAILED_RETRYABLE_LATER, details_for_failure
 
-
     def process(self):
 
         result_tuple = (0, 0, [], [], [], None, None)

src/ui/main_window.py

@@ -23,6 +23,11 @@ from PyQt5.QtWidgets import (
     QScrollArea, QListWidgetItem, QSizePolicy, QProgressBar, QAbstractItemView, QFrame,
     QMainWindow, QAction, QGridLayout, 
 )
+try:
+    from PIL import Image
+except ImportError:
+    Image = None
+
 from PyQt5.QtCore import Qt, QThread, pyqtSignal, QObject, QTimer, QSettings, QStandardPaths, QUrl, QSize, QProcess, QMutex, QMutexLocker, QCoreApplication
 from ..services.drive_downloader import download_mega_file as drive_download_mega_file ,download_gdrive_file ,download_dropbox_file 
 from ..core.workers import DownloadThread as BackendDownloadThread