From 8e918def5808e1beed34d88388b44c07ff486fed Mon Sep 17 00:00:00 2001
From: fles <fles@qq.com>
Date: Mon, 8 Jan 2024 00:45:50 +0800
Subject: [PATCH] add prompt

---
 turbo/api.py     | 2 ++
 turbo/index.html | 1 +
 turbo/main.js    | 4 ++--
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/turbo/api.py b/turbo/api.py
index bbaeac4..46f1b4f 100644
--- 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
index ddedf39..b6164c9 100644
--- 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
index 43fa0aa..8daeaa3 100644
--- 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]];
         }