Be more specific with flake8 ignores

Author: zquestz

.flake8

@@ -1,6 +1,5 @@
 [flake8]
 max-line-length = 79
-ignore = E402
 exclude =
     __pycache__,
     *.pyc,
@@ -11,9 +10,11 @@ exclude =
     dist,
     *.egg-info
 
-# E402: module level import not at top of file
-# This is ignored because gi.require_version() calls must come before gi imports
-# which is a required pattern for GTK/GObject applications
-
+# Per-file ignores for specific legitimate violations
 per-file-ignores =
-    models/factory.py:F401
+    # F401: Import model modules to register them automatically
+    models/factory.py:F401,
+    # E402: gi.require_version() must be called before gi imports
+    dialog.py:E402,
+    dialog_gtk.py:E402,
+    dream-prompter.py:E402

api.py

@@ -118,23 +118,31 @@ def edit_image(
             if not current_image_data:
                 return None, _("Failed to export current image")
 
-            if (progress_callback and not progress_callback(
-                _("Building Replicate edit request..."), PROGRESS_UPLOAD
-            )):
-                return None, _("Operation cancelled")
-
-            model_input = self._build_edit_input(
-                prompt, current_image_data, reference_images
-            )
-
-            if (progress_callback and not progress_callback(
-                _("Sending edit request to Replicate..."), PROGRESS_PROCESS
-            )):
-                return None, _("Operation cancelled")
-
-            output = self._execute_api_call(model_input)
-            if isinstance(output, str):
-                return None, output
+            with self._prepare_reference_images(
+                reference_images, self.model.max_reference_images_edit
+            ) as ref_files:
+                if (progress_callback and not progress_callback(
+                    _("Building Replicate edit request..."), PROGRESS_UPLOAD
+                )):
+                    return None, _("Operation cancelled")
+
+                from settings import get_model_settings
+                user_settings = get_model_settings(self.model.name)
+                model_input = self.model.build_edit_input(
+                    prompt=prompt,
+                    main_image=io.BytesIO(current_image_data),
+                    reference_images=ref_files if ref_files else None,
+                    user_settings=user_settings
+                )
+
+                if (progress_callback and not progress_callback(
+                    _("Sending edit request to Replicate..."), PROGRESS_PROCESS
+                )):
+                    return None, _("Operation cancelled")
+
+                output = self._execute_api_call(model_input)
+                if isinstance(output, str):
+                    return None, output
 
             return self._process_response(output, progress_callback)
 
@@ -171,67 +179,36 @@ def generate_image(
             )
 
         try:
-            if (progress_callback and not progress_callback(
-                _("Generating image with Replicate..."), PROGRESS_PREPARE
-            )):
-                return None, _("Operation cancelled")
-
-            model_input = self._build_generation_input(
-                prompt, reference_images
-            )
-
-            if (progress_callback and not progress_callback(
-                _("Sending request to Replicate..."), PROGRESS_PROCESS
-            )):
-                return None, _("Operation cancelled")
-
-            output = self._execute_api_call(model_input)
-            if isinstance(output, str):
-                return None, output
+            with self._prepare_reference_images(
+                reference_images, self.model.max_reference_images
+            ) as ref_files:
+                if (progress_callback and not progress_callback(
+                    _("Generating image with Replicate..."), PROGRESS_PREPARE
+                )):
+                    return None, _("Operation cancelled")
+
+                from settings import get_model_settings
+                user_settings = get_model_settings(self.model.name)
+                model_input = self.model.build_generation_input(
+                    prompt=prompt,
+                    reference_images=ref_files if reference_images else None,
+                    user_settings=user_settings
+                )
+
+                if (progress_callback and not progress_callback(
+                    _("Sending request to Replicate..."), PROGRESS_PROCESS
+                )):
+                    return None, _("Operation cancelled")
+
+                output = self._execute_api_call(model_input)
+                if isinstance(output, str):
+                    return None, output
 
             return self._process_response(output, progress_callback)
 
         except Exception as e:
             return None, _("Unexpected error: {error}").format(error=str(e))
 
-    def _build_edit_input(
-        self,
-        prompt: str,
-        current_image_data: bytes,
-        reference_images: Optional[List[str]] = None
-    ) -> dict:
-        """Build input parameters for edit operation."""
-        from settings import get_model_settings
-
-        with self._prepare_reference_images(
-            reference_images, self.model.max_reference_images_edit
-        ) as ref_files:
-            user_settings = get_model_settings(self.model.name)
-            return self.model.build_edit_input(
-                prompt=prompt,
-                main_image=io.BytesIO(current_image_data),
-                reference_images=ref_files if ref_files else None,
-                user_settings=user_settings
-            )
-
-    def _build_generation_input(
-        self,
-        prompt: str,
-        reference_images: Optional[List[str]] = None
-    ) -> dict:
-        """Build input parameters for generation operation."""
-        from settings import get_model_settings
-
-        with self._prepare_reference_images(
-            reference_images, self.model.max_reference_images
-        ) as ref_files:
-            user_settings = get_model_settings(self.model.name)
-            return self.model.build_generation_input(
-                prompt=prompt,
-                reference_images=ref_files if reference_images else None,
-                user_settings=user_settings
-            )
-
     def _bytes_to_pixbuf(
         self, image_bytes: bytes
     ) -> Optional[GdkPixbuf.Pixbuf]: