Update app.py

app.py

@@ -16,7 +16,6 @@ def preprocess_image(image):
   bin_image = cv.copyMakeBorder(bin_image, int(0.10 * image.shape[0]), int(0.05 * image.shape[0]), int(0.05 * image.shape[1]), int(0.10 * image.shape[1]), cv.BORDER_CONSTANT, value=(255, 255, 255))
   return bin_image
 
-#bin_image = preprocess_image(image)
 
 def split_image_into_lines(image):
   lines = []
@@ -52,7 +51,6 @@ def split_image_into_lines(image):
 
   return lines
 
-#lines = split_image_into_lines(bin_image)
 
 
 def generate_text(line):
@@ -119,24 +117,19 @@ max_width = 0.9
 out_image_width = 1500
 top_margin = 100
 
-#out_image = put_text(improved_text, font, font_scale, color, thickness, max_width, out_image_width, top_margin)
 
 def predict(input_path):
     image = cv.imread(input_path)
     bin_image = preprocess_image(image)
-    print("Preprocessing Complete!")
     lines = split_image_into_lines(bin_image)
-    print("Splitting complete!")
     improved_text = get_improved_result(lines)
-    print("Improvement done!")
     out_image = put_text(improved_text, font, font_scale, color, thickness, max_width, out_image_width, top_margin)
-    print("Out image generated!")
-    return out_image
+    return out_image, improved_text
 
 gradio_app = gr.Interface(
     predict,
-    inputs=gr.Image(label= '', sources=['upload', 'webcam'], type='filepath'),
-    outputs=[gr.Image(label= '')],
+    inputs=gr.Image(sources=['upload', 'webcam'], type='filepath'),
+    outputs=[gr.Image(), gradio.Textbox()],
     title="Extract Handwritten Text",
 )