Add more logs

app.py

@@ -80,23 +80,27 @@ def submit(input_file, old_text):
     if not input_file:
         gr.Error("Please upload a file to continue!")
         return gr.Textbox.update()
-
+    print("-"*5)
+    print("New input")
     # Send change to preprocessed image or to extracted text
     if input_file.name.endswith(".pdf"):
         text = process_pdf(Path(input_file.name), extract_only=True)
     else:
         text = process_image(Path(input_file.name), extract_only=True)
+    print("Extracted text")
     return text
 
 
 def categorize_extracted_text(extracted_text):
     category = categories.categorize_text(extracted_text)
+    print("Recognized category:", category)
     # gr.Info(f"Recognized category: {category}")
     return category
 
 
-def extract_from_category(category, extracted_text):
+def parse_from_category(category, extracted_text):
     if not category:
+        print("Updated with no category:", category)
         return (
             gr.Chatbot.update(None),
             gr.JSON.update(None),
@@ -104,7 +108,7 @@ def extract_from_category(category, extracted_text):
             gr.Button.update(interactive=False),
         )
     category = Category[category]
-    print("Extracting text from", category)
+    print("Parsing text from", category)
     chain = categories.category_modules[category].chain
     formatted_prompt = chain.prompt.format_prompt(
         text=extracted_text,
@@ -122,7 +126,9 @@ def extract_from_category(category, extracted_text):
     if len(formatted_prompt.messages) > 1:
         question += f"**System:**\n{formatted_prompt.messages[1].content}"
     question += f"\n\n**Human:**\n{formatted_prompt.messages[0].content}"
+    print("\tConstructed prompt")
     answer = result.generations[0][0].text
+    print("\tProcessed text")
     try:
         information = chain.output_parser.parse_with_prompt(answer, formatted_prompt)
         information = information.json() if information else {}
@@ -276,7 +282,7 @@ with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
                 [category],
             )
             category.change(
-                extract_from_category,
+                parse_from_category,
                 [category, extracted_text],
                 [chatbot, information, flag_incorrect_button, flag_irrelevant_button],
             )