import gradio as gr
import subprocess
import torch
import directories as Dir
import process as ps

#torch.hub.download_url_to_file('https://huggingface.co/spaces/Nebulae000/NoteCrawling/blob/main/noteimg2.jpg', 'noteimg2.jpg')

def HCR(im):
    ps.textDetection(im)
    ps.textRearrange()
    ps.textRecognition()
    hcr = ps.getHcrResult(Dir.txt_file_path)
    #subprocess.call('cd'+ Dir.yolo_dir, shell=True)
    #clearDir()

    return hcr

title = "HCR"
description = "[Postech X KITA] Elite Undergradute AI Strategy Program | AI Project Competition Team 2 OCR Part"
article = "<p style='text-align: center'> Handwritten Text Character Recognition Task \n" \
          "Text Detection + Text Rearrangement + Text Recognition\n" \
          "For More Information\n" \
            "<a href='https://bandalcom.github.io/'>GitBlog</a> | " \
          "<a href='https://github.com/bandalcom/Postech+KITA_AI-Project'>Source code</a></p>"

#<a href='https://github.com/bandalcom/Postech+KITA_AI-Project'>Source code</a>
#examples = [['zidane.jpg'], ['bus.jpg']]
torch.hub.download_url_to_file('https://ibb.co/JyMt31q', 'noteimg1.jpg')
examples = [['noteimg1.jpg']]
#examples=examples,
gr.Interface(fn=HCR, inputs="image", outputs = "text", title=title, description=description, examples=examples, article=article, analytics_enabled=False).launch(
    debug=True)