#!/usr/bin/env python
# coding: utf-8
#dosyayı py olarak kaydet ve komut satırını kullanarak streamlit run streamlit.py 
import streamlit as st
from tensorflow.keras.models import load_model
from PIL import Image
import numpy as np
import cv2
model=load_model('GTS_cnn.h5')
def process_image(img):
    img=img.resize((30,30))
    img=np.array(img)
    img=img/255.0
    img=np.expand_dims(img,axis=0)
    return img
st.title('GTS Trafik İşaretleri: :vertical_traffic_light: ')
st.write('Trafik işaretini yükleyin')
file=st.file_uploader('Bir resim seç', type= ['jpg','jpeg','png'])
class_names=[
    "Hız sınırı (20km/s)", "Hız sınırı (30km/s)", "Hız sınırı (50km/s)", "Hız sınırı (60km/s)", "Hız sınırı (70km/s)",
    "Hız sınırı (80km/s)", "Hız sınırı sonu (80km/s)", "Hız sınırı (100km/s)", "Hız sınırı (120km/s)", "Geçiş yasağı",
    "3.5 ton üstü araç geçişi yasak", "Kavşakta geçiş hakkı", "Ana yol", "Yol ver", "Dur", "Araç girişi yasak",
    "3.5 ton üstü araç girişi yasak", "Giriş yasak", "Genel dikkat", "Tehlikeli sol viraj", "Tehlikeli sağ viraj",
    "Çift viraj", "Engebeli yol", "Kaygan yol", "Yol sağdan daralıyor", "Yol çalışması", "Trafik ışıkları", "Yaya geçidi",
    "Çocuk geçidi", "Bisiklet geçidi", "Buzlanma/kar dikkati", "Yabani hayvanlar geçebilir", "Hız ve geçiş limitlerinin sonu",
    "İleri sağa dön", "İleri sola dön", "Sadece düz gidin", "Düz git veya sağa dön", "Düz git veya sola dön", "Sağdan devam et",
    "Soldan devam et", "Döner kavşak zorunlu", "Geçiş yasağı sonu", "3.5 ton üstü geçiş yasağı sonu"
]

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)
    st.write(class_names[predicted_class])