Update prompt/prompt_services.py

prompt/prompt_services.py

@@ -3,15 +3,18 @@ import openai, os, json, re
 openai.api_key = os.getenv("OPENAI_API_KEY")
 sys_prompt = os.getenv("SYS_PROMPT")
 
-def get_prompts(image_base64, category, user_prompt, sentiment, negative_prompt):
+def get_prompts(image_base64, category, user_prompt, sentiment, negative_prompt, num_variations):
     try:
+        prompt_examples = [f"Prompt {i + 1}" for i in range(num_variations)]
+        json_format_example = json.dumps({"variations": prompt_examples})
+
         if negative_prompt:
             message = [
                 {
                     "role": "system",
                     "content": f"""{sys_prompt}
-                            Return only a JSON with 2 variations of ad prompts for image generation, based on the input image.
-                            Respond in this JSON format:\n{{\"variations\": [\"Prompt 1\", \"Prompt 2\", ..., \"Prompt 10\"]}}"""
+                            Return only a JSON with {num_variations} variations of ad prompts for image generation, based on the input image.
+                            Respond in this JSON format:\n{json_format_example}"""
                 },
                 {
                     "role": "user",
@@ -29,8 +32,8 @@ def get_prompts(image_base64, category, user_prompt, sentiment, negative_prompt)
                 {
                     "role": "system",
                     "content": f"""{sys_prompt}
-                                        Return only a JSON with 2 variations of ad prompts for image generation, based on the input image.
-                                        Respond in this JSON format:\n{{\"variations\": [\"Prompt 1\", \"Prompt 2\", ..., \"Prompt 10\"]}}"""
+                                        Return only a JSON with {num_variations} variations of ad prompts for image generation, based on the input image.
+                                        Respond in this JSON format:\n{json_format_example}"""
                 },
                 {
                     "role": "user",
@@ -50,6 +53,7 @@ def get_prompts(image_base64, category, user_prompt, sentiment, negative_prompt)
         )
         content = response.choices[0].message.content.strip()
         content = re.sub(r"^```json\s*|\s*```$", "", content)
+        print(f"Prompt variations response: {content}")
         return json.loads(content)["variations"]
     except Exception:
-        return []
+        return []