fixing flux bug (#442)

* fixing flux bug

* added specific replicate version

apps/flux/app.py

@@ -1,20 +1,19 @@
-import os
-import base64
 import shutil
-import requests
 import replicate
 
 from PIL import Image
 from io import BytesIO
 
-
 def flux_image(prompt, filename):
-  output = replicate.run("black-forest-labs/flux-dev", input={"prompt": prompt})
+  try:
+    output = replicate.run("black-forest-labs/flux-dev", input={"prompt": prompt})
+  except Exception as e:
+    print("Couldn't generate that image. Please try a different prompt.")
+    print(f"Error: {e}")
 
-  header, encoded = output[0].url.split(',', 1)
+  encoded_image = output[0].read()
+  image = Image.open(BytesIO(encoded_image))
 
-  data = base64.b64decode(encoded)
-  image = Image.open(BytesIO(data))
   image.save(filename, format="JPEG")
 
   shutil.copy(filename, f".storage/{filename}")

apps/flux/requirements.txt

@@ -1 +1 @@
-replicate
+replicate==1.0.0