92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
import os
|
|
import requests
|
|
import subprocess
|
|
import time
|
|
|
|
from flask import Flask, jsonify, request
|
|
from flask_cors import CORS
|
|
from auth import get_meeting_token
|
|
|
|
app = Flask(__name__)
|
|
CORS(app)
|
|
|
|
|
|
def start_bot(bot_path, args=None):
|
|
daily_api_key = os.getenv("DAILY_API_KEY")
|
|
api_path = os.getenv("DAILY_API_PATH") or "https://api.daily.co/v1"
|
|
|
|
timeout = int(os.getenv("ROOM_TIMEOUT") or os.getenv("BOT_MAX_DURATION") or 300)
|
|
exp = time.time() + timeout
|
|
res = requests.post(
|
|
f"{api_path}/rooms",
|
|
headers={"Authorization": f"Bearer {daily_api_key}"},
|
|
json={
|
|
"properties": {
|
|
"exp": exp,
|
|
"enable_chat": True,
|
|
"enable_emoji_reactions": True,
|
|
"eject_at_room_exp": True,
|
|
"enable_prejoin_ui": False,
|
|
}
|
|
},
|
|
)
|
|
if res.status_code != 200:
|
|
return (
|
|
jsonify(
|
|
{
|
|
"error": "Unable to create room",
|
|
"status_code": res.status_code,
|
|
"text": res.text,
|
|
}
|
|
),
|
|
500,
|
|
)
|
|
room_url = res.json()["url"]
|
|
room_name = res.json()["name"]
|
|
|
|
meeting_token = get_meeting_token(room_name, daily_api_key, exp)
|
|
|
|
if args:
|
|
extra_args = " ".join([f'-{x[0]} "{x[1]}"' for x in args])
|
|
else:
|
|
extra_args = ""
|
|
|
|
otel_path = "opentelemetry-instrument" if os.getenv("USE_OTEL") else ""
|
|
print("using otel path: ", otel_path, os.getenv("USE_OTEL"))
|
|
proc = subprocess.Popen(
|
|
[
|
|
f"{otel_path} python {bot_path} -u {room_url} -t {meeting_token} -k {daily_api_key} {extra_args}"
|
|
],
|
|
shell=True,
|
|
bufsize=1,
|
|
)
|
|
|
|
# Don't return until the bot has joined the room, but wait for at most 2 seconds.
|
|
attempts = 0
|
|
while attempts < 20:
|
|
time.sleep(0.1)
|
|
attempts += 1
|
|
res = requests.get(
|
|
f"{api_path}/rooms/{room_name}/get-session-data",
|
|
headers={"Authorization": f"Bearer {daily_api_key}"},
|
|
)
|
|
if res.status_code == 200:
|
|
break
|
|
print(f"Took {attempts} attempts to join room {room_name}")
|
|
|
|
# Additional client config
|
|
config = {}
|
|
if os.getenv("CLIENT_VAD_TIMEOUT_SEC"):
|
|
config['vad_timeout_sec'] = float(os.getenv("CLIENT_VAD_TIMEOUT_SEC"))
|
|
else:
|
|
config['vad_timeout_sec'] = 1.5
|
|
|
|
return jsonify({"room_url": room_url, "token": meeting_token, "config": config}), 200
|
|
|
|
|
|
@app.route("/spin-up-kitty", methods=["POST"])
|
|
def spin_up_kitty():
|
|
return start_bot("./src/samples/foundational/06a-golden-kitty.py")
|
|
|
|
|