meshy generation

api_keys.py

@@ -1,2 +1,2 @@
-meshy_api_key = ""
+meshy_api_key = "msy_xLyNkJ7taYmF4g3KJ3w08W02lUVdcO1ZT0QK"
 gpt_api_key = "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"

app.py

@@ -4,6 +4,7 @@ import openai
 from prompt import setup_prompt
 from api_keys import gpt_api_key
 import utils
+import os
 
 openai.api_key = gpt_api_key 
 messages = []
@@ -19,7 +20,7 @@ reply = response["choices"][0]["message"]["content"]
 messages.append({"role": "assistant", "content": reply})
 print("\n" + reply + "\n")
 
-final_prompt = ""
+meshy_prompt = ""
 
 model_path =  "../house_light/model.glb"
 
@@ -28,13 +29,20 @@ cnt = 0
 
 def solve():
   global model_path
-  time.sleep(3)
-  return model_path
+  # time.sleep(3)
+  extracted_text = utils.extract_text_surrounded_by_backticks(meshy_prompt)
+  begin = meshy_prompt.find("{")
+  end = meshy_prompt.find("}")
+  print(meshy_prompt[begin:end+1])
+  payload = utils.text_to_3d_gen(meshy_prompt[begin:end+1])
+  taskid = utils.create_meshy_object(payload)
+  utils.download_model(taskid)
+  return os.path.join(os.path.dirname(os.path.abspath(__file__)), f"{taskid}.glb")
 
 def slow_echo(message, history):
   global cnt
   global messages
-  global final_prompt
+  global meshy_prompt
   global model_path
   messages.append({"role": "user", "content": message})
   response = openai.ChatCompletion.create(
@@ -45,13 +53,13 @@ def slow_echo(message, history):
   print("\n" + reply + "\n")
   if reply[len(reply)-1] == '1' or reply[len(reply)-2] == '1':
       print("start generating")
-      final_prompt = reply
+      meshy_prompt = reply
       reply = "Generating..."
-      cnt += 1
-      if cnt == 1 :
-        model_path = "../house_light/model.glb"
-      elif cnt == 2 :
-        model_path = "../house_dark/model.glb"
+      # cnt += 1
+      # if cnt == 1 :
+      #   model_path = "../house_light/model.glb"
+      # elif cnt == 2 :
+      #   model_path = "../house_dark/model.glb"
 
   for i in range(len(reply)):
     time.sleep(0.02)

prompt.py

@@ -31,7 +31,7 @@ Second, you should guide users step by step in the following procedure to help u
 6 - Confirmation: Show user the prompt you generated 
 - Style: <style> \n - Negative Prompt: <negative prompt> \n - Ary Style: <art style> \n - Texture Resolution: <resolution> \n <0>}. Note: everything in <> should be keywords, not a complete sentence or verbs. After showing, ask user if he/she wants to add more things
 7 - Output: Show the final prompt in JSON with the following keys:
-object_prompt, style_prompt, negative_prompt, art_style. Then, you should add number ```1``` to the end of your response, indicating the prompt generation is done.
+object_prompt, style_prompt, negative_prompt, art_style. Then, you must add number ```1``` to the end of your output response(not in json), indicating the prompt generation is done.
     - ```art_style``` key words to code conversion:
         - Realistic style -> realistic
         - 2.5D Cartoon style -> fake-3d-cartoon