Update app.py

app.py

@@ -137,6 +137,16 @@ def together_text_to_image(prompt: str):
     except Exception as e:
         return f"Error generating image from text: {e}"
 
+def image_to_base64(image_path):
+    try:
+        with open(image_path, "rb") as image_file:
+            base64_string = base64.b64encode(image_file.read()).decode('utf-8')
+            return base64_string
+    except FileNotFoundError:
+        return "Error: File not found. Please check the image path."
+    except Exception as e:
+        return f"An error occurred: {e}"
+
 def together_image_to_image(image_numpy, prompt: str):
     """
     Transforms an image based on a text prompt using the Together AI API.
@@ -161,7 +171,7 @@ def together_image_to_image(image_numpy, prompt: str):
             model="black-forest-labs/FLUX.1-kontext-max", # Hardcoded model as requested
             steps=40, # Hardcoded steps as requested
             prompt=prompt,
-            image_url=image_numpy
+            image_base64=image_to_base64(image_numpy)
         )
         return image_completion.data[0].url
     except Exception as e: