Upload app.py

app.py

@@ -2061,6 +2061,13 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
         def _generate_image(argv: List[Any]) -> Generator[Tuple[str, Optional[List[str]], Optional[str]], None, None]:
             # Delegate to existing generator
             yield from sd_gen_generate_pipeline(*argv)
+        
+        _API_SIG = inspect.signature(_signature_src)
+
+        def _bind_api_args(*args: Any, **kwargs: Any) -> Dict[str, Any]:
+            bound = _API_SIG.bind_partial(*args, **kwargs)
+            bound.apply_defaults()
+            return bound.arguments
 
         # 3) Signature-clone decorator: keep one signature across both endpoints
         def clone_signature(src):
@@ -2072,23 +2079,35 @@ with gr.Blocks(theme=args.theme, elem_id="main", fill_width=True, fill_height=Fa
 
         # 4) Implementations: both share the same signature via clone_signature
         @clone_signature(_signature_src)
-        def generate_image(**kwargs) -> Tuple[str, Optional[List[str]], Optional[str]]:
-            _load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
-            argv = _argv_from_kwargs(kwargs)
-            last = ("COMPLETE", None, None)
+        def generate_image(*args: Any, **kwargs: Any) -> Tuple[str, Optional[List[str]], Optional[str]]:
+            params = _bind_api_args(*args, **kwargs)
+            _load_model(
+                params["model_name"],
+                params["vae_model"],
+                params["task"],
+                params["controlnet_model"],
+            )
+            argv = _argv_from_kwargs(params)
+            last: Tuple[str, Optional[List[str]], Optional[str]] = ("COMPLETE", None, None)
             for last in _generate_image(argv):
                 pass
             return last
 
         @clone_signature(_signature_src)
-        def generate_image_stream(**kwargs) -> Generator[Tuple[str, Optional[List[str]], Optional[str]], None, None]:
-            _load_model(kwargs["model_name"], kwargs["vae_model"], kwargs["task"], kwargs["controlnet_model"])
-            argv = _argv_from_kwargs(kwargs)
+        def generate_image_stream(*args: Any, **kwargs: Any) -> Generator[Tuple[str, Optional[List[str]], Optional[str]], None, None]:
+            params = _bind_api_args(*args, **kwargs)
+            _load_model(
+                params["model_name"],
+                params["vae_model"],
+                params["task"],
+                params["controlnet_model"],
+            )
+            argv = _argv_from_kwargs(params)
             yield from _generate_image(argv)
 
         # 5) Register two APIs with identical, named kwargs
         gr.api(generate_image, api_name="generate_image", show_api=True, queue=True, concurrency_id="gpu")
-        gr.api(generate_image_stream, api_name="generate_image_stream", show_api=True, queue=True, concurrency_id="gpu")
+        gr.api(generate_image_stream, api_name="generate_image_stream", show_api=True, queue=True, concurrency_id="gpu").stream()
 
     gr.LoginButton()
     gr.DuplicateButton(value="Duplicate Space for private use (This demo does not work on CPU. Requires GPU Space)")