# -*- coding: utf-8 -*-
"""Copy of app.ipynb

Automatically generated by Colab.

Original file is located at
    https://colab.research.google.com/drive/1KoR_JrJMqzUq-XagPaxVUp874bVW_qNI
"""

#/default_exp app



#
#from fastai.vision.all import *
#!pip install gradio
import gradio as gr

#/export
def is_monet(x): return x[0].issupper

#/export
from fastai.vision.all import *

#/export
learn = load_learner('model.pkl')

#/export
categories = ('Manet', 'Monet')
def classify_image(img):
  pred, idx, probs = learn.predict(img)
  return dict(zip(categories, map(float, probs)))

# export
image = gr.Image()  # Image input without shape argument
label = gr.Label()  # Label output

# Define some example images (make sure these paths are correct)
examples = ['monet.jpg', 'manet2.jpg', 'manet1.jpeg']

# Create the Gradio interface
#intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)

# Create the Gradio interface with a title and description
intf = gr.Interface(
    fn=classify_image, 
    inputs=image, 
    outputs=label, 
    examples=examples,
    title="Monet vs Manet Image Classifier",  # Add title here
    description="Upload an image to classify it as either a Monet or Manet painting."  # Optional description
)

# Launch the interface with Inline=False to open in a separate window
intf.launch(share=True)

#!pip install nbdev

import os
os.listdir('/content')

import os
notebook_path = '/content/app.ipynb'
print(os.path.abspath(notebook_path))

import os
file_exists = os.path.isfile('/content/app.ipynb')
print(file_exists)  # This should return True if the file exists

import os

# Extract the path of the current notebook in Colab
notebook_path = '/content/app.ipynb'  # Replace with the actual path if different
print(f"Notebook is located at: {notebook_path}")

nbdev.export.nb_export('Copy of app.ipynb', 'app')
print('Export successful')