Update app.py

app.py

@@ -3,18 +3,29 @@ import requests
 import random
 import json
 import os
+from together import Together
+import base64
+from PIL import Image
+import io
+
+# --- Environment Variable for Together API Key ---
+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
 
 # --- Pixabay API Configuration ---
 PIXABAY_API_KEY = os.environ.get('PIXABAY_API_KEY')
 IMAGE_API_URL = 'https://pixabay.com/api/'
 VIDEO_API_URL = 'https://pixabay.com/api/videos/'
-PER_PAGE = 5 
+PER_PAGE = 5
 
 # --- Function to search Pixabay ---
 def search_pixabay(query: str, media_type: str, image_type: str, orientation: str, video_type: str):
     """
     Searches the Pixabay API for royalty-free stock images or videos based on user query and filters.
-    
+
     Args:
         query (str): The search term for finding media. If empty, an error is returned.
         media_type (str): Specifies the type of media to search for.
@@ -30,7 +41,7 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
         return None, None, "Please enter a search query."
 
     if not PIXABAY_API_KEY:
-        return None, None, "API Key not found. Please set the PIXABAY_API_KEY environment variable."
+        return None, None, "Pixabay API Key not found. Please set the PIXABAY_API_KEY environment variable."
 
     params = {
         'key': PIXABAY_API_KEY,
@@ -40,7 +51,6 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
         'safesearch': 'true'
     }
 
-    # Determine API endpoint and add media-specific parameters
     if media_type == "Image":
         api_url = IMAGE_API_URL
         params['image_type'] = image_type
@@ -54,8 +64,8 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
 
     try:
         response = requests.get(api_url, params=params)
-        response.raise_for_status() # Check for HTTP errors (like 404, 429, 500)
-        data = response.json() # Parse the JSON response
+        response.raise_for_status()
+        data = response.json()
 
         if data.get('totalHits', 0) == 0:
             return None, None, f"No results found for '{query}'."
@@ -64,12 +74,9 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
         if not hits:
              return None, None, f"No results found for '{query}'."
 
-        # Randomly select one result from the retrieved items (max 5)
         selected_hit = random.choice(hits)
 
-        # Extract the URL based on media type
         if media_type == "Image":
-            # Get the URL for the large image
             image_url = selected_hit.get('largeImageURL')
             if image_url:
                 # Return the image URL, None for video, and an empty status message
@@ -78,7 +85,6 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
                 return None, None, "Could not retrieve large image URL."
 
         elif media_type == "Video":
-            # Get the URL for the large video stream
             video_urls = selected_hit.get('videos', {})
             large_video = video_urls.get('large', {})
             video_url = large_video.get('url')
@@ -103,61 +109,155 @@ def search_pixabay(query: str, media_type: str, image_type: str, orientation: st
         # 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):
+    """
+    Generates an image from text using the Together AI API.
+
+    Args:
+        prompt (str): The text prompt for image generation.
+
+    Returns:
+        str: The URL of the generated image, or an error message.
+    """
+    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."
+
+    try:
+        image_completion = client.images.generate(
+            model="black-forest-labs/FLUX.1.1-pro", # Hardcoded model as requested
+            width=1024,
+            height=768,
+            steps=40, # Hardcoded steps as requested
+            prompt=prompt,
+        )
+        return image_completion.data[0].url
+    except Exception as e:
+        return f"Error generating image from text: {e}"
+
+def together_image_to_image(image_numpy, 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).
+        prompt (str): The text prompt for image transformation.
+
+    Returns:
+        str: The URL of the transformed image, or an error message.
+    """
+    if not client:
+        return "Together AI client not initialized. Please set the TOGETHER_API_KEY environment variable."
+    if image_numpy 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:
+        # Convert the NumPy array image to a PIL Image
+        img = Image.fromarray(image_numpy.astype('uint8'), 'RGB')
+
+        # Convert the PIL Image to base64
+        buffered = io.BytesIO()
+        img.save(buffered, format="PNG")
+        img_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+
+        image_completion = client.images.generate(
+            model="black-forest-labs/FLUX.1-kontext-max", # Hardcoded model as requested
+            steps=40, # Hardcoded steps as requested
+            prompt=prompt,
+            image_base64=img_base64
+        )
+        return image_completion.data[0].url
+    except Exception as e:
+        return f"Error transforming image: {e}"
+
+
 # --- Gradio Blocks Interface Definition ---
-with gr.Blocks(title="Pixabay Random Media Explorer") as demo:
-    gr.Markdown("## Pixabay Random Media Explorer (URL Output)")
-    gr.Markdown("Enter a search query, select media type and options, and get the URL for a random result from Pixabay.")
+with gr.Blocks(title="Media Generation and Search Explorer") as demo:
+    gr.Markdown("## Media Generation and Search Explorer")
+    gr.Markdown("Explore royalty-free media from Pixabay and generate/transform images using Together AI.")
 
