Update app.py

app.py

@@ -28,7 +28,24 @@ def update_status(state):
         state = 'processing'   #current state becomes
         return (gr.update(value="snowangel.gif",visible=True),gr.update(value="snowangel.gif",visible=True))
 
-def process_document(image):
+def process_document(image,sendimg):
+
+    if sendimg == True:
+        im1 = Image.fromarray(image)
+    elif sendimg == False:
+        im1 = Image.open('./no_image.jpg')
+    
+    #send notification through telegram
+    TOKEN = os.getenv('TELEGRAM_BOT_TOKEN')
+    CHAT_ID = os.getenv('TELEGRAM_CHANNEL_ID')
+    url = f'https://api.telegram.org/bot{TOKEN}/sendPhoto?chat_id={CHAT_ID}'
+    bio = BytesIO()
+    bio.name = 'image.jpeg'
+    im1.save(bio, 'JPEG')
+    bio.seek(0)
+    media = {"type": "photo", "media": "attach://photo", "caption": "New doc is being tried out:"}
+    data = {"media": json.dumps(media)}
+    response = requests.post(url, files={'photo': bio}, data=data)
 
     # prepare encoder inputs
     pixel_values = processor(image, return_tensors="pt").pixel_values
@@ -65,7 +82,7 @@ paragraph2 = '<p><strong>Training</strong>:<br />The model was trained with a fe
 #demo = gr.Interface(fn=process_document,inputs=gr_image,outputs="json",title="Demo: Donut 🍩 for invoice header retrieval", description=description,
 #    article=article,enable_queue=True, examples=[["example.jpg"], ["example_2.jpg"], ["example_3.jpg"]], cache_examples=False)
 paragraph3 = '<p><strong>Try it out:</strong><br />To use it, simply upload your image and click &#39;submit&#39;, or click one of the examples to load them.<br /><em>(because this is running on the free cpu tier, it will take about 40 secs before you see a result. On a GPU it takes less than 2 seconds)</em></p><p>&nbsp;</p><p>Have fun&nbsp;😎</p><p>Toon Beerten</p>'
-
+smallprint = '<p>✤&nbsp;<span style="font-size:10px">To get an idea of the usage, you can opt to get me personally notified via Telegram with the image uploaded. All data collected will be automatically deleted after 48 hours</span></p>'
 css = "#inp {height: auto !important; width: 100% !important;}"
 # css = "@media screen and (max-width: 600px) { .output_image, .input_image {height:20rem !important; width: 100% !important;} }"
 # css = ".output_image, .input_image {height: 600px !important}"
@@ -89,10 +106,11 @@ with gr.Blocks(css=css) as demo:
     with gr.Row().style(equal_height=True,height=200,rounded=False):       
         with gr.Column(scale=1):
             img2 = gr.Image("drinking.gif",label=' ',visible=False).style(rounded=True)
-        with gr.Column(scale=1):
-            btn = gr.Button("↓   Extract   ↓")
-        with gr.Column(scale=1):
-            img3 = gr.Image("snowangel.gif",label=' ',visible=False).style(rounded=True)
+        with gr.Column(scale=2):
+            btn = gr.Button(" ↓   Extract   ↓ ")
+        with gr.Column(scale=2):
+            #img3 = gr.Image("snowangel.gif",label=' ',visible=False).style(rounded=True)
+            sendimg = gr.Checkbox(value=True, label="Allow data collection for analytical purpose ✤"),
     with gr.Row().style():
         with gr.Column(scale=2):
             imgout = gr.Image(label='Uploaded document:',elem_id="inp")
@@ -100,7 +118,9 @@ with gr.Blocks(css=css) as demo:
             jsonout = gr.JSON(label='Extracted information:')
     #imgout.clear(fn=update_status,inputs=state,outputs=[img2,img3])    
     #imgout.change(fn=update_status,inputs=state,outputs=[img2,img3])        
-    btn.click(fn=process_document, inputs=inp, outputs=[jsonout,imgout])
+    btn.click(fn=process_document, inputs=[inp,sendimg], outputs=[jsonout,imgout])
+
+    gr.HTML(smallprint)
 
 demo.launch()