add feature

app.py

@@ -1,6 +1,8 @@
 from ultralytics import YOLO
 import streamlit as st
 from PIL import Image
+import requests
+from bs4 import BeautifulSoup
 model = YOLO('resource/best.pt')
 
 classes = ['Acrochordus-granulatus', 'Ahaetulla-nasuta', 'Ahaetulla-prasina', 'Argyrogena-fasciolata', 'Boiga-cyanea', 'Boiga-cynodon', 'Boiga-multomaculata', 'Boiga-trigonata', 'Bungarus-caeruleus', 'Bungarus-fasciatus', 'Cerberus-rynchops', 'Chrysopelea-ornata', 'Coelognathus-radiatus', 'Daboia-russelii', 'Dendrelaphis-cyanochloris', 'Dendrelaphis-pictus', 'Dendrelaphis-tristis', 'Echis-carinatus', 'Enhydris-enhydris', 'Fordonia-leucobalia', 'Fowlea-piscator', 'Gerarda-prevostiana', 'Homalopsis-buccata', 'Hydrophis-platurus', 'Indotyphlops-braminus', 'Laticauda-colubrina', 'Laticauda-laticaudata', 'Lycodon-aulicus', 'Macropisthodon-plumbicolor', 'Malayopython-reticulatus', 'Naja-kaouthia', 'Naja-naja', 'Oligodon-arnensis', 'Oreocryptophis-porphyraceus', 'Ovophis-monticola', 'Psammodynastes-pulverulentus', 'Ptyas-korros', 'Ptyas-mucosa', 'Python-bivittatus', 'Rhabdophis-subminiatus', 'Sinomicrurus-macclellandi', 'Trimeresurus-albolabris', 'Trimeresurus-erythrurus', 'Trimeresurus-popeiorum', 'Xenochrophis-piscator', 'Nil-object']
@@ -17,6 +19,12 @@ if photo:
     try:
         bx = model(image)[0].boxes[0]
         col1.write(f"Result: {(classes[int(bx.cls[0])])}")
+        txt = classes[int(bx.cls[0])]
+        genus, species = txt.split('-')
+        data = requests.get(f'https://reptile-database.reptarium.cz/species?genus={genus}&species={species}')
+        bs = BeautifulSoup(data.content, 'html.parser')
+        desc = bs.find('td', string='Comment').find_next_sibling('td').text
+        col1.write(desc)
     except:
         col1.write(f"Result: Could not detect")
 else:

requirements.txt

@@ -1,3 +1,5 @@
 torch
 ultralytics
-pillow
+pillow
+requests
+bs4