32 lines
1016 B
Python
32 lines
1016 B
Python
|
# SPDX-License-Identifier: Apache-2.0
|
||
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
|
||
|
|
from stream_client import stream
|
||
|
|
|
||
|
|
ICON = {"response": "SPEAK ", "silence": "silence ", "delegate": "DELEGATE"}
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("video")
|
||
|
|
parser.add_argument("--server", default="http://127.0.0.1:8070")
|
||
|
|
parser.add_argument("--query", default=None)
|
||
|
|
parser.add_argument("--fps", type=float, default=1.0)
|
||
|
|
parser.add_argument("--session-id", default="cli")
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
for tick in stream(args.video, args.server, args.session_id, args.query, args.fps):
|
||
|
|
if tick.action == "silence":
|
||
|
|
continue
|
||
|
|
line = f"[{tick.t:6.1f}s] {ICON.get(tick.action)} {tick.text}"
|
||
|
|
if tick.delegation:
|
||
|
|
line += f" (-> {tick.delegation.get('question', '')})"
|
||
|
|
print(line, flush=True)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|