Fix Gradio 6.0 compatibility and improve image handling

app.py

@@ -60,17 +60,17 @@ Response:"""
         return True, ""
 
 
-def search_anatomy_image(query: str) -> tuple[str, str]:
+def search_anatomy_image(query: str) -> tuple[list, str]:
     """
     Search for anatomy images using SERPAPI Google Images.
-    Returns (image_url, error_message)
+    Returns (list_of_image_urls, error_message)
     """
     try:
         params = {
             "engine": "google_images",
             "q": f"{query} anatomy diagram",
             "api_key": SERPAPI_KEY,
-            "num": 5,
+            "num": 10,  # Get more results for fallback
             "safe": "active"
         }
         
@@ -80,14 +80,23 @@ def search_anatomy_image(query: str) -> tuple[str, str]:
         data = response.json()
         
         if "images_results" in data and len(data["images_results"]) > 0:
-            # Get the first high-quality image
-            image_url = data["images_results"][0]["original"]
-            return image_url, ""
+            # Get multiple image URLs, filter out SVG files
+            image_urls = []
+            for img in data["images_results"]:
+                url = img.get("original", "")
+                # Skip SVG files and other problematic formats
+                if url and not url.lower().endswith('.svg'):
+                    image_urls.append(url)
+            
+            if image_urls:
+                return image_urls, ""
+            else:
+                return [], "No supported image formats found (SVG files excluded)."
         else:
-            return "", "No images found for this anatomy topic."
+            return [], "No images found for this anatomy topic."
     
     except Exception as e:
-        return "", f"Error searching for images: {str(e)}"
+        return [], f"Error searching for images: {str(e)}"
 
 
 def download_image(image_url: str) -> Image.Image:
@@ -95,7 +104,14 @@ def download_image(image_url: str) -> Image.Image:
     Download and return PIL Image from URL.
     """
     try:
-        response = requests.get(image_url, timeout=10)
+        # Add headers to mimic a browser request and avoid 403 errors
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
+            'Accept': 'image/avif,image/webp,image/apng,image/svg+xml,image/*,*/*;q=0.8',
+            'Accept-Language': 'en-US,en;q=0.9',
+            'Referer': 'https://www.google.com/'
+        }
+        response = requests.get(image_url, headers=headers, timeout=10)
         response.raise_for_status()
         img = Image.open(BytesIO(response.content))
         return img
@@ -155,19 +171,28 @@ def process_anatomy_query(query: str) -> tuple:
     if not is_valid:
         return None, "", validation_msg
     
-    # Search for image
-    image_url, img_error = search_anatomy_image(query)
+    # Search for images
+    image_urls, img_error = search_anatomy_image(query)
     
     # Generate educational information
     info = generate_anatomy_info(query)
     
-    # Download image if URL is available
+    # Try to download images from the list until one succeeds
     image = None
-    if image_url:
-        try:
-            image = download_image(image_url)
-        except Exception as e:
-            img_error = str(e)
+    download_error = ""
+    
+    if image_urls:
+        for url in image_urls[:5]:  # Try up to 5 images
+            try:
+                image = download_image(url)
+                download_error = ""  # Success!
+                break  # Stop trying once we get a valid image
+            except Exception as e:
+                download_error = str(e)
+                continue  # Try next image
+        
+        if not image and download_error:
+            img_error = f"Could not download images. Last error: {download_error}"
     
     # Prepare result
     error_message = ""
@@ -178,7 +203,7 @@ def process_anatomy_query(query: str) -> tuple:
 
 
 # Create Gradio interface
-with gr.Blocks(theme=gr.themes.Soft(), title="AnatomyBot - MBBS Anatomy Tutor") as demo:
+with gr.Blocks(title="AnatomyBot - MBBS Anatomy Tutor") as demo:
     gr.Markdown(
         """
         # 🩺 AnatomyBot - Your MBBS Anatomy Tutor
@@ -241,4 +266,4 @@ if __name__ == "__main__":
     if not HYPERBOLIC_API_KEY or HYPERBOLIC_API_KEY == "your_hyperbolic_api_key_here":
         print("⚠️  WARNING: HYPERBOLIC_API_KEY not configured in .env file")
     
-    demo.launch(share=False, server_name="0.0.0.0", server_port=7860)
+    demo.launch(share=False, server_name="0.0.0.0", server_port=7861)