import os
import shutil
import pickle
import numpy as np
import tensorflow as tf
from tensorflow.keras import datasets, layers, models

def load_cifar10_data(data_dir):
train_images = []
train_labels = []

for i in range(1, 6):
with open(os.path.join(data_dir, f'data_batch_{i}'), 'rb') as f:
data_dict = pickle.load(f, encoding='bytes')
images = data_dict[b'data']
labels = data_dict[b'labels']


train_images.extend(images)
train_labels.extend(labels)


train_images = np.array(train_images).reshape(-1, 3,
32, 32).transpose(0, 2, 3, 1)
train_labels = np.array(train_labels)


with open(os.path.join(data_dir, 'test_batch'), 'rb') as f:
data_dict = pickle.load(f, encoding='bytes')
test_images = data_dict[b'data'].reshape(
-1, 3, 32, 32).transpose(0, 2, 3, 1)
test_labels = np.array(data_dict[b'labels'])

return (train_images, train_labels), (test_images, test_labels)

def build_model():
# Define the model architecture
model = models.Sequential()
model.add(layers.Conv2D(
32, (3, 3), activation='relu', input_shape=(32, 32, 3)))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(64, (3, 3), activation='relu'))
model.add(layers.MaxPooling2D((2, 2)))
model.add(layers.Conv2D(64, (3, 3), activation='relu'))

# Add dense layers on top
model.add(layers.Flatten())
model.add(layers.Dense(64, activation='relu'))
model.add(layers.Dense(10))

return model

def train_model(model, train_images, train_labels, test_images, test_labels):
# Compile and train the model
model.compile(optimizer='adam',
loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),
metrics=['accuracy'])
history = model.fit(train_images, train_labels, epochs=50,
validation_data=(test_images, test_labels))

# Check if the model directory exists
if os.path.exists('model'):

# If it does, delete it
shutil.rmtree('model')

# Recreate the model directory
os.makedirs('model')

# Save the model
model.save('model/cifar-10-batches-py-model.keras')

return history

# Load and preprocess the CIFAR10 dataset
data_dir = 'cifar-10-batches-py'
(train_images, train_labels), (test_images,
test_labels) = load_cifar10_data(data_dir)

# Normalize pixel values to be between 0 and 1
train_images, test_images = train_images / 255.0, test_images / 255.0

# Build and train the model
model = build_model()
history = train_model(model, train_images, train_labels,
test_images, test_labels)

# Print the history dictionary
print(history.history)

from taipy.gui import Gui
from taipy.gui import Html
from tensorflow.keras import models
from PIL import Image
import tensorflow
import numpy as np


class_names = {

0: 'airplane',
1: 'automobile',
2: 'bird',
3: 'cat',
4: 'deer',
5: 'dog',
6: 'frog',
7: 'horse',
8: 'ship',
9: 'truck',
}


model = models.load_model("model/cifar-10-batches-py-model.keras")


def predict_image(model, path_to_img):
img = Image.open(path_to_img)
img = img.convert("RGB")
img = img.resize((32, 32))
data = np.asarray(img)
data = data / 255
logits = model.predict(np.array([data])[:1])
probs = tensorflow.nn.softmax(logits).numpy()

top_prob = probs.max()
top_pred = class_names[np.argmax(probs)]


return top_prob, top_pred


opt = tensorflow.keras.optimizers.legacy.Adam(learning_rate=0.1)


content = ""
img_path = "https://placehold.co/600?text=No+Image+Available&font=roboto"
prob = 0
pred = ""


html_page = Html("""

<div class="container">

<h1>Machine Learning Photo App</h1>

<p>There is a prediction indication that ranges from 0 to 100. The greater the value, the more certain the machine model is that the prediction is true. This all depends on the Neural Network Builder model that we generated. The longer you train the model, the smarter it will get. If it does not have enough training time, it will make incorrect predictions.</p>

<div class="attachment">

<div><taipy:file_selector extensions=".png">{content}</taipy:file_selector></div>

<div><p>Choose an image from your computer to upload</p></div>

</div>

<div>

#<taipy:image>{img_path}</taipy:image>

<taipy:indicator min="0" max="100" width="25vw" height="25vh" orientation="vertical" value="{prob}">{prob}</taipy:indicator>

</div>

<div class="prediction">

<p>Prediction:</p><div><taipy:text>{pred}</taipy:text></div>

</div>

</div>

""")


def on_change(state, var_name, var_val):
if var_name == "content":
top_prob, top_pred = predict_image(model, var_val)
state.prob = round(top_prob * 100)
state.pred = "Its a " + top_pred
state.img_path = var_val


app = Gui(page=html_page)

if __name__ == "__main__":

app.run(use_reloader=True, port=8000)