akhaliq HF Staff commited on
Commit
c8a3a45
·
1 Parent(s): b1bc991

Remove workflow lock subclass and update Gradio wheel URL

Browse files
Files changed (2) hide show
  1. app.py +1 -94
  2. 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
- class LockedWorkflow(gr.Workflow):
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]==6.17.3
 
1
+ gradio[oauth] @ https://huggingface.co/buckets/gradio/pypi-previews/resolve/eb2cebf05ac60e14832bcee09894832ab0137c9f/gradio-6.17.3-py3-none-any.whl