api_request_example.py

import httpx

train_url = "http://localhost:80/train"
predict_url = "http://localhost:80/predict"
get_url = "http://localhost:80/get_items"

payload = {
    "user_ids": [1, 2, 3, 4]
}

async def main():
    async with httpx.AsyncClient() as client:
        try:
            # Sending a GET request
            response = await client.get(get_url)

            # Checking if the request was successful (status code 200)
            if response.status_code == 200:
                # Printing the response content
                print("Response Content:")
                print(response.json())  # Assuming the response contains JSON data
            else:
                print(f"Request failed with status code: {response.status_code}")

        except httpx.RequestError as e:
            print("Error making the request:", e)

        a = input('run prediction? (Y/n)\n')
        if a not in ("Y", 'y'):
            return

        # Make the prediction
        response = await client.post(url=predict_url, json=payload)

        if response.status_code == 200:
            predictions = response.json()
            print(predictions)
        else:
            print(f"Prediction request failed with status code: {response.status_code}. Reason: {response.content}")

        a = input('run train? (Y/n)\n')
        if a not in ("Y", 'y'):
            return

        # Train the model
        response = await client.post(url=train_url, json={})

        if response.status_code == 200:
            status = response.json()
            print(status)
            if status.get('success') == 1:
                print("Training successful, proceeding with prediction.")
            else:
                print("Training failed, cannot proceed with prediction.")
        else:
            print(f"Training request failed with status code: {response.status_code}")

import asyncio
asyncio.run(main())