laxsvips commited on
Commit
148bfde
·
1 Parent(s): 4251885

Updates for generating prompt for Leo

Browse files
Files changed (1) hide show
  1. chat.py +14 -0
chat.py CHANGED
@@ -160,5 +160,19 @@ def initialize_game(game_id, user_id, user_input):
160
  save_file(fname, llm_prompt_op.choices[0]["message"]["content"])
161
  return llm_prompt_op.choices[0]["message"]["content"]
162
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
163
  if __name__ == '__main__':
164
  print("main")
 
160
  save_file(fname, llm_prompt_op.choices[0]["message"]["content"])
161
  return llm_prompt_op.choices[0]["message"]["content"]
162
 
163
+ def generate_image_prompt(game_id, user_id, user_input):
164
+ payload = list()
165
+ file_data = open_file(f"{file_path}/image_prompt_leo.txt").replace("<<PROMPT_FOR_IMG>>", user_input)
166
+ leo_input_msg = call_gpt(file_data)
167
+ leo_input = leo_input_msg.choices[0].message.content
168
+ timestamp_op = time()
169
+ timestring_op = timestamp_to_datetime(timestamp_op)
170
+ vector_op = perform_embedding(leo_input)
171
+ unique_id_op = str(uuid4)
172
+ metadata_op = {'speaker': 'BOT4LEO', 'user_id': user_id, 'game_id': game_id, 'timestring': timestring_op, 'text': leo_input}
173
+ payload.append((unique_id_op, vector_op, metadata_op))
174
+
175
+ return leo_input
176
+
177
  if __name__ == '__main__':
178
  print("main")