bounding box added

app.py

@@ -19,8 +19,11 @@ def get_response(input_img):
 
     '''
 
+    src_img = None
+    out_img = None
     if hasattr(input_img,'shape'):
         src_img=input_img.copy()
+        out_img=input_img.copy()
     outputs=text_detector.predict([input_img])
 
 
@@ -41,12 +44,12 @@ def get_response(input_img):
             # Find bounding box and crop
             x, y, w, h = cv2.boundingRect(pts)
             cropped = result[y:y + h, x:x + w, :]
-            # cv2.polylines(src_im, [box], True, color=(255, 255, 0), thickness=2)
+            cv2.polylines(out_img,[pts], True, color=(255, 255, 0), thickness=2)
             # cv2.imwrite(f"cropped/output_{i}_{j}.jpg",cropped)
             texts.append(ocr.predict([cropped])[0])
             j += 1
 
-    return "\n".join(texts)
+    return "\n".join(texts),out_img
 
 
 if __name__ == "__main__":
@@ -55,7 +58,7 @@ if __name__ == "__main__":
     iface = gr.Interface(
         fn=get_response,
         inputs=gr.Image(type="numpy"),  # Accepts image input
-        outputs=gr.Textbox(),
+        outputs=[gr.Textbox(),gr.Image(type="numpy")],
         title="Latin Text Detection & Recognition",
         description="Upload images for fast & accurate English(Latin) OCR. Works best with image format of invoice, documents"
     )