Upload app.py with huggingface_hub

app.py

@@ -335,6 +335,24 @@ server = Server(title="PASITA API", version="1.0.0", description=API_DESCRIPTION
 
 _V2_CONVERT_PATHS = frozenset({"/gradio_api/call/v2/convert", "/gradio_api/call/v2/convert/"})
 _V1_CONVERT_PATHS = frozenset({"/gradio_api/call/convert", "/gradio_api/call/convert/"})
+_QUEUE_JOIN_PATHS = frozenset({"/gradio_api/queue/join", "/gradio_api/queue/join/"})
+_CONVERT_API_NAMES = frozenset({"convert", "convert_1"})
+
+
+def _is_convert_fn(scope, fn_index) -> bool:
+    try:
+        app = scope.get("app")
+        if app is None or not hasattr(app, "get_blocks"):
+            return False
+        blocks = app.get_blocks()
+        fns = getattr(blocks, "fns", None)
+        if not fns or not isinstance(fn_index, int) or isinstance(fn_index, bool):
+            return False
+        if not 0 <= fn_index < len(fns):
+            return False
+        return getattr(fns[fn_index], "api_name", None) in _CONVERT_API_NAMES
+    except Exception:
+        return False
 _CONVERT_PARAMS = (("text", ""), ("num_candidates", 3), ("structured_fallback", True), ("max_output_tokens", None))
 
 
@@ -389,7 +407,8 @@ class _LegacyConvertBodyMiddleware:
         path = scope.get("path", "")
         is_v2 = path in _V2_CONVERT_PATHS
         is_v1 = path in _V1_CONVERT_PATHS
-        if not (is_v2 or is_v1):
+        is_queue = path in _QUEUE_JOIN_PATHS
+        if not (is_v2 or is_v1 or is_queue):
             await self.app(scope, receive, send)
             return
         body = b""
@@ -403,36 +422,39 @@ class _LegacyConvertBodyMiddleware:
         except Exception:
             parsed = None
         if isinstance(parsed, dict):
-            if is_v2 and "data" in parsed and "text" not in parsed:
-                data = parsed.get("data")
-                if isinstance(data, list):
-                    named: dict = {}
-                    for (name, _default), value in zip(_CONVERT_PARAMS, data):
-                        named[name] = value
-                    for key in ("session_hash", "event_id", "fn_index", "trigger_id", "batched"):
-                        if key in parsed:
-                            named[key] = parsed[key]
-                    parsed = named
-            if is_v2:
-                for name, default in _CONVERT_PARAMS:
-                    if name in parsed and (parsed[name] is None or parsed[name] == ""):
-                        parsed[name] = default
-                for name in ("num_candidates", "structured_fallback", "max_output_tokens"):
-                    if name in parsed:
-                        parsed[name] = _clean_convert_value(name, parsed[name])
-            elif is_v1 and isinstance(parsed.get("data"), list):
-                data = list(parsed["data"])
-                while len(data) < len(_CONVERT_PARAMS):
-                    data.append(None)
-                cleaned = []
-                for (name, default), value in zip(_CONVERT_PARAMS, data):
-                    if value is None or value == "":
-                        cleaned.append(default if name != "text" else "")
-                    else:
-                        cleaned.append(_clean_convert_value(name, value))
-                parsed = dict(parsed)
-                parsed["data"] = cleaned
-            body = json.dumps(parsed, default=str).encode("utf-8")
+            if is_queue and not _is_convert_fn(scope, parsed.get("fn_index")):
+                pass
+            else:
+                if is_v2 and "data" in parsed and "text" not in parsed:
+                    data = parsed.get("data")
+                    if isinstance(data, list):
+                        named: dict = {}
+                        for (name, _default), value in zip(_CONVERT_PARAMS, data):
+                            named[name] = value
+                        for key in ("session_hash", "event_id", "fn_index", "trigger_id", "batched"):
+                            if key in parsed:
+                                named[key] = parsed[key]
+                        parsed = named
+                if is_v2:
+                    for name, default in _CONVERT_PARAMS:
+                        if name in parsed and (parsed[name] is None or parsed[name] == ""):
+                            parsed[name] = default
+                    for name in ("num_candidates", "structured_fallback", "max_output_tokens"):
+                        if name in parsed:
+                            parsed[name] = _clean_convert_value(name, parsed[name])
+                elif isinstance(parsed.get("data"), list):
+                    data = list(parsed["data"])
+                    while len(data) < len(_CONVERT_PARAMS):
+                        data.append(None)
+                    cleaned = []
+                    for (name, default), value in zip(_CONVERT_PARAMS, data):
+                        if value is None or value == "":
+                            cleaned.append(default if name != "text" else "")
+                        else:
+                            cleaned.append(_clean_convert_value(name, value))
+                    parsed = dict(parsed)
+                    parsed["data"] = cleaned
+                body = json.dumps(parsed, default=str).encode("utf-8")
 
         async def receive_once():
             return {"type": "http.request", "body": body, "more_body": False}