Update app.py

app.py

@@ -28,6 +28,23 @@ For example:
 """
 
 txt_display_1 = 'content of the letter: '
+
+
+txt_prompt_2 = """The image contains an inventory stock level table with Item Id, Quantity, ROQ, Item Images, and Contact Phone. For reference, ROQ stands for Replenishment Order Quantity. 
+It is for use of reorder, that is, when quantity level is below ROQ, a reorder process will need to take place. Please provide a list of reorder items for all below ROQ. When responding please follow the below formats:
+
+** ROQ ({today's date})**
+Item ID, Shortage (Quantity - ROQ), Contact Phone
+(for all below ROQ items)
+
+For example:
+** ROQ (02/21/2024) **
+#11608, 70 (30-100), 1-858-7331029
+#61785, 5 (5-10), 1-858-1233
+
+"""
+txt_display_2 = 'today's inventory stock level: '
+
 import os
 GOOGLE_API_KEY=os.getenv('GOOGLE_API_KEY')
 genai.configure(api_key=GOOGLE_API_KEY)
@@ -52,6 +69,17 @@ def app1_response(img):
         response = vis_model.generate_content([txt_prompt_1,img])
         return response.text
 
+def app2_response(img):
+    if not img:
+        response = txt_model.generate_content(txt_prompt_2)
+        return response
+
+    else:
+        img = PIL.Image.open(img)
+        response = vis_model.generate_content([txt_prompt_2,img])
+        return response.text
+
+
 # SMS service ends in March 2024, to restore service @Sinch Simple Text
 def send_SMS(resp_text):
     url = SMS_URL
@@ -72,8 +100,8 @@ def send_SMS(resp_text):
 # gradio block
 with gr.Blocks() as app1:
     with gr.Column():
-        gr.Markdown("```for email and/or instruction, paste screenshot here...```")
-        image_box = gr.Image(type="filepath")
+        gr.Markdown("```for email and/or assigment descriptions, paste screenshot here...```")
+        image_box = gr.Image(label="email screen", type="filepath")
         btn1 = gr.Button("generate To-Dos")
         out1 = gr.Textbox(label="here are the actionables...")
         btn2 = gr.Button("Send to Mobile")
@@ -97,13 +125,13 @@ with gr.Blocks() as app1:
 with gr.Blocks() as app2:
     with gr.Column():
         gr.Markdown("```Stock-Out Shamer, paste ERP Inv screenshot here...```")
-        image_box = gr.Image(type="filepath")
+        image_box = gr.Image(label="ERP/Inventory screen",type="filepath")
         btn1 = gr.Button("check ROQ")
         out1 = gr.Textbox(label="here is the watch list...")
         btn2 = gr.Button("send out reminders")
         out2 = gr.Textbox(label="response or feed back?")
     
-        btn1.click(fn=app1_response, inputs=[image_box], outputs=out1)
+        btn1.click(fn=app2_response, inputs=[image_box], outputs=out1)
         btn2.click(fn=send_SMS, inputs=out1, outputs=out2)
 
     gr.Markdown("""