Update app.py

app.py

@@ -379,24 +379,25 @@ def postprocess_dataframes(result_tables):
     for idx, table_df in enumerate(result_tables):
         result_df = pd.DataFrame()
         print("--1")
-        print(result_df)
+        print(table_df)
         print("--2")
-        print(result_df.to_json(orient="records"))
-        # for col in table_df.columns:
-        #     if col.lower().startswith("item"):
-        #         result_df["name"] = table_df[col].copy()
-        #     if (
-        #         col.lower().startswith("total")
-        #         or col.lower().startswith("amount")
-        #         or col.lower().startswith("cost")
-        #     ):
-        #         result_df["amount"] = table_df[col].copy()
-        print("--3")
-        print(result_df.columns)
+        print(table_df.to_json(orient="records"))
+        for col in table_df.columns:
+            if col.lower().startswith("item"):
+                result_df["name"] = table_df[col].copy()
+            if (
+                col.lower().startswith("total")
+                or col.lower().startswith("amount")
+                or col.lower().startswith("cost")
+            ):
+                result_df["amount"] = table_df[col].copy()
+        
         if len(result_df.columns) == 0:
             result_df["name"] = table_df.iloc[:, 0].copy()
             result_df["amount"] = table_df.iloc[:, 1].copy()
-
+        print("--3")
+        print(result_df.columns)
+        
         result_df["cost_code"] = ""
         # res["Table" + str(idx)] = result_df.to_json(orient="records")
         res=result_df.to_json(orient="records")