main.py

from urllib import response
import boto3
import requests
import os
import time
import json

SESSION = boto3.Session(profile_name='jpmc')
REGION='us-east-2'
PRESIGNED_S3_URL = "https://gaina-demo-agentcore.s3.us-east-2.amazonaws.com/input/test-agentcore.txt?response-content-disposition=inline&X-Amz-Content-Sha256=UNSIGNED-PAYLOAD&X-Amz-Security-Token=IQoJb3JpZ2luX2VjENv%2F%2F%2F%2F%2F%2F%2F%2F%2F%2FwEaCXVzLWVhc3QtMiJHMEUCIQCz3DgB5ZpsUU5dJkiBtL%2Fbm4TtCAKevQ%2ByvKLiO3eVOAIgSeq1sX7ZkqoMFcUR%2BFtk8MR8PIpmz1ltJLn2PANB%2BWQq8QMIpP%2F%2F%2F%2F%2F%2F%2F%2F%2F%2FARAAGgw2NjgwNDE4ODQyNzEiDOqAzWpcT0lMc2YlPSrFA9OaBgSw52hhurt4dG18yk49XbtPXyTfmEkvU%2FP1rJngqV6litTag3EbYMHKHzyV4rQLlkOXdBCqldGcRlic1QBUwVgROzywGj6W0UvTsiFia%2FLHFaRxV3X9ktvVFwxjLQqXCOodzA%2BpHaPzExCDOZPY%2B3f93M2sRVr6FhQJYOJ0JCjlHK3s94%2BfCvVVblXyXF%2FOHwgiqrlfom7xBEQRgu5LVsvd6%2BIx0T5N9L3WNpdmI5A18bqWLOJPMsJrvibmAp3kYP0VGqjFr9VYT8K2r5DSIPHl3RBXz%2BVmggvVdbaBR1xSiSUQfj%2FbTcXaGvh%2FuTSQcg2m1hP7entPdRqhwTOXJQHwZhKV00tVGTdWfHIVoRwNmWBGMU%2BojknCXvYb4aIiu%2Fzevy%2FC%2F33fvcgWUlfSr3C8vakFBUUnw14heLabpV9bR3KWDkYdwu57I00xYQIJL%2Fw8qlI%2Br8%2FwClXLNlSb7kcABMEZeJWUy4qvB%2F0jTP61qPabJWFr5oyPoRsIDq7vZCHlJSgKwGEfKiZ6rGU6aCpE4INMvdGxOlZiVF0kZJYr5ywBVf5r3BW6TFyPFFdvFJ3yt%2B3s5pp0PfKhzcClB%2BCBCjCLlo3OBjq6Ak00h3xHXLSo0p0Xm60Zq8nuSCku1DrhW2%2F9TKnsKpscdUKuWv%2BJJgDclQy6SGDH1phylOHR860HvpshrMB%2F6Ic3kgyAioJ9xR0wHpPEqXfWAJz7hDGY1r3ppue5XocSlTP%2BptQFuPbdqVV%2FRvqCKtmzSKIfIj6aahT6QP69UVS6knMbBq7mCJG7GN48Y%2BJcJBKpE56%2BIghoFh1QKLCQZd%2FQtG749tZNThKOaeuunqFk1KMOkkyF15oSWHiyQoAt%2BpYb%2F4%2FDuvAtno78f5TNZW28ItmrfrD2mLC5gX8vxqxL6oehEcQTFOYL250NwfhQexv741tJM3lPlo9Uo2yiZqQvOtJo12MM5GBlefV69Y6Y9K0Xojmpxe4jjv8KG%2BpsH9uVw%2Buh18eheDVbp06FauWaS8MH7KlJcsvI&X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Credential=ASIAZXCTGGJX2M73V6IE%2F20260325%2Fus-east-2%2Fs3%2Faws4_request&X-Amz-Date=20260325T025842Z&X-Amz-Expires=43200&X-Amz-SignedHeaders=host&X-Amz-Signature=b6be5098b6fb304257569668fbd01e691525fb1bf9b768ac1f737a81064a26bc"
PUBLIC_S3 = "https://s3.amazonaws.com/globalnightlight/201204/SVDNB_npp_d20120404_t1831351_e1837155_b02261_c20120405003716697413_noaa_ops.rade9.co.json"


