api-nodes: fixed dynamic pricing format; import comfy_io directly (#10336)

Author: bigcat88

comfy_api/latest/__init__.py

@@ -114,7 +114,9 @@ class Types:
  ComfyAPISync: Type[comfy_api.latest.generated.ComfyAPISyncStub.ComfyAPISyncStub]
 ComfyAPISync = create_sync_class(ComfyAPI_latest)
 
-comfy_io = io # create the new alias for io
+# create new aliases for io and ui
+IO = io
+UI = ui
 
 __all__ = [
  "ComfyAPI",
@@ -124,6 +126,7 @@ class Types:
  "Types",
  "ComfyExtension",
  "io",
- "comfy_io",
+ "IO",
  "ui",
+ "UI",
 ]

comfy_api_nodes/apinode_utils.py

@@ -3,6 +3,7 @@
 import io
 import logging
 import mimetypes
+import os
 from typing import Optional, Union
 from comfy.utils import common_upscale
 from comfy_api.input_impl import VideoFromFile
@@ -702,3 +703,16 @@ def image_tensor_pair_to_batch(
  "center",
  ).movedim(1, -1)
  return torch.cat((image1, image2), dim=0)
+
+
+def get_size(path_or_object: Union[str, io.BytesIO]) -> int:
+ if isinstance(path_or_object, str):
+ return os.path.getsize(path_or_object)
+ return len(path_or_object.getvalue())
+
+
+def validate_container_format_is_mp4(video: VideoInput) -> None:
+ """Validates video container format is MP4."""
+ container_format = video.get_container_format()
+ if container_format not in ["mp4", "mov,mp4,m4a,3gp,3g2,mj2"]:
+ raise ValueError(f"Only MP4 container format supported. Got: {container_format}")

comfy_api_nodes/apis/client.py

@@ -845,7 +845,7 @@ def _display_text_on_node(self, text: str):
  if not self.node_id:
  return
  if self.extracted_price is not None:
- text = f"Price: {self.extracted_price}$\n{text}"
+ text = f"Price: ${self.extracted_price}\n{text}"
  PromptServer.instance.send_progress_text(text, self.node_id)
 
  def _display_time_progress_on_node(self, time_completed: int | float):