Update app_chatgpt.py

app_chatgpt.py

@@ -14,34 +14,50 @@ session = boto3.Session(
     aws_secret_access_key=os.getenv("secret_access_key")
 )
 
-# ✅ Upload reconciled results and original files to S3
 def upload_files_to_s3(erp_file_path, external_file_path, content, bucket_name):
     s3 = session.client("s3")
     session_id = str(uuid.uuid4())
 
-    erp_filename = f"{os.path.splitext(os.path.basename(erp_file_path))[0]}_{session_id}.csv"
-    bank_filename = f"{os.path.splitext(os.path.basename(external_file_path))[0]}_{session_id}.csv"
+    base_erp = os.path.splitext(os.path.basename(erp_file_path))[0]
+    base_bank = os.path.splitext(os.path.basename(external_file_path))[0]
+
+    # Add session ID to filenames
+    erp_filename = f"{base_erp}_{session_id}.csv"
+    bank_filename = f"{base_bank}_{session_id}.csv"
     txt_filename = f"{session_id}.txt"
 
     # Write reconciliation output to local file
     with open(txt_filename, "w", encoding="utf-8") as file:
         file.write(content)
 
-    # Define S3 keys
+    # === Separate folders by type ===
     erp_s3_key = f"ERP Statements/{erp_filename}"
     bank_s3_key = f"Bank Statements/{bank_filename}"
     txt_s3_key = f"Reconciliation Results/{txt_filename}"
 
+    # === Combined folder for each run ===
+    combined_prefix = f"Combined Files/{session_id}/"
+    erp_combined_key = combined_prefix + erp_filename
+    bank_combined_key = combined_prefix + bank_filename
+    txt_combined_key = combined_prefix + txt_filename
+
+    # ✅ Upload to type-based folders
     s3.upload_file(erp_file_path, bucket_name, erp_s3_key)
     s3.upload_file(external_file_path, bucket_name, bank_s3_key)
     s3.upload_file(txt_filename, bucket_name, txt_s3_key)
 
+    # ✅ Upload to combined folder
+    s3.upload_file(erp_file_path, bucket_name, erp_combined_key)
+    s3.upload_file(external_file_path, bucket_name, bank_combined_key)
+    s3.upload_file(txt_filename, bucket_name, txt_combined_key)
+
     os.remove(txt_filename)
 
     return {
         "erp_s3_key": erp_s3_key,
         "bank_s3_key": bank_s3_key,
         "result_s3_key": txt_s3_key,
+        "combined_keys": [erp_combined_key, bank_combined_key, txt_combined_key],
         "session_id": session_id
     }
 
@@ -178,4 +194,4 @@ with gr.Blocks(css="""
         outputs=[status, result]
     )
 
-iface.launch()
+iface.launch()