Update app.py

app.py

@@ -5,6 +5,7 @@ import torchvision.transforms as transforms
 from torchvision import models
 import gradio as gr
 import webbrowser
+from rembg import remove  # 🆕 Background removal
 
 # 🔧 Set device
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -36,8 +37,9 @@ CONFIDENCE_THRESHOLD = 85.0  # Minimum confidence considered safe enough to show
 
 def classify_mushroom(image: Image.Image):
     try:
-        image = image.convert("RGB")
-        tensor = transform(image).unsqueeze(0).to(device)
+        image = image.convert("RGBA")  # Required format for rembg
+        image_no_bg = remove(image).convert("RGB")  # 🆕 Remove background
+        tensor = transform(image_no_bg).unsqueeze(0).to(device)
 
         with torch.no_grad():
             outputs = model(tensor)
@@ -57,10 +59,11 @@ def classify_mushroom(image: Image.Image):
         print(f"❌ Error: {e}")
         return "Error", "ผิดพลาด", "N/A", "Invalid image. Please upload a valid mushroom photo."
 
-# 🔗 Open user manual
+# 🔗 Open user manual in new browser tab
 MANUAL_URL = "https://drive.google.com/drive/folders/19lUCEaLstrRjCzqpDlWErhRd1EXWUGbf?usp=sharing"
 def open_manual():
-    webbrowser.open(MANUAL_URL)
+    import webbrowser
+    webbrowser.open_new_tab(MANUAL_URL)  # 🆕 Ensure it opens in new tab
     return "", "", "", "Opening user manual..."
 
 # 🎛️ Gradio UI