Update app.py

app.py

@@ -53,16 +53,13 @@ def search_pixabay(query: str = "", media_type: str = "Image", image_type: str =
         video_type (str): Filter results by video type (used only if media_type is "Video"). 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
+        str: URL of the found media or status/error message
     """
     if not query:
-        return None, None, "Please enter a search query."
+        return "Please enter a search query."
 
     if not PIXABAY_API_KEY:
-        return None, None, "Pixabay API Key not found. Please set the PIXABAY_API_KEY environment variable."
+        return "Pixabay API Key not found. Please set the PIXABAY_API_KEY environment variable."
 
     params = {
         'key': PIXABAY_API_KEY,
@@ -80,7 +77,7 @@ def search_pixabay(query: str = "", media_type: str = "Image", image_type: str =
         api_url = VIDEO_API_URL
         params['video_type'] = video_type
     else:
-        return None, None, "Invalid media type selected."
+        return "Invalid media type selected."
 
     try:
         response = requests.get(api_url, params=params)
@@ -88,20 +85,20 @@ def search_pixabay(query: str = "", media_type: str = "Image", image_type: str =
         data = response.json()
 
         if data.get('totalHits', 0) == 0:
-            return None, None, f"No results found for '{query}'."
+            return f"No results found for '{query}'."
 
         hits = data.get('hits', [])
         if not hits:
-             return None, None, f"No results found for '{query}'."
+             return f"No results found for '{query}'."
 
         selected_hit = random.choice(hits)
 
         if media_type == "Image":
             image_url = selected_hit.get('largeImageURL')
             if image_url:
-                return image_url, None, ""
+                return image_url
             else:
-                return None, None, "Could not retrieve large image URL."
+                return "Could not retrieve large image URL."
 
         elif media_type == "Video":
             video_urls = selected_hit.get('videos', {})
@@ -109,21 +106,23 @@ def search_pixabay(query: str = "", media_type: str = "Image", image_type: str =
             video_url = large_video.get('url')
 
             if video_url:
-                return None, video_url, ""
+                return video_url
             else:
                  medium_video = video_urls.get('medium', {})
                  video_url = medium_video.get('url')
                  if video_url:
-                     return None, video_url, "Using medium quality video."
+                     return video_url
                  else:
-                    return None, None, "Could not retrieve video URL."
+                    return "Could not retrieve video URL."
 
     except requests.exceptions.RequestException as e:
-        return None, None, f"API request error: {e}"
+        return f"API request error: {e}"
     except json.JSONDecodeError:
-         return None, None, "Error decoding API response."
+         return "Error decoding API response."
     except Exception as e:
-        return None, None, f"An unexpected error occurred: {e}"
+        return f"An unexpected error occurred: {e}"
+
+# In the Gradio interface section, replace the Pixabay search outputs with:
 
 def together_text_to_image(prompt: str = "", width: int = 1024, height: int = 1024):
     """
@@ -222,41 +221,33 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
             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")
-
+    
         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 pixabay_video_options_col:
             pixabay_video_type_input = gr.Radio(["all", "film", "animation"], label="Video Type", value="all")
-
-        pixabay_status_output = gr.Textbox(label="Status", interactive=False)
-
-        with gr.Row():
-            pixabay_image_output = gr.Image(label="Result Image (URL)", interactive=False)
-            pixabay_video_output = gr.Video(label="Result Video (URL)", interactive=False)
-
+    
+        pixabay_output = gr.Textbox(label="Result", interactive=False)
+    
         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))
             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))
             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))
+    
         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],
             api_name=False,
             show_api=False
         )
-
+    
         pixabay_search_button.click(
             fn=search_pixabay,
             inputs=[
@@ -266,7 +257,7 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
                 pixabay_orientation_input,
                 pixabay_video_type_input
             ],
-            outputs=[pixabay_image_output, pixabay_video_output, pixabay_status_output]
+            outputs=pixabay_output
         )
 
     with gr.Tab("Together AI - Text to Image"):
@@ -277,7 +268,7 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
             together_text_to_image_width = gr.Slider(label="Width", value=1024, minimum=512, maximum=1440)
             together_text_to_image_height = gr.Slider(label="Height", value=1024, minimum=512, maximum=1440)
             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_output = gr.Textbox(label="Generated Image (URL)", type="filepath", interactive=False)
 
         together_text_to_image_button.click(
             fn=together_text_to_image,
@@ -292,7 +283,7 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
             together_image_input = gr.Image(label="Upload or paste an image", type="filepath", 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_output = gr.Textbox(label="Transformed Image (URL)", type="filepath", interactive=False)
 
         together_image_to_image_button.click(
             fn=together_image_to_image,
@@ -313,7 +304,7 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
             )
             tts_generate_button = gr.Button("Generate Audio")
     
-        tts_audio_output = gr.Audio(label="Generated Audio", interactive=False)
+        tts_audio_output = gr.Textbox(label="Generated Audio (url)", interactive=False)
     
         def text_to_speech(text: str = "", voice: str = ""):
             """
@@ -324,7 +315,7 @@ with gr.Blocks(title="Media Generation and Search Explorer") as demo:
                 voice (str): The voice to use for speech synthesis. All available voices are: calm lady, meditation lady, storyteller lady, wise lady, teacher lady, wise man, customer support man, tutorial man, helpful woman, customer support lady, asmr lady, pleasant man, professional woman, reading lady, reading man. Default is Helpful Woman.
     
             Returns:
-                str: Path to the generated audio file or error message
+                url (str): Give url to the generated audio file or error message
             """
             if not client:
                 return None, "Together AI client not initialized. Please set the TOGETHER_API_KEY environment variable."