Update app.py

app.py

@@ -169,6 +169,27 @@ def flag_if_shared(flag_method):
             return flag_method(request, *args, **kwargs)
     return proxy
 
+def process_and_output_files(input_files):
+    output_data = []
+
+    for file in input_files:
+        # Extract and categorize text for each file
+        text = extract_text(file)
+        category = categorize_text(text)
+        chatbot_response = query(category, text)
+        parsed_info = parse(category, chatbot_response)
+
+        # Append the relevant data for this file to the output_data list
+        output_data.append({
+            "File Name": file.name,
+            "Extracted Text": text,
+            "Category": category,
+            "Chatbot Response": chatbot_response[0][1],
+            "Parsed Information": parsed_info,
+        })
+
+    return output_data
+
 
 with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
     gr.Markdown("<center><h1>Automatic Reimbursement Tool Demo</h1></center>")
@@ -197,7 +218,7 @@ with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
                     label="Input receipt",
                     show_label=True,
                     type="file",
-                    file_count="single",
+                    file_count="multiple",
                     file_types=["image", ".pdf"],
                 )
                 input_file.change(
@@ -295,26 +316,9 @@ with gr.Blocks(title="Automatic Reimbursement Tool Demo") as page:
                 [input_file],
                 [extracted_text, category, chatbot, information],
             ).then(
-                extract_text,
+                process_and_output_files,
                 [input_file],
-                [extracted_text],
-            ).then(
-                categorize_text,
-                [extracted_text],
-                [category],
-            ).then(
-                query,
-                [category, extracted_text],
-                [chatbot],
-                queue=True,
-            ).then(
-                parse,
-                [category, chatbot],
                 [information],
-            ).then(
-                activate_flags,
-                None,
-                [flag_incorrect_button, flag_irrelevant_button],
             ).then(
                 flag_if_shared(flag_method),
                 [