Updated output file creation variables for Lambda direct redaction runs

entrypoint_router.py

@@ -1,22 +0,0 @@
-import os
-
-if __name__ == "__main__":
-    run_direct_mode = os.getenv("RUN_DIRECT_MODE", "0")
-
-    if run_direct_mode == "1":
-
-        print("Attempting to import lambda_handler from lambda_entrypoint")
-        # Invoke the lambda handler
-        from lambda_entrypoint import lambda_handler
-
-        print("Imported lambda_handler from lambda_entrypoint")
-
-    else:
-        # Gradio App execution
-        from app import app, max_queue_size, max_file_size  # Replace with actual import if needed
-        from tools.auth import authenticate_user
-
-        if os.getenv("COGNITO_AUTH", "0") == "1":
-            app.queue(max_size=max_queue_size).launch(show_error=True, auth=authenticate_user, max_file_size=max_file_size)
-        else:
-            app.queue(max_size=max_queue_size).launch(show_error=True, inbrowser=True, max_file_size=max_file_size)

lambda_entrypoint.py

@@ -106,11 +106,15 @@ def lambda_handler(event, context):
             print(f"Unexpected error: {str(e)}")
             raise e
 
+        print("Now uploading files from:", output_dir)
+        
         # Upload output files back to S3
         for root, _, files in os.walk(output_dir):
             for file_name in files:
+                print("file_name:", file_name)
                 local_file_path = os.path.join(root, file_name)
-                output_key = f"{os.path.dirname(input_key)}/output/{file_name}"
+                output_key = f"output/{file_name}"
+                print("Output location is:", output_key)
                 upload_file_to_s3(local_file_path, bucket_name, output_key)
 
     return {"statusCode": 200, "body": "Processing complete."}