Remove workflow lock subclass and update Gradio wheel URL
Browse files- app.py +1 -94
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -239,100 +239,7 @@ def generate_ideogram_image(prompt: str) -> dict | None:
|
|
| 239 |
raise e
|
| 240 |
|
| 241 |
|
| 242 |
-
|
| 243 |
-
def _build(self):
|
| 244 |
-
from gradio.components.workflowcanvas import WorkflowCanvas
|
| 245 |
-
from gradio.workflow import (
|
| 246 |
-
get_token,
|
| 247 |
-
call_space,
|
| 248 |
-
call_model,
|
| 249 |
-
fetch_dataset,
|
| 250 |
-
get_dataset_schema,
|
| 251 |
-
_workflow_from_bind,
|
| 252 |
-
)
|
| 253 |
-
from gradio.utils import get_space
|
| 254 |
-
|
| 255 |
-
def _load_initial() -> str | None:
|
| 256 |
-
try:
|
| 257 |
-
with open(self._workflow_file, encoding="utf-8") as f:
|
| 258 |
-
return f.read()
|
| 259 |
-
except FileNotFoundError:
|
| 260 |
-
if self._bound:
|
| 261 |
-
return _workflow_from_bind(
|
| 262 |
-
self._bound, self._edges, name=self._workflow_name
|
| 263 |
-
)
|
| 264 |
-
return None
|
| 265 |
-
|
| 266 |
-
bound = self._bound
|
| 267 |
-
|
| 268 |
-
def call_fn(data, _token=None) -> str:
|
| 269 |
-
if _token:
|
| 270 |
-
t_str = _token.token if hasattr(_token, "token") else _token
|
| 271 |
-
if t_str:
|
| 272 |
-
workflow_token.set(str(t_str))
|
| 273 |
-
fn_name = data[0] if data else ""
|
| 274 |
-
try:
|
| 275 |
-
args_json = data[1] if len(data) > 1 else "[]"
|
| 276 |
-
fn = bound.get(fn_name)
|
| 277 |
-
if fn is None:
|
| 278 |
-
return json.dumps(
|
| 279 |
-
{
|
| 280 |
-
"error": f"No function '{fn_name}' bound to this workflow",
|
| 281 |
-
"error_type": "unknown",
|
| 282 |
-
"suggestion": "Check the bind= argument to Workflow()",
|
| 283 |
-
}
|
| 284 |
-
)
|
| 285 |
-
args = json.loads(args_json)
|
| 286 |
-
if not isinstance(args, list):
|
| 287 |
-
args = [args]
|
| 288 |
-
result = fn(*args)
|
| 289 |
-
result = list(result) if isinstance(result, (list, tuple)) else [result]
|
| 290 |
-
return json.dumps(result)
|
| 291 |
-
except Exception as e:
|
| 292 |
-
return json.dumps(
|
| 293 |
-
{"error": str(e), "error_type": "unknown", "suggestion": ""}
|
| 294 |
-
)
|
| 295 |
-
|
| 296 |
-
# Mocked/neutralized methods for locked read-only mode:
|
| 297 |
-
def save_workflow(data, _token=None) -> str:
|
| 298 |
-
return "ok"
|
| 299 |
-
|
| 300 |
-
def list_bound_fns(_data=None, _token=None) -> str:
|
| 301 |
-
return "[]"
|
| 302 |
-
|
| 303 |
-
def search_spaces(data, token=None) -> str:
|
| 304 |
-
return "[]"
|
| 305 |
-
|
| 306 |
-
def search_models(data, token=None) -> str:
|
| 307 |
-
return "[]"
|
| 308 |
-
|
| 309 |
-
def search_datasets(data, token=None) -> str:
|
| 310 |
-
return "[]"
|
| 311 |
-
|
| 312 |
-
server_functions = [
|
| 313 |
-
get_token,
|
| 314 |
-
call_space,
|
| 315 |
-
call_model,
|
| 316 |
-
fetch_dataset,
|
| 317 |
-
search_spaces,
|
| 318 |
-
search_models,
|
| 319 |
-
search_datasets,
|
| 320 |
-
get_dataset_schema,
|
| 321 |
-
call_fn,
|
| 322 |
-
list_bound_fns,
|
| 323 |
-
save_workflow,
|
| 324 |
-
]
|
| 325 |
-
|
| 326 |
-
with self:
|
| 327 |
-
if get_space() is not None and os.getenv("OAUTH_CLIENT_ID"):
|
| 328 |
-
gr.LoginButton(visible=False)
|
| 329 |
-
WorkflowCanvas(
|
| 330 |
-
value=_load_initial,
|
| 331 |
-
server_functions=server_functions,
|
| 332 |
-
)
|
| 333 |
-
|
| 334 |
-
|
| 335 |
-
demo = LockedWorkflow(bind=[generate_prompt, generate_image, generate_z_image, edit_image, generate_ideogram_image])
|
| 336 |
|
| 337 |
if __name__ == "__main__":
|
| 338 |
demo.launch()
|
|
|
|
| 239 |
raise e
|
| 240 |
|
| 241 |
|
| 242 |
+
demo = gr.Workflow(bind=[generate_prompt, generate_image, generate_z_image, edit_image, generate_ideogram_image])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 243 |
|
| 244 |
if __name__ == "__main__":
|
| 245 |
demo.launch()
|
requirements.txt
CHANGED
|
@@ -1 +1 @@
|
|
| 1 |
-
gradio[oauth]
|
|
|
|
| 1 |
+
gradio[oauth] @ https://huggingface.co/buckets/gradio/pypi-previews/resolve/eb2cebf05ac60e14832bcee09894832ab0137c9f/gradio-6.17.3-py3-none-any.whl
|