Update app.py

app.py

@@ -92,7 +92,9 @@ 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.")
     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',
@@ -103,6 +105,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")
 
     return {'extracted_text': extracted_text}
 
@@ -120,6 +123,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.")
     
     categorization_prompt = [
         SystemMessage(content=categorization_system_message),
@@ -134,7 +138,9 @@ 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.")
     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: 
@@ -164,6 +170,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")
 
     total_bill = 0
     total_taxes_and_charges = 0
@@ -183,6 +190,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")
     
     final_output_system_message = """ 
         You are an expert accountant tasked to generate the expense claim report.
@@ -272,6 +280,8 @@ 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")
+    
     pdf = MarkdownPdf(toc_level=3)
 
     pdf.add_section(Section(output['final_output']))