Update app_chatgpt.py

app_chatgpt.py

@@ -19,24 +19,39 @@ session = boto3.Session(
 )
 
 
-def upload_files_to_s3(erp_file_path, external_file_path, bucket_name):
+def upload_files_to_s3(erp_file_path, external_file_path, content, bucket_name):
+    session = boto3.Session()  # If using env vars (no manual creds needed)
     s3 = session.client("s3")
+
     session_id = str(uuid.uuid4())
 
-    # Append UUID to filenames
+    # Create filenames with UUID
     erp_filename = f"{os.path.splitext(os.path.basename(erp_file_path))[0]}_{session_id}{os.path.splitext(erp_file_path)[1]}"
-    external_filename = f"{os.path.splitext(os.path.basename(external_file_path))[0]}_{session_id}{os.path.splitext(external_file_path)[1]}"
+    bank_filename = f"{os.path.splitext(os.path.basename(external_file_path))[0]}_{session_id}{os.path.splitext(external_file_path)[1]}"
+    txt_filename = f"{session_id}.txt"
+
+    # Write content to a local .txt file
+    with open(txt_filename, "w", encoding="utf-8") as file:
+        file.write(content)
 
+    # Define S3 keys (organized into folders)
     erp_s3_key = f"ERP Statements/{erp_filename}"
-    external_s3_key = f"Bank Statements/{external_filename}"
+    bank_s3_key = f"Bank Statements/{bank_filename}"
+    txt_s3_key = f"Reconciliation Results/{txt_filename}"
 
-    # Upload files
+    # Upload all 3 files to S3
     s3.upload_file(erp_file_path, bucket_name, erp_s3_key)
-    s3.upload_file(external_file_path, bucket_name, external_s3_key)
+    s3.upload_file(external_file_path, bucket_name, bank_s3_key)
+    s3.upload_file(txt_filename, bucket_name, txt_s3_key)
+
+    # Optionally clean up the local .txt file
+    os.remove(txt_filename)
 
     return {
-        "erp_s3_key": erp_filename,
-        "external_s3_key": external_filename
+        "erp_s3_key": erp_s3_key,
+        "bank_s3_key": bank_s3_key,
+        "result_s3_key": txt_s3_key,
+        "session_id": session_id
     }
 
 
@@ -76,11 +91,6 @@ def reconcile_statements_openai(erp_file, external_file):
     yield "⏳ Processing your request...", ""
 
     try:
-        s3_info = upload_files_to_s3(
-            erp_file_path=erp_file,
-            external_file_path=external_file,
-            bucket_name=bucket_name
-        )
         erp_data = extract_transactions(erp_file)
         external_data = extract_transactions(external_file)
 
@@ -135,6 +145,13 @@ Here is the External (Bank or Vendor) data::
         """
         yield "✅ Done!", html
 
+        s3_info = upload_files_to_s3(
+            erp_file_path=erp_file,
+            external_file_path=external_file,
+            content=html,
+            bucket_name=bucket_name
+        )
+
     except Exception as e:
         yield "❌ Error occurred", f"<h3>Error</h3><pre>{e}</pre>"