import cv2
import gradio as gr
import os
import functools
from PIL import Image
from rembg import remove 
from io import BytesIO
import numpy as np
import torch
from torch.autograd import Variable
from torchvision import transforms
import torch.nn.functional as F
import gdown
import matplotlib.pyplot as plt
import warnings
warnings.filterwarnings("ignore")
import requests

@functools.lru_cache()
def get_url_im(t):
    user_agent = {'User-agent': 'gradio-app'}
    response = requests.get(t, headers=user_agent)
    return (BytesIO(response.content))


def inference(image):
  im_path = get_url_im(image)
  im = Image.open((im_path))
  

  return im, im , im


title = "Bg remover for sarvm catalog"
description = "Bg remover for sarvm catalog"
article = "<div><center><img src='https://visitor-badge.glitch.me/badge?page_id=max_skobeev_dis_cmp_public' alt='visitor badge'></center></div>"

interface = gr.Interface(
    fn=inference,
    inputs=gr.Textbox(label="Text or Image URL", interactive=True),
    outputs=["image", "image", "image"],
    title=title ,
    description=description,
    article=article,
    allow_flagging='never',
    cache_examples=False,
    ).queue().launch(show_error=True, share = True)