Spaces:
Sleeping
Sleeping
Jon Taylor
commited on
Commit
•
f66623b
1
Parent(s):
98b5e62
moved buffer into loop
Browse files- app/bot.py +5 -5
- frontend/app/components/Call.js +2 -0
- frontend/app/components/CreateRoom.js +61 -55
- frontend/app/components/Joining.js +8 -6
app/bot.py
CHANGED
@@ -101,11 +101,6 @@ class DailyVision(EventHandler):
|
|
101 |
})
|
102 |
|
103 |
def process_frames(self):
|
104 |
-
if self.__params_buffer:
|
105 |
-
print("New buffer params found")
|
106 |
-
print(self.__params_buffer)
|
107 |
-
self.__params = self.__pipeline.InputParams(**self.__params_buffer)
|
108 |
-
|
109 |
params = self.__params
|
110 |
|
111 |
while not self.__app_quit:
|
@@ -123,6 +118,11 @@ class DailyVision(EventHandler):
|
|
123 |
#video_frame = self.__queue.get(timeout=5)
|
124 |
video_frame = self.__image_buffer
|
125 |
|
|
|
|
|
|
|
|
|
|
|
126 |
if not video_frame == None:
|
127 |
image = Image.frombytes("RGB", (video_frame.width, video_frame.height), video_frame.buffer)
|
128 |
result_image = self.__pipeline.predict(params, image)
|
|
|
101 |
})
|
102 |
|
103 |
def process_frames(self):
|
|
|
|
|
|
|
|
|
|
|
104 |
params = self.__params
|
105 |
|
106 |
while not self.__app_quit:
|
|
|
118 |
#video_frame = self.__queue.get(timeout=5)
|
119 |
video_frame = self.__image_buffer
|
120 |
|
121 |
+
if self.__params_buffer:
|
122 |
+
print("New buffer params found")
|
123 |
+
params = self.__pipeline.InputParams(**self.__params_buffer)
|
124 |
+
self.__params_buffer = None
|
125 |
+
|
126 |
if not video_frame == None:
|
127 |
image = Image.frombytes("RGB", (video_frame.width, video_frame.height), video_frame.buffer)
|
128 |
result_image = self.__pipeline.predict(params, image)
|
frontend/app/components/Call.js
CHANGED
@@ -14,6 +14,8 @@ const STATE_LEFT = "left";
|
|
14 |
const BOT_STATE_STARTING = "bot_starting";
|
15 |
const BOT_STATE_STARTED = "bot_started";
|
16 |
|
|
|
|
|
17 |
export default function Call() {
|
18 |
const daily = useDaily();
|
19 |
const [callState, setCallState] = useState(STATE_IDLE);
|
|
|
14 |
const BOT_STATE_STARTING = "bot_starting";
|
15 |
const BOT_STATE_STARTED = "bot_started";
|
16 |
|
17 |
+
//@TODO change to portrait constraints and update UI
|
18 |
+
//@TODO Disable simulcast
|
19 |
export default function Call() {
|
20 |
const daily = useDaily();
|
21 |
const [callState, setCallState] = useState(STATE_IDLE);
|
frontend/app/components/CreateRoom.js
CHANGED
@@ -32,65 +32,71 @@ export default function CreateRoom({ onCreateRoom }) {
|
|
32 |
}
|
33 |
|
34 |
return (
|
35 |
-
<
|
36 |
-
<div className="
|
37 |
-
<
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
<
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
|
|
|
|
|
|
|
|
|
|
53 |
|
54 |
-
|
55 |
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
-
|
67 |
-
|
68 |
|
69 |
-
|
70 |
-
|
71 |
-
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
83 |
</div>
|
84 |
-
|
85 |
-
|
86 |
-
onClick={() => onCreateRoom(roomUrl)}
|
87 |
-
className="mt-3 inline-flex w-full items-center justify-center rounded-md bg-indigo-600 px-3 py-2 text-sm font-semibold text-white shadow-sm hover:bg-indigo-500 focus-visible:outline focus-visible:outline-2 focus-visible:outline-offset-2 focus-visible:outline-indigo-600 sm:ml-3 sm:mt-0 sm:w-auto"
|
88 |
-
>
|
89 |
-
Join
|
90 |
-
</button>
|
91 |
-
</div>
|
92 |
-
)}
|
93 |
</div>
|
94 |
-
</
|
95 |
);
|
96 |
}
|
|
|
32 |
}
|
33 |
|
34 |
return (
|
35 |
+
<main className="min-h-screen flex items-center justify-center bg-zinc-100">
|
36 |
+
<div className="flex-1 bg-white shadow sm:rounded-lg max-w-xl mx-auto">
|
37 |
+
<div className="px-4 py-5 sm:p-6">
|
38 |
+
<h3 className="text-base font-semibold leading-6 text-gray-900">
|
39 |
+
Stable Diffusion + WebRTC Latency Demo
|
40 |
+
</h3>
|
41 |
+
<div className="mt-2 text-sm text-gray-500">
|
42 |
+
<p>
|
43 |
+
Demonstrates implementating WebRTC via daily.co to create
|
44 |
+
real-time vision inference. Currently implemented using SD 1.5
|
45 |
+
Turbo with Control Net.
|
46 |
+
</p>
|
47 |
+
</div>
|
48 |
+
<div className="mt-5">
|
49 |
+
<Button
|
50 |
+
rightSection={<IconArrowRight stroke={2} className="h-5 w-5" />}
|
51 |
+
size="md"
|
52 |
+
loading={fetching}
|
53 |
+
onClick={() => create()}
|
54 |
+
>
|
55 |
+
Join
|
56 |
+
</Button>
|
57 |
+
</div>
|
58 |
|
59 |
+
<hr className="my-6" />
|
60 |
|
61 |
+
<div
|
62 |
+
className="inline-flex items-center gap-1 text-sm text-gray-500 cursor-pointer"
|
63 |
+
onClick={() => setShowManual(!showManual)}
|
64 |
+
>
|
65 |
+
Join an existing room
|
66 |
+
<IconChevronRight
|
67 |
+
stoke={2}
|
68 |
+
className={`w-4 h-4 text-gray-300 transition-transform ${
|
69 |
+
showManual && "rotate-90"
|
70 |
+
}`}
|
71 |
+
/>
|
72 |
+
</div>
|
73 |
|
74 |
+
{showManual && (
|
75 |
+
<div className="mt-5 sm:flex sm:items-center">
|
76 |
+
<div className="w-full">
|
77 |
+
<label htmlFor="room" className="sr-only">
|
78 |
+
Daily Room URL
|
79 |
+
</label>
|
80 |
+
<input
|
81 |
+
type="text"
|
82 |
+
name="room"
|
83 |
+
id="room"
|
84 |
+
className="block w-full rounded-md border-0 py-1.5 text-gray-900 shadow-sm ring-1 ring-inset ring-gray-300 placeholder:text-gray-400 focus:ring-2 focus:ring-inset focus:ring-indigo-600 sm:text-sm sm:leading-6"
|
85 |
+
placeholder="https://..."
|
86 |
+
onChange={(e) => setRoomUrl(e.target.value)}
|
87 |
+
/>
|
88 |
+
</div>
|
89 |
+
<button
|
90 |
+
type="submit"
|
91 |
+
onClick={() => onCreateRoom(roomUrl)}
|
92 |
+
className="mt-3 inline-flex w-full items-center justify-center rounded-md bg-indigo-600 px-3 py-2 text-sm font-semibold text-white shadow-sm hover:bg-indigo-500 focus-visible:outline focus-visible:outline-2 focus-visible:outline-offset-2 focus-visible:outline-indigo-600 sm:ml-3 sm:mt-0 sm:w-auto"
|
93 |
+
>
|
94 |
+
Join
|
95 |
+
</button>
|
96 |
</div>
|
97 |
+
)}
|
98 |
+
</div>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
99 |
</div>
|
100 |
+
</main>
|
101 |
);
|
102 |
}
|
frontend/app/components/Joining.js
CHANGED
@@ -11,12 +11,14 @@ export default function Join({ roomUrl, onJoin }) {
|
|
11 |
});
|
12 |
|
13 |
return (
|
14 |
-
<
|
15 |
-
<div className="
|
16 |
-
<
|
17 |
-
|
18 |
-
|
|
|
|
|
19 |
</div>
|
20 |
-
</
|
21 |
);
|
22 |
}
|
|
|
11 |
});
|
12 |
|
13 |
return (
|
14 |
+
<main className="min-h-screen flex items-center justify-center bg-zinc-200">
|
15 |
+
<div className="bg-white shadow sm:rounded-lg max-w-xl mx-auto">
|
16 |
+
<div className="px-4 py-5 sm:p-6">
|
17 |
+
<h3 className="text-base font-semibold leading-6 text-gray-900">
|
18 |
+
Joining...
|
19 |
+
</h3>
|
20 |
+
</div>
|
21 |
</div>
|
22 |
+
</main>
|
23 |
);
|
24 |
}
|