Commit
·
b2f617d
1
Parent(s):
2081532
Set queue to False
Browse files- chat_interface_preference.py +11 -11
chat_interface_preference.py
CHANGED
|
@@ -308,13 +308,13 @@ class ChatInterface(Blocks):
|
|
| 308 |
self._setup_api()
|
| 309 |
|
| 310 |
def _setup_events(self) -> None:
|
| 311 |
-
|
| 312 |
-
|
| 313 |
[self.textbox.submit, self.submit_btn_one.click] if self.submit_btn_one else [self.textbox.submit]
|
| 314 |
)
|
| 315 |
-
|
| 316 |
submit_triggers_two = [self.submit_btn_two.click]
|
| 317 |
-
for _fn, _triggers in [(
|
| 318 |
submit_event = (
|
| 319 |
on(
|
| 320 |
_triggers,
|
|
@@ -339,7 +339,7 @@ class ChatInterface(Blocks):
|
|
| 339 |
concurrency_limit=cast(Union[int, Literal["default"], None], self.concurrency_limit),
|
| 340 |
)
|
| 341 |
)
|
| 342 |
-
self._setup_stop_events(
|
| 343 |
|
| 344 |
partial_fn_a, partial_fn_b, partial_fn_ab = (
|
| 345 |
functools.partial(self._log_fn, log="a"),
|
|
@@ -356,13 +356,13 @@ class ChatInterface(Blocks):
|
|
| 356 |
[self.saved_input, self.chatbot_state],
|
| 357 |
[self.chatbot, self.saved_input, self.chatbot_state],
|
| 358 |
show_api=False,
|
| 359 |
-
queue=
|
| 360 |
).then(
|
| 361 |
async_lambda(lambda x: x),
|
| 362 |
[self.saved_input],
|
| 363 |
[self.textbox],
|
| 364 |
show_api=False,
|
| 365 |
-
queue=
|
| 366 |
)
|
| 367 |
|
| 368 |
# if self.retry_btn:
|
|
@@ -430,14 +430,14 @@ class ChatInterface(Blocks):
|
|
| 430 |
None,
|
| 431 |
[self.submit_btn_one, self.stop_btn],
|
| 432 |
show_api=False,
|
| 433 |
-
queue=
|
| 434 |
)
|
| 435 |
event_to_cancel.then(
|
| 436 |
async_lambda(lambda: (Button(visible=True), Button(visible=False))),
|
| 437 |
None,
|
| 438 |
[self.submit_btn_one, self.stop_btn],
|
| 439 |
show_api=False,
|
| 440 |
-
queue=
|
| 441 |
)
|
| 442 |
else:
|
| 443 |
for event_trigger in event_triggers:
|
|
@@ -446,14 +446,14 @@ class ChatInterface(Blocks):
|
|
| 446 |
None,
|
| 447 |
[self.stop_btn],
|
| 448 |
show_api=False,
|
| 449 |
-
queue=
|
| 450 |
)
|
| 451 |
event_to_cancel.then(
|
| 452 |
async_lambda(lambda: Button(visible=False)),
|
| 453 |
None,
|
| 454 |
[self.stop_btn],
|
| 455 |
show_api=False,
|
| 456 |
-
queue=
|
| 457 |
)
|
| 458 |
self.stop_btn.click(
|
| 459 |
None,
|
|
|
|
| 308 |
self._setup_api()
|
| 309 |
|
| 310 |
def _setup_events(self) -> None:
|
| 311 |
+
submit_fn_one = self._stream_fn if self.is_generator else self._submit_fn
|
| 312 |
+
submit_triggers_one = (
|
| 313 |
[self.textbox.submit, self.submit_btn_one.click] if self.submit_btn_one else [self.textbox.submit]
|
| 314 |
)
|
| 315 |
+
submit_fn_two = functools.partial(submit_fn_one, n_generations=2)
|
| 316 |
submit_triggers_two = [self.submit_btn_two.click]
|
| 317 |
+
for _fn, _triggers in [(submit_fn_one, submit_triggers_one), (submit_fn_two, submit_triggers_two)]:
|
| 318 |
submit_event = (
|
| 319 |
on(
|
| 320 |
_triggers,
|
|
|
|
| 339 |
concurrency_limit=cast(Union[int, Literal["default"], None], self.concurrency_limit),
|
| 340 |
)
|
| 341 |
)
|
| 342 |
+
self._setup_stop_events(_triggers, submit_event)
|
| 343 |
|
| 344 |
partial_fn_a, partial_fn_b, partial_fn_ab = (
|
| 345 |
functools.partial(self._log_fn, log="a"),
|
|
|
|
| 356 |
[self.saved_input, self.chatbot_state],
|
| 357 |
[self.chatbot, self.saved_input, self.chatbot_state],
|
| 358 |
show_api=False,
|
| 359 |
+
queue=False,
|
| 360 |
).then(
|
| 361 |
async_lambda(lambda x: x),
|
| 362 |
[self.saved_input],
|
| 363 |
[self.textbox],
|
| 364 |
show_api=False,
|
| 365 |
+
queue=False,
|
| 366 |
)
|
| 367 |
|
| 368 |
# if self.retry_btn:
|
|
|
|
| 430 |
None,
|
| 431 |
[self.submit_btn_one, self.stop_btn],
|
| 432 |
show_api=False,
|
| 433 |
+
queue=False,
|
| 434 |
)
|
| 435 |
event_to_cancel.then(
|
| 436 |
async_lambda(lambda: (Button(visible=True), Button(visible=False))),
|
| 437 |
None,
|
| 438 |
[self.submit_btn_one, self.stop_btn],
|
| 439 |
show_api=False,
|
| 440 |
+
queue=False,
|
| 441 |
)
|
| 442 |
else:
|
| 443 |
for event_trigger in event_triggers:
|
|
|
|
| 446 |
None,
|
| 447 |
[self.stop_btn],
|
| 448 |
show_api=False,
|
| 449 |
+
queue=False,
|
| 450 |
)
|
| 451 |
event_to_cancel.then(
|
| 452 |
async_lambda(lambda: Button(visible=False)),
|
| 453 |
None,
|
| 454 |
[self.stop_btn],
|
| 455 |
show_api=False,
|
| 456 |
+
queue=False,
|
| 457 |
)
|
| 458 |
self.stop_btn.click(
|
| 459 |
None,
|