Doc

app.py

@@ -12,12 +12,32 @@ from documentation import DOC_CONTENT
 def auto_label(
     image: Image.Image, imid: int, dataset: ImageDataset
 ) -> tuple[str, ImageDataset]:
+    """Generate automatic label for a single image using AI model.
+
+    Args:
+        image: PIL Image to generate label for
+        imid: Image ID in the dataset
+        dataset: Current ImageDataset instance
+
+    Returns:
+        Tuple of (generated_label_text, updated_dataset)
+    """
     text = get_task_response("<MORE_DETAILED_CAPTION>", image)
     ds = dataset.update_label(imid, text)
     return text, ds
 
 
 def label_changed(label: str, imid: int, dataset: ImageDataset) -> ImageDataset:
+    """Handle label text change event for an image.
+
+    Args:
+        label: New label text
+        imid: Image ID in the dataset
+        dataset: Current ImageDataset instance
+
+    Returns:
+        Updated ImageDataset with new label
+    """
     return dataset.update_label(imid, label)
 
 
@@ -121,6 +141,15 @@ with gr.Blocks(title="Labelizer", fill_width=True) as demo:
 
     @gr.render(inputs=[dataset, is_labeling_in_progress])
     def render_grid(ds, is_labeling_in_progress):
+        """Render the image grid with labels and controls.
+
+        Args:
+            ds: Current ImageDataset instance
+            is_labeling_in_progress: Whether labeling is currently in progress
+
+        Returns:
+            None - renders UI components directly
+        """
         if not ds or len(ds.images) == 0:
             gr.Markdown(DOC_CONTENT)
             return