add prompt

chaojie · Jan 7, 2024 · 8e918de · 8e918de
1 parent b0f9432
commit 8e918de
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/turbo/api.py b/turbo/api.py
@@ -130,6 +130,8 @@ def handle_message(message):
 @socketio.on('camera_poses')
 def handle_message(camera_poses):
     print(f'camera_poses:{request.sid} {camera_poses}')
+
+    prompt["60"]["inputs"]["prompt"] = camera_poses["prompt"]
     prompt["60"]["inputs"]["camera"] = camera_poses["camera_poses"]
     prompt["60"]["inputs"]["traj"] = camera_poses["trajs"]
     images = get_images(ws, prompt)

diff --git a/turbo/index.html b/turbo/index.html
@@ -13,6 +13,7 @@
         <div id="info">
 			<button id="btn_translate">Translate</button>
 			<button id="btn_rotate">Rotate</button>
+			<input id="txt_prompt" value="a rose swaying in the wind" />
 			<button id="btn_addpoint" style="display:none;">Add Point</button>
 			<!--"W" translate | "E" rotate--><br />
             <!--textarea id="tb_result" style="width:256px; height:128px;">[]</textarea><br/-->

diff --git a/turbo/main.js b/turbo/main.js
@@ -135,7 +135,7 @@ function detect_change(){
 
     if(JSON.stringify(cmatrix)!=JSON.stringify(matrix)){
         if(userDrawnPixels1024.length>=16){
-            socket.emit('camera_poses', {roomid:roomid,camera_poses:JSON.stringify(cposes),trajs:JSON.stringify(userDrawnPixels1024)});
+            socket.emit('camera_poses', {roomid:roomid,prompt:document.getElementById('txt_prompt').value,camera_poses:JSON.stringify(cposes),trajs:JSON.stringify(userDrawnPixels1024)});
             userDrawnPixels1024=[userDrawnPixels1024[userDrawnPixels1024.length-1]];
             cposes=[matrix];
         }else{
@@ -144,7 +144,7 @@ function detect_change(){
         cmatrix=matrix;
     }else{
         if(cposes.length>1||userDrawnPixels1024.length>=16){
-            socket.emit('camera_poses', {roomid:roomid,camera_poses:JSON.stringify(cposes),trajs:JSON.stringify(userDrawnPixels1024)});
+            socket.emit('camera_poses', {roomid:roomid,prompt:document.getElementById('txt_prompt').value,camera_poses:JSON.stringify(cposes),trajs:JSON.stringify(userDrawnPixels1024)});
             userDrawnPixels1024=[userDrawnPixels1024[userDrawnPixels1024.length-1]];
         }