patient_id can be informed manually or by image

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-#from pyzbar.pyzbar import decode
+from pyzbar.pyzbar import decode
 from lambdas import upload_models, predict
 import base64
 from io import BytesIO, StringIO
@@ -32,7 +32,7 @@ def login(username, password) -> bool:
             restaurant_name = str(row['restaurant_name'])
             mode = 'waste'
             possible_modes = str(row.get('modes')).split(':')
-            possible_shifts = {i.split(':')[0]: i.split(':')[1] for i in str(row.get('shifts')).split('-')} \
+            possible_shifts = {i.split(':')[0]: int(i.split(':')[1]) for i in str(row.get('shifts')).split('-')} \
                 if row.get('shifts') else {'no shift': None}
 
             config_aux = {'restaurant_id': restaurant_id,
@@ -113,6 +113,7 @@ with (gr.Blocks(head=prefer_frontal_cam_html, css=css_image_aspect_ratio) as blo
 
 
             gr.Markdown(""" * Do you have any doubt? Please, see the documentation tab. \n
+                            * Tienes alguna duda? Consulta la pestaña de documentación
                             * Tens un dubte? Consulta la pestanya de documentació. """)
 
             logout_button = gr.Button(link='/logout', value='Logout / Exit ↩', size='sm')
@@ -129,9 +130,25 @@ with (gr.Blocks(head=prefer_frontal_cam_html, css=css_image_aspect_ratio) as blo
                           type='pil',
                           show_share_button=False,
                           show_download_button=False,
-                          height=720, width=1280, min_width=0,
+                          height='720px', width='1280px', min_width=0,
                           elem_id="image-container")
 
+
+            ## Patient ID searcher.
+            with gr.Accordion(open=False):
+                def get_barcode_from_image(image):
+                    if not image:
+                        raise gr.Error('No image has been taken yet')
+                    d = decode(image)
+                    if not d:
+                        raise gr.Error('Cannot find a barcode on the Image')
+                    barcode = d[0].data.decode('utf8') if d else None
+                    return barcode
+                get_barcode_button = gr.Button('Get Patient ID (Barcode) from Image')
+                eater_id = gr.Textbox(label='Patient ID', interactive=True, value=None,
+                                      placeholder='Write the patient id or take a photo')
+                get_barcode_button.click(fn=get_barcode_from_image, inputs=[im], outputs=eater_id)
+
             ## Patient ID searcher. Uncomment if needed
             # with gr.Accordion(open=False):
             #     eater_id = gr.Textbox(label='Patient Identification', placeholder='Searching Patient ID')
@@ -148,10 +165,10 @@ with (gr.Blocks(head=prefer_frontal_cam_html, css=css_image_aspect_ratio) as blo
             #     final_value = new_value if new_value != default_value else current_value
             #     current_eater_id['value'] = final_value
             #     return final_value
-            eater_id = None
+            # eater_id = None
 
             b = gr.Button('PRESS TO PREDICT')
-            b.click(fn=predict_app, inputs=[im], outputs=gr.Info()) # add eater id to input if needed
+            b.click(fn=predict_app, inputs=[im, eater_id], outputs=gr.Info())  # add eater id to input if needed
 
         with gr.Tab(label='ℹ️ Status', id=2):
             gr.Markdown(' Press the button to see the status of the Application and technical information')

requirements.txt

@@ -1,2 +1,3 @@
 pillow
 boto3
+pyzbar