Update app.py

app.py

@@ -9,8 +9,7 @@ from together import Together
 TOGETHER_API_KEY = os.environ.get('TOGETHER_API_KEY')
 
 # --- Together API Client Configuration ---
-# Initialize the Together client only if the API key is available
-client = Together(api_key=TOGETHER_API_KEY) if TOGETHER_API_KEY else None
+client = Together(api_key=TOGETHER_API_KEY)
 
 # --- Pixabay API Configuration ---
 PIXABAY_API_KEY = os.environ.get('PIXABAY_API_KEY')
@@ -18,21 +17,50 @@ IMAGE_API_URL = 'https://pixabay.com/api/'
 VIDEO_API_URL = 'https://pixabay.com/api/videos/'
 PER_PAGE = 5
 
-# --- Function to search Pixabay ---
-def search_pixabay(query: str, media_type: str, image_type: str, orientation: str, video_type: str):
+def image_to_url(image_path):
+    """
+    Uploads an image to a temporary hosting service and returns its URL.
+
+    Args:
+        image_path (str): The path to the image file to be uploaded.
+
+    Returns:
+        str: The URL of the uploaded image, or an error message if the upload fails.
+    """
+    try:
+        url = 'https://uguu.se/upload'
+
+        with open(image_path, 'rb') as f:
+            files = {'files[]': (image_path, f)}
+            response = requests.post(url, files=files)
+            response_json = response.json()
+            return response_json['files'][0]['url']
+    except FileNotFoundError:
+        return "Error: File not found. Please check the image path."
+    except Exception as e:
+        return f"An error occurred: {e}"
+
+def search_pixabay(query: str = "", media_type: str = "Image", image_type: str = "all",
+                  orientation: str = "all", video_type: str = "all"):
     """
-    Searches the Pixabay API for royalty-free stock images or videos based on user query and filters.
+    Searches the Pixabay API for royalty-free stock images or videos based on user's query and filters.
 
     Args:
-        query (str): The search term for finding media. If empty, an error is returned.
+        query (str): The search term for finding stock media.
         media_type (str): Specifies the type of media to search for.
-                         Accepted values are "Image" or "Video".
+                         Accepted values are "Image" or "Video". Defaults to "Image".
         image_type (str): Filter results by image type (used only if media_type is "Image").
-                          Accepted values: "all", "photo", "illustration", "vector".
+                          Accepted values: "all", "photo", "illustration", "vector". Defaults to "all".
         orientation (str): Filter results by image orientation (used only if media_type is "Image").
-                           Accepted values: "all", "horizontal", "vertical".
+                           Accepted values: "all", "horizontal", "vertical". Defaults to "all".
         video_type (str): Filter results by video type (used only if media_type is "Video").
-                          Accepted values: "all", "film", "animation".
+                          Accepted values: "all", "film", "animation". Defaults to "all".
+
+    Returns:
+        tuple: Contains three elements:
+            - str: URL of the found image (or None if not found or media_type is "Video")
+            - str: URL of the found video (or None if not found or media_type is "Image")
+            - str: Status message or error message
     """
     if not query:
         return None, None, "Please enter a search query."
@@ -56,7 +84,6 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
         api_url = VIDEO_API_URL
         params['video_type'] = video_type
     else:
-        # This case should not be reachable with the Gradio Radio component
         return None, None, "Invalid media type selected."
 
     try:
@@ -76,7 +103,6 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
         if media_type == "Image":
             image_url = selected_hit.get('largeImageURL')
             if image_url:
-                # Return the image URL, None for video, and an empty status message
                 return image_url, None, ""
             else:
                 return None, None, "Could not retrieve large image URL."
@@ -87,10 +113,8 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
             video_url = large_video.get('url')
 
             if video_url:
-                # Return None for image, the video URL, and an empty status message
                 return None, video_url, ""
             else:
-                 # Fallback to medium quality if large is not available
                  medium_video = video_urls.get('medium', {})
                  video_url = medium_video.get('url')
                  if video_url:
@@ -103,20 +127,18 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
     except json.JSONDecodeError:
          return None, None, "Error decoding API response."
     except Exception as e:
-        # Catch any other unexpected errors
         return None, None, f"An unexpected error occurred: {e}"
 
