from client.AIGCClient import AIGCClient

aigc_client = AIGCClient(token="8bf14f27fc3a486788f3383452422d72", base_url="https://aigc-api.aiddit.com")


def main():
    produce_plan_ids = [
        "20260509113208212653909",
        "20260509113221856771441",
        "20260509113329478111972",
        "20260509113453897708339",
        "20260509113509961697272",
        "20260509113531215757335",
        "20260509113756891904084",
        "20260509113817127161918",
        "20260509113833884726698",
        "20260509113937229233085"
    ]
    for produce_plan_id in produce_plan_ids:
        # produce_plan = aigc_client.get_produce_plan_by_id(produce_plan_id)
        # inputSources = produce_plan['inputSourceGroups'][0]['inputSources']
        # produce_plan['inputSourceGroups'][0]['inputSources'] = [inputSources[0]]

        # aigc_client.save_produce_plan(produce_plan)

        print(produce_plan_id)


if __name__ == '__main__':
    main()