Fixed sme things and added dalle option

code/imagesGeneration.py

@@ -1,7 +1,16 @@
 import requests
 import os
-async def generate(prompt,path):
-    r = requests.get(f"http://localhost:8000/generate_image?prompt={prompt}&path={path}")
-    return r.json()
-#print the current working directory
-print(os.getcwd())
+import openai
+async def generate(prompt, path, mode, openai_key):
+    #r = requests.get(f"http://localhost:8000/generate_image?prompt={prompt}&path={path}")
+    if mode == "sd": 
+        r = requests.get(f"https://localhost:8000/generate_image?prompt={prompt}&path={path}")
+        return "image generated"
+    if mode == "dalle":
+        openai.api_key = openai_key
+        img = await openai.Image.acreate(
+                prompt=prompt,
+                n=1,
+                size="1024x1024",
+            )
+        return img