import os
os.system('pip install -r requirements-colab.txt')
os.system('pip install gradio tensorflow numpy')
from deoldify import device
from deoldify.device_id import DeviceId
device.set(device=DeviceId.GPU0)

import torch

if not torch.cuda.is_available():
    print('GPU not available.')

import fastai
from deoldify.visualize import *
import warnings
warnings.filterwarnings("ignore", category=UserWarning, message=".*?Your .*? set is empty.*?")
colorizer = get_image_colorizer(artistic=True)
# source_url = 'https://i.pinimg.com/originals/09/c2/22/09c222cc275efa908bdc140a750ca4e4.jpg'
# image_path_1 = colorizer.plot_transformed_image_from_url(url=source_url, render_factor=render_factor, compare=True, watermarked=watermarked)
# image_path_2 = colorizer.plot_transformed_image(path=source_path, render_factor=render_factor, compare=True)
# print("image_path",image_path_1)

render_factor=10

import gradio as gr
import tensorflow as tf
import numpy as np
import os

def Colorize(address,watermark,URL,render_factor):
    text_box_output = "Successful"
    watermarked = True
    output_address = np.array([
    [0,0],
    [0,0],
], dtype=np.uint8)
    if watermark == "Watermark On":
      watermarked = True
      render_factor = int(render_factor)
    elif watermark == "Watermark Off":
      watermarked = False
      render_factor = int(render_factor)
    else:
      text_box_output = "Please set the watermark setting."
    if URL != '':
      output_address = colorizer.plot_transformed_image_from_url(url=URL, render_factor=render_factor, compare=True, watermarked=watermarked)
    else:
      if address == None:
        text_box_output = "Please provide the image"
      else:
       output_address = colorizer.plot_transformed_image(path=address, render_factor=render_factor, compare=True, watermarked=watermarked)
    return text_box_output,output_address
with gr.Blocks() as demo:
    image_box=gr.Image(type="filepath")
    image_output_box = gr.Image()
   # text_box_1 = gr.Textbox(value="10",label="The render factor")
    text_box_2 = gr.Textbox(label='URL')
    drop_down_1 = gr.Dropdown(['Watermark On','Watermark Off'],label='WaterMark')
    slider = gr.Slider(7,40,step=1)
    text_box_output = gr.Textbox(label="Status")
    slider.release(Colorize, inputs=[image_box,drop_down_1,text_box_2,slider], outputs=[text_box_output,image_output_box])
    #b1=gr.Button("Colorized!")
    #b1.click(Colorize, inputs=[image_box,drop_down_1,text_box_2,text_box_1], outputs= image_output_box)

demo.queue().launch(share=True,debug=True)