updates

Author: parishwolfe

product_pipeline.py

@@ -9,7 +9,7 @@
 from util.ai_util import ai_util
 from util.image_util import create_text_image
 from res.models.tshirt import tshirt_from_ai_list
-from res.prompts.tshirt import user_message
+from res.prompts.tshirt import user_message, blueprint_6_description
 from util.github_util import GithubUploader
 
 from dotenv import load_dotenv
@@ -20,7 +20,7 @@
 # Set up argument parser
 parser = argparse.ArgumentParser(description="Utility to handle patterns and ideas.")
 parser.add_argument('-p', '--patterns', type=int, default=3, 
-                    help='Number of patterns, default is 10')
+                    help='Number of patterns, default is 3')
 parser.add_argument('idea', type=str, 
                     help='The Idea to generate patterns for')
 
@@ -48,10 +48,10 @@ def main():
     response = ai.chat(
         messages=[
             {"role": "system", "content": "You are a helpful chatbot"},
-            {"role": "user", "content": user_message % (number_of_patterns, idea)}
+            {"role": "user", "content": user_message % (number_of_patterns, idea) + blueprint_6_description},
         ],
         output_model=tshirt_from_ai_list,
-    ) #TODO - TEST TO MAKE SURE THIS WORKS using both f string and %s formatting
+    )
 
     # Parse the response
     parsed_response = json.loads(response)

res/prompts/__init__.py

@@ -1 +1 @@
-from .tshirt import general_product_description, user_message
+from .tshirt import user_message, blueprint_6_description

res/prompts/tshirt.py

@@ -1,10 +1,10 @@
 # General Product Description for blueprint 6
-general_product_description = "Unisex Gildan T-Shirt - Made with medium fabric (5.3 oz/yd² (180 g/m²)) consisting of 100% cotton for year-round comfort that is sustainable and highly durable. The classic fit of this shirt ensures a comfy, relaxed wear while the crew neckline adds that neat, timeless look that can blend into any occasion, casual or semi-formal."
+blueprint_6_description = """
+General Product Description:
+Unisex Gildan T-Shirt - Made with medium fabric (5.3 oz/yd² (180 g/m²)) consisting of 100% cotton for year-round comfort that is sustainable and highly durable. The classic fit of this shirt ensures a comfy, relaxed wear while the crew neckline adds that neat, timeless look that can blend into any occasion, casual or semi-formal.
+"""
 
 # AI prompt
-user_message = f"""
-I have a t-shirt business and I need your help to come up with creative, fun, unique, and interesting t-shirt designs. I would like you to generate %s patterns for %s. The title is the name of the pattern and it will be displayed as the product name. The description is a short description of the product, and the description of the item it will be printed on can be found below. You can use parts of this in the description, but keep it witty yet classy. The tshirt_text is the text that will be printed on the t-shirt. This field is your time to shine, I believe in your ability to help me create amazing products! The marketing_tags are the tags that will be used to market the t-shirt, these should be relevant to the idea and usable for marketing purposes. Please make sure that the patterns are unique and creative. Thank you!
-
-General Product Description:
-{general_product_description}
+user_message = """
+I have a t-shirt business and I need your help to come up with creative, fun, unique, and interesting t-shirt designs. I would like you to generate %s patterns relating tho this idea: %s. The title is the name of the pattern and it will be displayed as the product name. The description is a short description of the product, and the description of the item it will be printed on can be found below. You can use parts of this in the description, but keep it witty yet classy. The tshirt_text is the text that will be printed on the t-shirt. This field is your time to shine, I believe in your ability to help me create amazing products! The marketing_tags are the tags that will be used to market the t-shirt, these should be relevant to the idea and usable for marketing purposes. Please make sure that the patterns are unique and creative. Thank you!
 """