GitLab CI commited on
Commit
7f7ae8a
·
1 Parent(s): b0b3162

Update game build from GitLab CI

Browse files
server/ActionProcessor.py CHANGED
@@ -1,9 +1,8 @@
1
  import datetime
 
2
  from threading import Thread
3
  from multiprocessing import Queue
4
  from typing import Dict, Any, List
5
- import json
6
- import re
7
  import logging
8
  import sys
9
  from mistralai import Mistral
@@ -34,7 +33,7 @@ class ActionProcessor(Thread):
34
  def __init__(
35
  self,
36
  text_queue: "Queue[str]",
37
- action_queue: "Queue[str]",
38
  mistral_api_key: str,
39
  ):
40
  super().__init__()
@@ -50,22 +49,19 @@ class ActionProcessor(Thread):
50
  {
51
  "role": "system",
52
  "content": """
53
- You are an AI Agent in a video game. You're listening to a parent giving their baby instructions, translate them into game functions. The following game functions are:
54
-
55
- - DropBleach: Ask the baby to drop the bleach (or 'Javel').
56
- - DropSyringe: Ask the baby to drop the syringe.
57
- - DropFork: Ask the baby to drop the fork.
58
- - GoToLivingRoom: Ask the baby to go to the living room.
59
- - GoToKitchen: Ask the baby to go to the kitchen.
60
- - GoToBedroom: Ask the baby to go to the bedroom.
61
- - StopCrying: Ask the baby to stop crying.
62
- - Come: Ask the baby to come towards the camera, or follow the sound of the voice.
63
- - None: The baby does not need to take any action.
64
-
65
- You must also determine whether the parenting sentiment behind the command is positive or negative, based on the following criteria:
66
- - If the statement includes the baby’s name ("Harold"), it is considered positive parenting.
67
- - If the statement consists only of direct commands without affectionate or encouraging words (e.g., "good boy," "better"), it is considered negative parenting.
68
- Your response should strictly follow this JSON format as an array of strings:
69
 
70
  ```json
71
  [action,sentiment]
@@ -77,7 +73,10 @@ Output: ["DropFork", "negative"]
77
 
78
  Input: "Harold, please don't drink the bleach!"
79
  Output: ["DropBleach", "positive"]
80
- """,
 
 
 
81
  },
