app.py

import gradio as gr
import os
import cv2 
import shutil
import sys
from subprocess import call
from utils import Restoration

os.system("pip install dlib")
os.system('bash setup.sh')

def run_cmd(command):
    try:
        call(command, shell=True)
    except KeyboardInterrupt:
        print("Process interrupted")
        sys.exit(1)

def run(image):
    # os.makedirs("Temp")
    # os.makedirs("Temp/input")
    # print(type(image))
    # cv2.imwrite("Temp/input/input_img.png", image)

    # command = ("python run.py --input_folder "
    #         + "Temp/input"
    #         + " --output_folder "
    #         + "Temp"
    #         + " --GPU "
    #         + "-1"
    #         + " --with_scratch")
    # run_cmd(command)

    # result = cv2.imread("Temp/final_output/input_img.png")
    # shutil.rmtree("Temp")

    result = Restoration(image)

    return result
    
iface = gr.Interface(fn=run, inputs="image", outputs="image").launch(debug=True, share=True)