from typing import Any, Dict, List, Optional, Tuple, Generator
import gradio as gr

import inspect
def _signature_src(
    prompt: str
) -> Tuple[str, Optional[List[str]], Optional[str]]:
    raise NotImplementedError

def clone_signature(src):
    def deco(dst):
        dst.__signature__ = inspect.signature(src)
        dst.__annotations__ = src.__annotations__.copy()
        return dst
    return deco

@clone_signature(_signature_src)
def generate_image1(**kwargs) -> Tuple[str, Optional[List[str]], Optional[str]]:
    return ("COMPLETE", kwargs["prompt"], None)

def generate_image2(prompt: str) -> Tuple[str, Optional[List[str]], Optional[str]]:
    return ("COMPLETE", prompt, None)

with gr.Blocks() as app:
    gr.api(fn=generate_image1, api_name="generate_image1", show_api=True, queue=True, concurrency_id="gpu")
    gr.api(fn=generate_image2, api_name="generate_image2", show_api=True, queue=True, concurrency_id="gpu")

app.launch(show_error=True)