import streamlit as st 
from tensorflow.keras.models import load_model
from PIL import Image
import numpy as np

model=load_model('capstone_cnn_model.h5')

def process_image(img):
    img=img.resize((32,32)) #boyutunu 32 x 32 pixel yaptik
    img=np.array(img)
    img=img/255.0 #normalize ettik
    img=np.expand_dims(img,axis=0)
    return img

st.title("Trafik Levhaları Tanıma :traffic_light: :vertical_traffic_light: :construction:")
st.write("Trafik levhası resmi ekleyin")

file=st.file_uploader('Bir Resim Sec',type=['jpg','jpeg','png'])

if file is not None:
    img=Image.open(file)
    st.image(img,caption='yuklenen resim')
    image= process_image(img)
    prediction=model.predict(image)
    predicted_class=np.argmax(prediction)
    
    class_names=["maksimum hız 20 km/h",  #0
                 "maksimum hız 30 km/h",  #1
                 "maksimum hız 50 km/h",  #2
                 "maksimum hız 60 km/h",  #3
                 "maksimum hız 70 km/h",  #4
                 "maksimum hız 80 km/h"   #5
                 "80 km/hhız limiti sonu",#6
                 "maksimum hız limiti 100 km/h", #7
                 "maksimum hız limiti 120 km/h", #8
                 "sollamak yasak", #9
                 "ağır tonajlı araçlara sollama yasağı", #10
                 "kavşak yaklaşıyor, öncelikli yoldasınız", #11
                 "geçiş öncelikli yol", #12
                 "yol ver",  #13
                 "dur",  #14                         
                 "motorlu araçlar giremez",  #15     
                 "ağır tonajlı araçlar için giriş yasağı", #16     
                 "motorlu araçlar için giriş yasağı", #17            
                 "dikkat", #18                                               
                 "sola doğru viraj",  #19                      
                 "sağa doğru viraj",  #20                      
                 "soldan başlayan çift viraj", #21             
                 "kasisli yol", #22
                 "kaygan yol",  #23              
                 "yol sağdan daralıyor", #24                   
                 "inşaat var",  #25                            
                 "dikkat, trafik ışığı ile kontrol edilen kavşak yaklaşıyor", #26
                 "yayalara dikkat",  #27                       
                 "dikkat, meskun mahal",  #28                  
                 "bisikletlere dikkat",   #29                  
                 "kar ve buzdan dolayı kayma tehlikesi",  #30  
                 "dikkat, yabani hayvan çıkabilir",  #31       
                 "tüm yasakların sonu",  #32                   
                 "sağa dönme mecburiyeti",  #33                
                 "sola dönme mecburiyeti",   #34               
                 "düz devam etme mecburiyeti", #35             
                 "düz gidilebilir veya sağa dönülebilir",   #36
                 "sola dönülebilir veya düz gidilebilir",   #37
                 "sağdan gidiniz",  #38                        
                 "soldan gidiniz",  #39                       
                 "dönel kavşak",    #40                      
                 "sollama yasağı sonu",   #41                  
                 "ağır tonajlı arabalara sollama yasağının sonu"] #42
    
    st.write(class_names[predicted_class])