Update app.py

app.py

@@ -2395,43 +2395,46 @@ def identify_headers_and_save_excel(pdf_path, model,LLM_prompt):
             #             subject_body_map[subject.strip()] = " ".join(body)
 
             # df["body"] = df["text"].map(subject_body_map)
-
             subject_body_map = {}
-
+            
             def process_obj(obj):
                 if not isinstance(obj, dict):
                     return
-
+            
                 subject = obj.get("Subject")
                 body = obj.get("BodyText", [])
-
+            
                 if subject:
                     if isinstance(body, list):
                         body_text = " ".join(body)
                     else:
                         body_text = str(body)
-
+            
                     subject_body_map[subject.strip()] = body_text
-
-
+            
+            
             for item in jsons:
-
-                # Case: flat list of dicts (like your example)
+            
+                # Case: flat list of dicts
                 if isinstance(item, dict):
                     process_obj(item)
-
-                # Case: nested list of dicts
+            
+                # Case: nested list
                 elif isinstance(item, list):
                     for obj in item:
                         process_obj(obj)
-
-                    output_path = os.path.abspath("header_analysis_output.xlsx")
-                    df.to_excel(output_path, index=False, engine="openpyxl")
-                    
-                    print("--- Processed DataFrame ---")
-                    print(df)
             
-                    return output_path
+        
+            df["body"] = df["text"].map(subject_body_map).fillna("")
+            
+            output_path = os.path.abspath("header_analysis_output.xlsx")
+            df.to_excel(output_path, index=False, engine="openpyxl")
+            
+            print("--- Processed DataFrame ---")
+            print(df)
+            
+            return output_path
+
 
     except Exception as e:
         logger.error(f"Critical error in processing: {str(e)}")