82
  {
83
  "role": "user",
@@ -91,7 +90,7 @@ Output: ["DropBleach", "positive"]
91
  + [
92
  {
93
  "role": "assistant",
94
- "content": "[",
95
  "prefix": True,
96
  }
97
  ],
@@ -117,8 +116,14 @@ Output: ["DropBleach", "positive"]
117
  candidate = self.text_buffers[1]
118
 
119
  if len(self.text_buffers[0]) < len(candidate) >= len(self.text_buffers[2]):
120
- action_and_sentiment = self.get_action_and_sentiment(candidate)
121
- action, sentiment = action_and_sentiment.split(",")
 
 
 
 
 
 
122
 
123
  if action not in self.valid_action:
124
  action = "None"
@@ -143,7 +148,7 @@ Output: ["DropBleach", "positive"]
143
 
144
  # If we got a valid action, add it to the action queue
145
  if action:
146
- self.action_queue.put(json.dumps(action))
147
 
148
  except Exception as e:
149
  logger.error(f"Error processing text: {str(e)}")
 
1
  import datetime
2
+ import json
3
  from threading import Thread
4
  from multiprocessing import Queue
5
  from typing import Dict, Any, List
 
 
6
  import logging
7
  import sys
8
  from mistralai import Mistral
 
33
  def __init__(
34
  self,
35
  text_queue: "Queue[str]",
36
+ action_queue: "Queue[Dict[str, Any]]",
37
  mistral_api_key: str,
38
  ):
39
  super().__init__()
 
49
  {
50
  "role": "system",
51
  "content": """
52
+ You are a transcription expert. You're listening to a parent speaking to a baby. Your goal
53
+ is to determine what the baby is asked to do and what the parent's sentiment is.
54
+
55
+ The following interpretations are possible:
56
+ - DropBleach: The parent asks to drop the bleach (or 'Javel').
57
+ - DropSyringe: The parent asks to drop the syringe.
58
+ - DropFork: The parent asks to drop the fork.
59
+ - GoToLivingRoom: The parent asks to go to the living room.
60
+ - GoToKitchen: The parent asks to go to the kitchen.
61
+ - GoToBedroom: The parent asks to go to the bedroom.
62
+ - StopCrying: The parent asks to stop crying.
63
+ - Come: The parent asks to come.
64
+ - None: Others instructions are not relevant.
 
 
 
65
 
66
  ```json
67
  [action,sentiment]
 
73
 
74
  Input: "Harold, please don't drink the bleach!"
75
  Output: ["DropBleach", "positive"]
76
+
77
+ Input: "I'm so tired of this."
78
+ Output: ["None", "negative"]
79
+ """,
80
  },
81
  {
82
  "role": "user",
 
90
  + [
91
  {
92
  "role": "assistant",
93
+ "content": '["',
94
  "prefix": True,
95
  }
96
  ],
 
116
  candidate = self.text_buffers[1]
117
 
118
  if len(self.text_buffers[0]) < len(candidate) >= len(self.text_buffers[2]):
119
+ action_and_sentiment = json.loads(self.get_action_and_sentiment(candidate))
120
+ if (
121
+ not isinstance(action_and_sentiment, list)
122
+ or len(action_and_sentiment) != 2
123
+ ):
124
+ return None
125
+
126
+ action, sentiment = action_and_sentiment
127
 
128
  if action not in self.valid_action:
129
  action = "None"
 
148
 
149
  # If we got a valid action, add it to the action queue
150
  if action:
151
+ self.action_queue.put(action)
152
 
153
  except Exception as e:
154
  logger.error(f"Error processing text: {str(e)}")
server/__main__.py CHANGED
@@ -32,7 +32,7 @@ STATIC_DIR = (
32
 
33
  audio_queue: "Queue[io.BytesIO]" = Queue()
34
  text_queue: "Queue[str]" = Queue()
35
- action_queue: "Queue[str]" = Queue()
36
 
37
  app = Flask(__name__, static_folder=STATIC_DIR)
38
 
 
32
 
33
  audio_queue: "Queue[io.BytesIO]" = Queue()
34
  text_queue: "Queue[str]" = Queue()
35
+ action_queue: "Queue[Dict[str, Any]]" = Queue()
36
 
37
  app = Flask(__name__, static_folder=STATIC_DIR)
38
 
server/static/godot/index.html CHANGED
@@ -97,7 +97,7 @@ body {
97
 
98
  <script src="index.js"></script>
99
  <script>
100
- const GODOT_CONFIG = {"args":[],"canvasResizePolicy":2,"ensureCrossOriginIsolationHeaders":false,"executable":"index","experimentalVK":false,"fileSizes":{"index.pck":870464,"index.wasm":35376909},"focusCanvas":true,"gdextensionLibs":[]};
101
  const GODOT_THREADS_ENABLED = false;
102
  const engine = new Engine(GODOT_CONFIG);
103
 
 
97
 
98
  <script src="index.js"></script>
99
  <script>
100
+ const GODOT_CONFIG = {"args":[],"canvasResizePolicy":2,"ensureCrossOriginIsolationHeaders":false,"executable":"index","experimentalVK":false,"fileSizes":{"index.pck":870496,"index.wasm":35376909},"focusCanvas":true,"gdextensionLibs":[]};
101
  const GODOT_THREADS_ENABLED = false;
102
  const engine = new Engine(GODOT_CONFIG);
103
 
server/static/godot/index.pck CHANGED
Binary files a/server/static/godot/index.pck and b/server/static/godot/index.pck differ