Update app/helperCodes/common_functions.py

app/helperCodes/common_functions.py

@@ -117,20 +117,28 @@ def generate_image(prompt: str) -> Image.Image:
     and applies image properties to the resulting image.
     """
     replicate_client = Client(api_token=os.getenv("REPLICATE_KEY"))
+    # inputs = {
+    #     "prompt": f"{prompt}.",
+    #     "go_fast": True,
+    #     "guidance": 3,
+    #     "megapixels": "1",
+    #     "num_outputs": 1,
+    #     "aspect_ratio": "1:1",
+    #     "output_format": "jpg",
+    #     "output_quality": 50,
+    #     "prompt_strength": 0.8,
+    #     "num_inference_steps": 28
+    # }
+    # output = replicate_client.run("black-forest-labs/flux-dev", input=inputs)
+    # image_data = output[0].read()
+
     inputs = {
         "prompt": f"{prompt}.",
-        "go_fast": True,
-        "guidance": 3,
-        "megapixels": "1",
-        "num_outputs": 1,
         "aspect_ratio": "1:1",
-        "output_format": "jpg",
-        "output_quality": 50,
-        "prompt_strength": 0.8,
-        "num_inference_steps": 28
+        "magic_prompt_option": "Off"
     }
-    output = replicate_client.run("black-forest-labs/flux-dev", input=inputs)
-    image_data = output[0].read()
+    output = replicate_client.run("ideogram-ai/ideogram-v2", input=inputs)
+    image_data = output.read()
 
     image = Image.open(io.BytesIO(image_data))
     image = add_random_border(image)