Update app.py

app.py

@@ -25,8 +25,15 @@ mushroom_species = {
 }
 
 # 🎨 Image preprocessing
-transform = transforms.Compose([
-    transforms.Resize((224, 224)),
+# ✅ Define transforms for training and validation
+train_transform = transforms.Compose([
+    transforms.Resize((256, 256)),
+    transforms.RandomResizedCrop(224, scale=(0.8, 1.0)),
+    transforms.RandomHorizontalFlip(p=0.5),
+    transforms.RandomVerticalFlip(p=0.3),
+    transforms.RandomRotation(degrees=45),
+    transforms.ColorJitter(brightness=0.3, contrast=0.3, saturation=0.2, hue=0.05),
+    transforms.RandomAffine(degrees=0, translate=(0.1, 0.1), scale=(0.9, 1.1)),
     transforms.ToTensor(),
     transforms.Normalize([0.485, 0.456, 0.406],
                          [0.229, 0.224, 0.225])
@@ -59,12 +66,9 @@ 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 in new browser tab
+# 🔗 Open user manual (link version) 🆕
 MANUAL_URL = "https://drive.google.com/drive/folders/19lUCEaLstrRjCzqpDlWErhRd1EXWUGbf?usp=sharing"
-def open_manual():
-    import webbrowser
-    webbrowser.open_new_tab(MANUAL_URL)  # 🆕 Ensure it opens in new tab
-    return "", "", "", "Opening user manual..."
+manual_link = f"<a href=\"{MANUAL_URL}\" target=\"_blank\">📄 Open User Manual</a>"
 
 # 🎛️ Gradio UI
 if __name__ == "__main__":
@@ -73,7 +77,7 @@ if __name__ == "__main__":
         gr.Markdown("Upload a mushroom photo or use your camera to check if it’s edible or poisonous.\nอัปโหลดรูปเห็ดหรือใช้กล้องเพื่อตรวจสอบว่าเห็ดกินได้หรือมีพิษ")
 
         with gr.Row():
-            image_input = gr.Image(type="pil", label="📷 Upload or Capture Mushroom Image")
+            image_input = gr.Image(type="pil", label="📷 Upload or Capture Mushroom Image", source="upload")  # 🆕 safer source usage
             with gr.Column():
                 label_en = gr.Textbox(label="🧬 Prediction (English)")
                 label_th = gr.Textbox(label="🔁 คำทำนาย (ภาษาไทย)")
@@ -81,7 +85,6 @@ if __name__ == "__main__":
                 label_hint = gr.Textbox(label="🏷️ Likely Species (Based on Training Data)")
 
         classify_btn = gr.Button("🔍 Classify")
-        manual_btn = gr.Button("Open User Manual 📄")
 
         classify_btn.click(
             fn=classify_mushroom,
@@ -89,13 +92,8 @@ if __name__ == "__main__":
             outputs=[label_en, label_th, confidence, label_hint]
         )
 
-        manual_btn.click(
-            fn=open_manual,
-            inputs=[],
-            outputs=[label_en, label_th, confidence, label_hint]
-        )
-
         gr.Markdown("---")
-        gr.Markdown("App version: 1.0.1 | Updated: August 2025")
+        gr.Markdown(manual_link)  # 🆕 Display clickable user manual
+        gr.Markdown("App version: 1.0.2 | Updated: August 2025")  # 🆕 Version bump
 
     demo.launch()