Update app.py

app.py

@@ -110,26 +110,55 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
         return None, None, f"An unexpected error occurred: {e}"
 
 # --- Together AI Image Generation Functions ---
-def together_text_to_image(prompt: str):
+def together_text_to_image(prompt: str = "", width: int = 1024, height: int = 1024):
     """
-    Generates an image from text using the Together AI API.
+    Generates an image from a text prompt using the Together AI API and the FLUX.1.1-pro model.
 
     Args:
-        prompt (str): The text prompt for image generation.
+        prompt (str): The text prompt to generate the image from. This must be a non-empty string.
+        width (int, optional): The width of the generated image in pixels. 
+            Must be between 512 and 1440. Will auto-adjust if out of bounds. Default is 1024.
+        height (int, optional): The height of the generated image in pixels. 
+            Must be between 512 and 1440. Will auto-adjust if out of bounds. Default is 1024.
 
     Returns:
-        str: The URL of the generated image, or an error message.
+        str: The URL of the generated image if successful, or an error message if not.
     """
     if not client:
         return "Together AI client not initialized. Please set the TOGETHER_API_KEY environment variable."
     if not prompt:
         return "Please enter a prompt for text-to-image generation."
 
+    # Clamp dimensions while preserving aspect ratio
+    min_size, max_size = 512, 1440
+    if width < min_size or width > max_size or height < min_size or height > max_size:
+        aspect_ratio = width / height
+
+        # Adjust based on which dimension is more out of bounds
+        if width < min_size or height < min_size:
+            if width < height:
+                width = min_size
+                height = int(round(width / aspect_ratio))
+            else:
+                height = min_size
+                width = int(round(height * aspect_ratio))
+        elif width > max_size or height > max_size:
+            if width > height:
+                width = max_size
+                height = int(round(width / aspect_ratio))
+            else:
+                height = max_size
+                width = int(round(height * aspect_ratio))
+
+        # Re-clamp just in case rounding pushed a value out of range
+        width = max(min_size, min(width, max_size))
+        height = max(min_size, min(height, max_size))
+
     try:
         image_completion = client.images.generate(
-            model="black-forest-labs/FLUX.1.1-pro", # Hardcoded model as requested
-            width=1024,
-            height=768,
+            model="black-forest-labs/FLUX.1.1-pro",  # Hardcoded model
+            width=width,
+            height=height,
             steps=40,
             prompt=prompt,
         )
@@ -137,6 +166,7 @@ def together_text_to_image(prompt: str):
     except Exception as e:
         return f"Error generating image from text: {e}"
 
+
 def image_to_url(image_path):
     try:
         url = 'https://uguu.se/upload'
@@ -244,11 +274,13 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
         gr.Warning("This requires setting the TOGETHER_API_KEY environment variable.")
         with gr.Row():
             together_text_to_image_prompt = gr.Textbox(label="Enter your prompt", scale=2)
+            together_text_to_image_width = gr.Slider("Width", value=1024)
+            together_text_to_image_height = gr.Slider("Height", value=1024)
             together_text_to_image_button = gr.Button("Generate Image", scale=1)
         together_text_to_image_output = gr.Image(label="Generated Image (URL)", type="filepath", interactive=False)
 
         together_text_to_image_button.click(
-            fn=together_text_to_image,
+            fn=[together_text_to_image, together_text_to_image_width, together_text_to_image_height],
             inputs=together_text_to_image_prompt,
             outputs=together_text_to_image_output,
         )