Create app.py

app.py

@@ -0,0 +1,63 @@
+import pytesseract
+from PIL import Image
+import pandas as pd
+import os
+import gradio as gr
+import numpy as np
+import gradio as gr
+
+
+def ocr_df_using_pytesseract(image):
+    pytesseract.pytesseract.tesseract_cmd =r"C:\Users\amold\Desktop\Upwork\pdf to image and pytesseract\tesseact_exe\Tesseract-OCR\tesseract.exe"
+
+    #image = Image.open(example['image_path'])
+
+    width, height = image.size
+
+    # apply ocr to the image 
+    ocr_df = pytesseract.image_to_data(image, output_type='data.frame')
+    float_cols = ocr_df.select_dtypes('float').columns
+    ocr_df = ocr_df.dropna().reset_index(drop=True)
+    ocr_df[float_cols] = ocr_df[float_cols].round(0).astype(int)
+    ocr_df = ocr_df.replace(r'^\s*$', np.nan, regex=True)
+    ocr_df = ocr_df.dropna().reset_index(drop=True)
+
+    ocr_df
+
+    ocr_df['X1']=ocr_df['left']
+
+    ocr_df['Y1']=ocr_df['top']
+
+    ocr_df['X2']= ocr_df['left'] + ocr_df['width']
+
+    ocr_df['Y2']= ocr_df['top'] + ocr_df['height']
+
+    return ocr_df
+
+
+def image_to_text(image):
+    ocr_df= ocr_df_using_pytesseract(image)
+
+
+    grouped_text = ocr_df.groupby(['block_num', 'line_num'])['text'].agg(' '.join).reset_index()
+
+    # sort the text by line numbers within each block
+    grouped_text = grouped_text.sort_values(['block_num', 'line_num'])
+
+    # join the text by blocks and add newlines
+    result = ''
+    for i, row in grouped_text.iterrows():
+        if i > 0 and row['block_num'] != grouped_text.loc[i-1, 'block_num']:
+            result += '\n\n'
+        result += row['text'].rstrip() + '\n'
+
+    return result
+
+
+
+demo = gr.Interface(fn=image_to_text, 
+                    inputs= gr.Image(type="pil"), 
+                    outputs=["text"],
+                    title="Menu image to text",
+                    description= "Upload Menu image here")
+demo.launch(share=True)