Update app.py

app.py

@@ -4,15 +4,24 @@ import random
 import json
 import os
 
+# --- 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
-
-def search_pixabay(query: str, media_type: str, image_type: str = 'all', orientation: str = 'all', video_type: str = 'all'):
+PER_PAGE = 5 
+
+# --- Function to search Pixabay ---
+def search_pixabay(query: str, media_type: str, image_type: str, orientation: str, video_type: str):
+    """
+    Searches Pixabay API and returns a random result URL and status message.
+    Takes all type/orientation inputs, but uses them based on media_type.
+    """
     if not query:
         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."
+
     params = {
         'key': PIXABAY_API_KEY,
         'q': query,
@@ -21,6 +30,7 @@ def search_pixabay(query: str, media_type: str, image_type: str = 'all', orienta
         'safesearch': 'true'
     }
 
+    # Determine API endpoint and add media-specific parameters
     if media_type == "Image":
         api_url = IMAGE_API_URL
         params['image_type'] = image_type
@@ -29,12 +39,13 @@ def search_pixabay(query: str, media_type: str, image_type: str = 'all', orienta
         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:
         response = requests.get(api_url, params=params)
-        response.raise_for_status()
-        data = response.json()
+        response.raise_for_status() # Check for HTTP errors (like 404, 429, 500)
+        data = response.json() # Parse the JSON response
 
         if data.get('totalHits', 0) == 0:
             return None, None, f"No results found for '{query}'."
@@ -43,23 +54,30 @@ def search_pixabay(query: str, media_type: str, image_type: str = 'all', orienta
         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
                 return image_url, None, ""
             else:
                 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')
 
             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:
@@ -72,51 +90,65 @@ def search_pixabay(query: str, media_type: str, image_type: str = 'all', orienta
     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}"
 
-# Gradio input components
-image_type_input = gr.Radio(["all", "photo", "illustration", "vector"], label="Image Type", value="all", visible=True)
-orientation_input = gr.Radio(["all", "horizontal", "vertical"], label="Orientation", value="all", visible=True)
-video_type_input = gr.Radio(["all", "film", "animation"], label="Video Type", value="all", visible=False)
-media_type_radio = gr.Radio(["Image", "Video"], label="Media Type", value="Image")
+# --- 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.")
 
-# Function to toggle input visibility
-def update_inputs(media_type):
-    if media_type == "Image":
-        return gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
-    elif media_type == "Video":
-        return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
-    else:
-        return gr.update(visible=True), gr.update(visible=True), gr.update(visible=False)
-
-media_type_radio.change(
-    fn=update_inputs,
-    inputs=media_type_radio,
-    outputs=[image_type_input, orientation_input, video_type_input]
-)
-
-# Gradio output components
-output_components = [
-    gr.Image(label="Result URL", type="filepath", interactive=False),
-    gr.Video(label="Result URL", interactive=False),
-    gr.Textbox(label="Status", interactive=False)
-]
-
-# Define the Gradio interface
-interface = gr.Interface(
-    fn=search_pixabay,
-    inputs=[
-        gr.Textbox(label="Search Query", placeholder="e.g., yellow flowers"),
-        media_type_radio,
-        image_type_input,
-        orientation_input,
-        video_type_input
-    ],
-    outputs=output_components,
-    title="Pixabay Random Media Explorer (URL Output)",
-    description="Get a random media result from Pixabay (URL output)."
-)
-
-# Launch the app
+    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")
+
+    # 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=False) as video_options_col:
+        video_type_input = gr.Radio(["all", "film", "animation"], label="Video Type", value="all")
+
+    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)
+
+    # 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
+    )
+
+# --- Launch the Gradio app ---
 if __name__ == "__main__":
-    interface.launch()
+    demo.launch()