Update app.py

app.py

@@ -92,9 +92,8 @@ def text_extractor(state: State):
     ]
 
     print("I have access to the Open AI API for text extraction.")
-    gr.Info("I have access to the Open AI API for text extraction.")
+    gr.Info("I have access to the Open AI API for text extraction. I need to format the input image in the Open AI format", duration=2)
     print("I need to format the input image in the Open AI format")
-    gr.Info("I need to format the input image in the Open AI format")
 
     response = client.chat.completions.create(
         model='gpt-4o-mini',
@@ -105,7 +104,7 @@ def text_extractor(state: State):
     extracted_text = response.choices[0].message.content
 
     print("Extracted text from the input image")
-    gr.Info("Extracted text from the input image")
+    gr.Info("Extracted text from the input image", duration=2)
 
     return {'extracted_text': extracted_text}
 
@@ -123,7 +122,7 @@ def categorizer(state: State):
         """
 
     print("Categorizing items in the input text to one of: Alcoholic Drinks, Non-Alcoholic Drinks and Food.")
-    gr.Info("Categorizing items in the input text to one of: Alcoholic Drinks, Non-Alcoholic Drinks and Food.")
+    gr.Info("Categorizing items in the input text to one of: Alcoholic Drinks, Non-Alcoholic Drinks and Food.", duration=1)
     
     categorization_prompt = [
         SystemMessage(content=categorization_system_message),
@@ -138,9 +137,8 @@ def categorizer(state: State):
 def verifier(state: State):
 
     print("I now have to retrieve relevant sections of the company policy to exclude items that are not reimbursable.")
-    gr.Info("I now have to retrieve relevant sections of the company policy to exclude items that are not reimbursable.")
+    gr.Info("I now have to retrieve relevant sections of the company policy to exclude items that are not reimbursable. I will use the search tool to execute this step", duration=2)
     print("I will use the search tool to execute this step.")
-    gr.Info("I will use the search tool to execute this step.")
 
     for document in company_policy_document:
         if document.page_content.find(f'{type_of_expense}') != -1: 
@@ -170,7 +168,7 @@ def verifier(state: State):
 def estimator(state: State):
 
     print("Calculating the revised total amount and taxes")
-    gr.Info("Calculating the revised total amount and taxes")
+    gr.Info("Calculating the revised total amount and taxes", duration=1)
 
     total_bill = 0
     total_taxes_and_charges = 0
@@ -190,7 +188,7 @@ def estimator(state: State):
 
 def formatter(state: State):
     print("Formatting the output into a markdown file")
-    gr.Info("Formatting the output into a markdown file")
+    gr.Info("Formatting the output into a markdown file", duration=1)
     
     final_output_system_message = """ 
         You are an expert accountant tasked to generate the expense claim report.
@@ -280,7 +278,7 @@ def claim_generator(input_bill_path, progress=gr.Progress()):
     output = chain.invoke({'image_path': input_bill_path})
     progress(0.25)
 
-    print("Converting the markdown file to pdf")
+    gr.Info("Converting the markdown file to pdf", duration=2)
     
     pdf = MarkdownPdf(toc_level=3)