CODE = """
import requests
import os
import time

s3_url="<<S3_URL>>"


DOWNLOAD_URL = s3_url
WORKSPACE    = "."

# ── Download: GET the file from S3 via presigned URL ──────────────────────
print("📥 Downloading file from S3...")

response = requests.get(DOWNLOAD_URL, timeout=30)

if response.status_code != 200:
    raise RuntimeError(
        f"Download failed — HTTP {response.status_code}: {response.text[:200]}"
    )
else:
    print(f"Download successful! - {len(response.content)} bytes")

file_content = response.content.decode('utf-8')
print(file_content)
    
#local_input_path = os.path.join(WORKSPACE, f"data-{time.strftime('%d_%H-%M-%S')}.csv")
# local_input_path=f"data-{time.strftime('%d_%H-%M-%S')}.csv"
# with open(local_input_path, "wb") as f:
#     f.write(response.content)

# file_size_kb = len(response.content) / 1024
# print(f"Downloaded {file_size_kb:.1f} KB → {local_input_path}")
"""

def main(s3_url):
    print("Hello from jpmc-sandbox!")

    DOWNLOAD_URL = s3_url
    WORKSPACE    = "./workspace"

    # ── Download: GET the file from S3 via presigned URL ──────────────────────
    print("📥 Downloading file from S3...")

    response = requests.get(DOWNLOAD_URL, timeout=30)

    if response.status_code != 200:
        raise RuntimeError(
            f"Download failed — HTTP {response.status_code}: {response.text[:200]}"
        )

    local_input_path = os.path.join(WORKSPACE, f"data-{time.strftime('%d_%H-%M-%S')}.csv")
    with open(local_input_path, "wb") as f:

        f.write(response.content)

    file_size_kb = len(response.content) / 1024

def presigned(s3_url):
    modified_code = CODE.replace("<<S3_URL>>", s3_url)

    data_client = SESSION.client("bedrock-agentcore", region_name=REGION)

    session    = data_client.start_code_interpreter_session(
        codeInterpreterIdentifier="gaina_code_interpreter-Ui8myMn4G0",
        name="s3-download-session",
        sessionTimeoutSeconds=900
    )

    session_id = session["sessionId"]
    print(f"Session started: {session_id}\n{'─'*50}")


    # client = data_client.client('bedrock-agentcore', region_name='us-east-2')
    response = data_client.invoke_code_interpreter(
        codeInterpreterIdentifier='gaina_code_interpreter-Ui8myMn4G0',
        sessionId=session_id,  # 1-40 alphanumeric chars only
        name='executeCode',
        arguments={
            'code':modified_code,
            'language':'python'
        }
    )

    # Handle the event stream from Bedrock
    print(f"Session ID from response: {response['sessionId']}")
    
    # Collect all events from the stream
    events_data = []
    event_stream = response['stream']
    
    for event in event_stream:
        # Convert each event to a dictionary
        event_dict = dict(event)
        events_data.append(event_dict)
        
        if 'executionOutput' in event:
            output = event['executionOutput']
            print("Execution Output:", output)
        elif 'executionError' in event:
            error = event['executionError']
            print("Execution Error:", error)
        elif 'sessionState' in event:
            state = event['sessionState']
            print("Session State:", state)
    
    # Convert to JSON
    response_json = json.dumps(events_data, indent=2, default=str)
    print("\nFull response as JSON:")
    print(response_json)
    #         print("Session State:", state)




if __name__ == "__main__":
    presigned(PRESIGNED_S3_URL)
    # presigned(PUBLIC_S3)
    # main(PUBLIC_S3)