Upload folder using huggingface_hub

app.py

@@ -0,0 +1,153 @@
+# app.py
+"""
+Built with anycoder
+"""
+import gradio as gr
+from daggr import GradioNode, FnNode, InferenceNode, Graph
+
+# --- Helper to parse resolution strings ---
+def parse_resolution_string(res_str):
+    """
+    Extracts the width and height from the resolution string.
+    Example: "1024x1024 ( 1:1 )" -> (1024, 1024)
+    """
+    if not res_str:
+        return None, None
+    
+    # Split by 'x' to get width and height parts
+    parts = res_str.split('x')
+    if len(parts) >= 2:
+        try:
+            w = int(parts[0].strip())
+            h = int(parts[1].split('(')[0].strip())
+            return w, h
+        except ValueError:
+            pass
+    return None, None
+
+# --- Helper to map resolution category to string ---
+def map_resolution_cat(res_cat_value):
+    """
+    Maps the numeric category (e.g., '1024') to the full resolution string
+    (e.g., '1024x1024 ( 1:1 )').
+    """
+    # This mapping corresponds to the order in the Z-Image API documentation
+    # 0: '720x720 ( 1:1 )', 1: '896x512 ( 16:9 )', ...
+    # We construct the mapping based on the default list provided in the docs.
+    
+    # The API docs list specific strings. We map the numeric key to these strings.
+    # Note: The API documentation for /update_res_choices returns a list of strings.
+    # We must match the input 'res_cat_value' (int/str) to the correct index.
+    
+    # The documentation lists the output as a list of strings.
+    # We will map the index to the string.
+    
+    # Let's define the resolution strings based on the documentation provided
+    # to ensure consistency.
+    resolution_map = {
+        '0': '720x720 ( 1:1 )',
+        '1': '896x512 ( 16:9 )',
+        '2': '512x896 ( 9:16 )',
+        '3': '832x544 ( 3:2 )',
+        '4': '544x832 ( 2:3 )',
+        '5': '800x576 ( 4:3 )',
+        '6': '576x800 ( 3:4 )',
+        '7': '1024x1024 ( 1:1 )',
+        '8': '1152x896 ( 9:7 )',
+        '9': '896x1152 ( 7:9 )',
+        '10': '1152x864 ( 4:3 )',
+        '11': '864x1152 ( 3:4 )',
+        '12': '1248x832 ( 3:2 )',
+        '13': '832x1248 ( 2:3 )',
+        '14': '1280x720 ( 16:9 )',
+        '15': '720x1280 ( 9:16 )',
+        '16': '1344x576 ( 21:9 )',
+        '17': '576x1344 ( 9:21 )',
+        '18': '1280x1280 ( 1:1 )',
+        '19': '1440x1120 ( 9:7 )',
+        '20': '1120x1440 ( 7:9 )',
+        '21': '1472x1104 ( 4:3 )',
+        '22': '1104x1472 ( 3:4 )',
+        '23': '1536x1024 ( 3:2 )',
+        '24': '1024x1536 ( 2:3 )',
+        '25': '1536x864 ( 16:9 )',
+        '26': '864x1536 ( 9:16 )',
+        '27': '1680x720 ( 21:9 )',
+        '28': '720x1680 ( 9:21 )',
+    }
+    
+    # Handle if input is int or string
+    key = str(res_cat_value)
+    return resolution_map.get(key, '1024x1024 ( 1:1 )')
+
+# --- Node 1: Resolution Updater ---
+# Updates the resolution dropdown state before generation
+update_resolution_node = GradioNode(
+    space_or_url="Tongyi-MAI/Z-Image",
+    api_name="/update_res_choices",
+    inputs={
+        "res_cat_value": gr.Textbox(label="Resolution Category Value", value="1024"),
+    },
+    outputs={
+        "updated_resolution": gr.Textbox(label="Updated Resolution String"),
+    },
+    postprocess=lambda original, target: target.get("updated_resolution", "1024x1024 ( 1:1 )"),
+)
+
+# --- Node 2: Image Generator ---
+# Generates the image based on the resolved resolution
+generate_image_node = GradioNode(
+    space_or_url="Tongyi-MAI/Z-Image",
+    api_name="/generate",
+    inputs={
+        "prompt": gr.Textbox(label="Prompt", placeholder="Describe your image...", value="Hello!!"),
+        "negative_prompt": gr.Textbox(label="Negative Prompt", placeholder="Blurry, low quality...", value="Hello!!"),
+        "resolution": gr.Textbox(label="Resolution", value="1024x1024 ( 1:1 )"),
+        "seed": gr.Number(label="Seed", value=42),
+        "num_inference_steps": gr.Number(label="Inference Steps", value=30, precision=0),
+        "guidance_scale": gr.Number(label="Guidance Scale (CFG)", value=4),
+        "cfg_normalization": gr.Checkbox(label="CFG Normalization", value=False),
+        "random_seed": gr.Checkbox(label="Random Seed", value=True),
+        "gallery_images": gr.Gallery(label="Gallery Images", visible=False),
+    },
+    outputs={
+        "images": gr.Gallery(label="Generated Images"),
+        "seed_used": gr.Textbox(label="Seed Used"),
+        "seed_number": gr.Number(label="Seed"),
+    },
+)
+
+# --- Node 3: Image Preview (FnNode) ---
+# Extracts the first image from the gallery for preview
+preview_node = FnNode(
+    fn=lambda images: images[0] if images and len(images) > 0 else None,
+    inputs={"images": gr.Gallery(label="Images")},
+    outputs={"preview": gr.Image(label="Preview")},
+)
+
+# --- Node 4: Metadata Logger (FnNode) ---
+# Logs metadata about the generation
+log_metadata_node = FnNode(
+    fn=lambda prompt, seed, res: f"Prompt: {prompt}\nSeed: {seed}\nResolution: {res}",
+    inputs={
+        "prompt": gr.Textbox(label="Prompt"),
+        "seed": gr.Number(label="Seed"),
+        "res": gr.Textbox(label="Resolution"),
+    },
+    outputs={"log": gr.Textbox(label="Generation Log", interactive=False)},
+)
+
+# --- Build the Graph ---
+graph = Graph(
+    name="Z-Image Workflow",
+    nodes=[
+        update_resolution_node,
+        generate_image_node,
+        preview_node,
+        log_metadata_node
+    ]
+)
+
+# --- Launch ---
+if __name__ == "__main__":
+    graph.launch()

requirements.txt

@@ -0,0 +1,2 @@
+daggr
+gradio