Update app.py

app.py

@@ -6,13 +6,19 @@ import gradio as gr
 import torch
 import easyocr
 
-torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/english.png', 'english.png')
-torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/thai.jpg', 'thai.jpg')
-torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/french.jpg', 'french.jpg')
-torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/chinese.jpg', 'chinese.jpg')
-torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/japanese.jpg', 'japanese.jpg')
-torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/korean.png', 'korean.png')
-torch.hub.download_url_to_file('https://i.imgur.com/mwQFd7G.jpeg', 'Hindi.jpeg')
+# Download examples
+urls = {
+    "english.png": "https://github.com/JaidedAI/EasyOCR/raw/master/examples/english.png",
+    "thai.jpg": "https://github.com/JaidedAI/EasyOCR/raw/master/examples/thai.jpg",
+    "french.jpg": "https://github.com/JaidedAI/EasyOCR/raw/master/examples/french.jpg",
+    "chinese.jpg": "https://github.com/JaidedAI/EasyOCR/raw/master/examples/chinese.jpg",
+    "japanese.jpg": "https://github.com/JaidedAI/EasyOCR/raw/master/examples/japanese.jpg",
+    "korean.png": "https://github.com/JaidedAI/EasyOCR/raw/master/examples/korean.png",
+    "Hindi.jpeg": "https://i.imgur.com/mwQFd7G.jpeg"
+}
+
+for filename, url in urls.items():
+    torch.hub.download_url_to_file(url, filename)
 
 def draw_boxes(image, bounds, color='yellow', width=2):
     draw = ImageDraw.Draw(image)
@@ -24,59 +30,55 @@ def draw_boxes(image, bounds, color='yellow', width=2):
 def inference(img_path, lang):
     reader = easyocr.Reader(lang)
     bounds = reader.readtext(img_path)
+    
     im = PIL.Image.open(img_path)
     draw_boxes(im, bounds)
-    im.save('result.jpg')
+    im.save("result.jpg")
 
-    # return dataframe with only text + confidence
     df = pd.DataFrame(bounds)[[1, 2]]
     df.columns = ["text", "confidence"]
+    
+    return "result.jpg", df
 
-    return 'result.jpg', df
-
-title = 'EasyOCR'
-description = """
-EasyOCR demo supports 80+ languages.  
-Upload an image and choose a language to extract text.
-"""
-article = "<p style='text-align: center'><a href='https://www.jaided.ai/easyocr/'>EasyOCR Website</a> | <a href='https://github.com/JaidedAI/EasyOCR'>GitHub Repo</a></p>"
+choices = [
+    "abq","ady","af","ang","ar","as","ava","az","be","bg","bh","bho","bn","bs","ch_sim","ch_tra",
+    "che","cs","cy","da","dar","de","en","es","et","fa","fr","ga","gom","hi","hr","hu","id","inh","is",
+    "it","ja","kbd","kn","ko","ku","la","lbe","lez","lt","lv","mah","mai","mi","mn","mr","ms","mt","ne",
+    "new","nl","no","oc","pi","pl","pt","ro","ru","rs_cyrillic","rs_latin","sck","sk","sl","sq","sv","sw",
+    "ta","tab","te","th","tjk","tl","tr","ug","uk","ur","uz","vi"
+]
 
 examples = [
-    ['english.png', ['en']],
-    ['thai.jpg', ['th']],
-    ['french.jpg', ['fr', 'en']],
-    ['chinese.jpg', ['ch_sim', 'en']],
-    ['japanese.jpg', ['ja', 'en']],
-    ['korean.png', ['ko', 'en']],
-    ['Hindi.jpeg', ['hi', 'en']]
+    ["english.png", ["en"]],
+    ["thai.jpg", ["th"]],
+    ["french.jpg", ["fr", "en"]],
+    ["chinese.jpg", ["ch_sim", "en"]],
+    ["japanese.jpg", ["ja", "en"]],
+    ["korean.png", ["ko", "en"]],
+    ["Hindi.jpeg", ["hi", "en"]]
 ]
 
-css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
+with gr.Blocks(css=".output_image, .input_image {height: 40rem; width: 100%;}") as demo:
 
-choices = [
-    "abq","ady","af","ang","ar","as","ava","az","be","bg","bh","bho","bn","bs","ch_sim","ch_tra","che",
-    "cs","cy","da","dar","de","en","es","et","fa","fr","ga","gom","hi","hr","hu","id","inh","is","it",
-    "ja","kbd","kn","ko","ku","la","lbe","lez","lt","lv","mah","mai","mi","mn","mr","ms","mt","ne",
-    "new","nl","no","oc","pi","pl","pt","ro","ru","rs_cyrillic","rs_latin","sck","sk","sl","sq","sv",
-    "sw","ta","tab","te","th","tjk","tl","tr","ug","uk","ur","uz","vi"
-]
+    gr.Markdown("# **EasyOCR**")
+    gr.Markdown("Upload an image and select languages to extract text.")
+
+    with gr.Row():
+        image_input = gr.Image(type="filepath", label="Input Image")
+        lang_input = gr.CheckboxGroup(choices=choices, value=["en"], label="Language")
+
+    run_btn = gr.Button("Run OCR")
+
+    with gr.Row():
+        output_img = gr.Image(label="Output")
+        output_df = gr.Dataframe(headers=["text", "confidence"], label="Detected Text")
+
+    gr.Examples(examples, [image_input, lang_input], [output_img, output_df], fn=inference)
 
-demo = gr.Interface(
-    fn=inference,
-    inputs=[
-        gr.Image(type="filepath", label="Input Image"),
-        gr.CheckboxGroup(choices=choices, value=["en"], label="Language")
-    ],
-    outputs=[
-        gr.Image(label="Output"),
-        gr.Dataframe(label="Detected Text")
-    ],
-    title=title,
-    description=description,
-    article=article,
-    examples=examples,
-    css=css,
-    enable_queue=True
-)
+    run_btn.click(
+        inference,
+        inputs=[image_input, lang_input],
+        outputs=[output_img, output_df]
+    )
 
-demo.launch(debug=True)
+demo.launch()