implement db

src/image_gen.py

@@ -45,3 +45,8 @@ def icon_gen(prompt: str, quality: Literal["hd", "standard"], output: Path = "./
     with open(image_filepath, "wb") as f:
         f.write(image_response.content)
     print(image_id)
+
+
+if __name__ == "__main__":
+    dog_emoji = image_style("dog in a hat on a beach with a drink", style=Styles.cartoon)
+    icon_gen(dog_emoji, output=Path("../output"), quality="hd")