-    with gr.Row():
-        query_input = gr.Textbox(label="Search Query", placeholder="e.g., yellow flowers", scale=2)
-        media_type_radio = gr.Radio(["Image", "Video"], label="Media Type", value="Image", scale=1)
-        search_button = gr.Button("Search")
+    with gr.Tab("Pixabay Search"):
+        gr.Markdown("Search for royalty-free images and videos on Pixabay.")
+        gr.Warning("This requires setting the PIXABAY_API_KEY environment variable.")
+        with gr.Row():
+            pixabay_query_input = gr.Textbox(label="Search Query", placeholder="e.g., yellow flowers", scale=2)
+            pixabay_media_type_radio = gr.Radio(["Image", "Video"], label="Media Type", value="Image", scale=1)
+            pixabay_search_button = gr.Button("Search")
 
-    # Containers to hold media-specific inputs, initially hidden or shown
-    with gr.Column(visible=True) as image_options_col:
-        image_type_input = gr.Radio(["all", "photo", "illustration", "vector"], label="Image Type", value="all")
-        orientation_input = gr.Radio(["all", "horizontal", "vertical"], label="Orientation", value="all")
+        with gr.Column(visible=True) as pixabay_image_options_col:
+            pixabay_image_type_input = gr.Radio(["all", "photo", "illustration", "vector"], label="Image Type", value="all")
+            pixabay_orientation_input = gr.Radio(["all", "horizontal", "vertical"], label="Orientation", value="all")
 
-    with gr.Column(visible=False) as video_options_col:
-        video_type_input = gr.Radio(["all", "film", "animation"], label="Video Type", value="all")
+        with gr.Column(visible=False) as pixabay_video_options_col:
+            pixabay_video_type_input = gr.Radio(["all", "film", "animation"], label="Video Type", value="all")
 
-    status_output = gr.Textbox(label="Status", interactive=False)
+        pixabay_status_output = gr.Textbox(label="Status", interactive=False)
 
-    # Outputs will appear side-by-side in a row
-    with gr.Row():
-        image_output = gr.Image(label="Result URL", type="filepath", interactive=False)
-        video_output = gr.Video(label="Result URL", interactive=False)
+        with gr.Row():
+            pixabay_image_output = gr.Image(label="Result Image (URL)", type="filepath", interactive=False)
+            pixabay_video_output = gr.Video(label="Result Video (URL)", type="filepath", interactive=False)
 
-    # Logic to toggle visibility of input columns based on media type selection
-    def update_inputs_blocks(media_type):
-        if media_type == "Image":
-            return gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False) # show image cols, hide video cols, show image outputs, hide video outputs
-        elif media_type == "Video":
-            return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=True) # hide image cols, show video cols, hide image outputs, show video outputs
-        else:
-            # Default state
-            return gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
-
-    # Trigger the update_inputs_blocks function when media_type_radio changes
-    media_type_radio.change(
-        fn=update_inputs_blocks,
-        inputs=media_type_radio,
-        outputs=[image_options_col, video_options_col, image_output, video_output] # Update visibility of columns AND output components
-    )
-
-    # Trigger the search_pixabay function when the search button is clicked
-    # Pass all relevant inputs, even if some are hidden
-    search_button.click(
-        fn=search_pixabay,
-        inputs=[
-            query_input,
-            media_type_radio,
-            image_type_input,
-            orientation_input,
-            video_type_input
-        ],
-        outputs=[image_output, video_output, status_output] # Map function outputs to Gradio components
-    )
+        # Logic to toggle visibility of input columns and output components based on media type selection
+        def update_pixabay_inputs_blocks(media_type):
+            if media_type == "Image":
+                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)
+            else:
+                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]
+        )
+
+        # Trigger the search_pixabay function when the search button is clicked
+        pixabay_search_button.click(
+            fn=search_pixabay,
+            inputs=[
+                pixabay_query_input,
+                pixabay_media_type_radio,
+                pixabay_image_type_input,
+                pixabay_orientation_input,
+                pixabay_video_type_input
+            ],
+            outputs=[pixabay_image_output, pixabay_video_output, pixabay_status_output]
+        )
+
+    with gr.Tab("Together AI - Text to Image"):
+        gr.Markdown("Generate an image from a text prompt using Together AI.")
+        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_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,
+            inputs=together_text_to_image_prompt,
+            outputs=together_text_to_image_output,
+        )
+
+    with gr.Tab("Together AI - Image to Image"):
+        gr.Markdown("Transform an uploaded image based on a text prompt using Together AI.")
+        gr.Warning("This requires setting the TOGETHER_API_KEY environment variable.")
+        with gr.Row():
+            together_image_input = gr.Image(label="Upload or paste an image", scale=2)
+            together_image_to_image_prompt = gr.Textbox(label="Enter your transformation prompt", scale=2)
+            together_image_to_image_button = gr.Button("Transform Image", scale=1)
+        together_image_to_image_output = gr.Image(label="Transformed Image (URL)", type="filepath", interactive=False)
+
+        together_image_to_image_button.click(
+            fn=together_image_to_image,
+            inputs=[together_image_input, together_image_to_image_prompt],
+            outputs=together_image_to_image_output,
+        )
 
 # --- Launch the Gradio app ---
 if __name__ == "__main__":