-# --- Together AI Image Generation Functions ---
 def together_text_to_image(prompt: str = "", width: int = 1024, height: int = 1024):
     """
     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 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.
+        prompt (str): The text prompt to generate the image from.
+        width (int): The width of the generated image in pixels.
+            Must be between 512 and 1440. Defaults to 1024.
+        height (int): The height of the generated image in pixels.
+            Must be between 512 and 1440. Defaults to 1024.
 
     Returns:
         str: The URL of the generated image if successful, or an error message if not.
@@ -131,7 +153,6 @@ def together_text_to_image(prompt: str = "", width: int = 1024, height: int = 10
     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
@@ -147,13 +168,12 @@ def together_text_to_image(prompt: str = "", width: int = 1024, height: int = 10
                 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
+            model="black-forest-labs/FLUX.1.1-pro",
             width=width,
             height=height,
             steps=40,
@@ -163,27 +183,12 @@ def together_text_to_image(prompt: str = "", width: int = 1024, height: int = 10
     except Exception as e:
         return f"Error generating image from text: {e}"
 
-
-def image_to_url(image_path):
-    try:
-        url = 'https://uguu.se/upload'
-    
-        with open(image_path, 'rb') as f:
-            files = {'files[]': (image_path, f)}
-            response = requests.post(url, files=files)
-            response_json = response.json()
-            return response_json['files'][0]['url']
-    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):
+def together_image_to_image(image_path: str = None, prompt: str = ""):
     """
     Transforms an image based on a text prompt using the Together AI API.
 
     Args:
-        image_numpy (numpy.ndarray): The input image as a NumPy array (provided by Gradio).
+        image_path (str): The path to the input image file.
         prompt (str): The text prompt for image transformation.
 
     Returns:
@@ -191,26 +196,26 @@ def together_image_to_image(image_numpy, prompt: str):
     """
     if not client:
         return "Together AI client not initialized. Please set the TOGETHER_API_KEY environment variable."
-    if image_numpy is None:
+    if image_path is None:
         return "Please upload or paste an image for image-to-image transformation."
     if not prompt:
         return "Please enter a prompt for image transformation."
 
     try:
-
-        print(image_numpy)
+        image_url = image_to_url(image_path)
+        if isinstance(image_url, str) and image_url.startswith("Error"):
+            return image_url
 
         image_completion = client.images.generate(
-            model="black-forest-labs/FLUX.1-kontext-max", # Hardcoded model as requested
-            steps=40, # Hardcoded steps as requested
+            model="black-forest-labs/FLUX.1-kontext-max",
+            steps=40,
             prompt=prompt,
-            image_url=image_to_url(image_numpy)
+            image_url=image_url
         )
         return image_completion.data[0].url
     except Exception as e:
         return f"Error transforming image: {e}"
 
-
 # --- Gradio Blocks Interface Definition ---
 with gr.Blocks(title="Media Generation and Search Explorer") as demo:
     gr.Markdown("## Media Generation and Search Explorer")
@@ -237,25 +242,27 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
             pixabay_image_output = gr.Image(label="Result Image (URL)", interactive=False)
             pixabay_video_output = gr.Video(label="Result Video (URL)", interactive=False)
 
-        # Logic to toggle visibility of input columns and output components based on media type selection
         def update_pixabay_inputs_blocks(media_type):
+            """Updates the visibility of input columns based on selected media type."""
             if media_type == "Image":
-                return gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+                return (gr.update(visible=True), gr.update(visible=False),
+                        gr.update(visible=True), gr.update(visible=False))
             elif media_type == "Video":
-                return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=True)
+                return (gr.update(visible=False), gr.update(visible=True),
+                        gr.update(visible=False), gr.update(visible=True))
             else:
-                return gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
+                return (gr.update(visible=True), gr.update(visible=False),
+                        gr.update(visible=True), gr.update(visible=False))
 
-        # Trigger the update_pixabay_inputs_blocks function when media_type_radio changes
         pixabay_media_type_radio.change(
             fn=update_pixabay_inputs_blocks,
             inputs=pixabay_media_type_radio,
-            outputs=[pixabay_image_options_col, pixabay_video_options_col, pixabay_image_output, pixabay_video_output],
+            outputs=[pixabay_image_options_col, pixabay_video_options_col,
+                    pixabay_image_output, pixabay_video_output],
             api_name=False,
             show_api=False
         )
 
-        # Trigger the search_pixabay function when the search button is clicked
         pixabay_search_button.click(
             fn=search_pixabay,
             inputs=[