Update app.py

app.py

@@ -90,6 +90,8 @@ def text_extractor(state: State):
         }
     ]
 
+    print("I have access to the Open AI API for text extraction.")
+    print("I need to format the input image in the Open AI format")
 
     response = client.chat.completions.create(
         model='gpt-4o-mini',
@@ -99,6 +101,8 @@ def text_extractor(state: State):
 
     extracted_text = response.choices[0].message.content
 
+    print("Extracted text from the input image")
+
     return {'extracted_text': extracted_text}
 
 def categorizer(state: State):
@@ -113,6 +117,8 @@ def categorizer(state: State):
         Do not output anything except the above fields in your JSON output.
         Do not delimit the JSON with any extra tags (e.g., ``` or ```JSON).
         """
+
+    print("Categorizing items in the input text to one of: Alcoholic Drinks, Non-Alcoholic Drinks and Food.")
     
     categorization_prompt = [
         SystemMessage(content=categorization_system_message),
@@ -126,6 +132,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.")
+    print("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: 
             relevant_company_policy = document.page_content
@@ -153,6 +162,8 @@ def verifier(state: State):
 
 def estimator(state: State):
 
+    print("Calculating the revised total amount and taxes")
+
     total_bill = 0
     total_taxes_and_charges = 0
 
@@ -170,6 +181,8 @@ def estimator(state: State):
 
 
 def formatter(state: State):
+    print("Formatting the output into a markdown file")
+    
     final_output_system_message = """ 
         You are an expert accountant tasked to generate the expense claim report.
         Generate the expense claim report based on the calculated total amount to be reimbursed and other details available to you.