radames commited on
Commit
3060840
β€’
1 Parent(s): bf7c37f
Files changed (1) hide show
  1. frontend/src/routes/+page.svelte +11 -3
frontend/src/routes/+page.svelte CHANGED
@@ -82,13 +82,18 @@
82
  if (!_prompt || $isLoading == true) return;
83
  $loadingState = 'Pending';
84
  $isLoading = true;
 
85
  const sessionHash = crypto.randomUUID();
86
 
87
- const payload = {
88
  fn_index: 2,
89
- data: [_prompt],
90
  session_hash: sessionHash
91
  };
 
 
 
 
 
92
  const websocket = new WebSocket(PUBLIC_WS_ENDPOINT);
93
  // websocket.onopen = async function (event) {
94
  // websocket.send(JSON.stringify({ hash: sessionHash }));
@@ -104,9 +109,12 @@
104
  const data = JSON.parse(event.data);
105
  $loadingState = '';
106
  switch (data.msg) {
 
 
 
107
  case 'send_data':
108
  $loadingState = 'Sending Data';
109
- websocket.send(JSON.stringify(payload));
110
  break;
111
  case 'queue_full':
112
  $loadingState = 'Queue full';
 
82
  if (!_prompt || $isLoading == true) return;
83
  $loadingState = 'Pending';
84
  $isLoading = true;
85
+
86
  const sessionHash = crypto.randomUUID();
87
 
88
+ const hashpayload = {
89
  fn_index: 2,
 
90
  session_hash: sessionHash
91
  };
92
+
93
+ const datapayload = {
94
+ data: [_prompt]
95
+ };
96
+
97
  const websocket = new WebSocket(PUBLIC_WS_ENDPOINT);
98
  // websocket.onopen = async function (event) {
99
  // websocket.send(JSON.stringify({ hash: sessionHash }));
 
109
  const data = JSON.parse(event.data);
110
  $loadingState = '';
111
  switch (data.msg) {
112
+ case 'send_hash':
113
+ websocket.send(JSON.stringify(hashpayload));
114
+ break;
115
  case 'send_data':
116
  $loadingState = 'Sending Data';
117
+ websocket.send(JSON.stringify({ ...hashpayload, ...datapayload }));
118
  break;
119
  case 'queue_full':
120
  $loadingState = 'Queue full';