main.py

from add_image_to_production import add_image_to_production
from checker import checker
from create_youtube_video import create_youtube_video
from dir_checker import dir_checker
from flask_client import flask_client
from image_generation_custom import image_generator_custom
from welcome import welcome


app, jsonify,request = flask_client()

INTERVAL = 5
MAXTOKENS = 2500

@app.route('/api/create/1', methods=['GET'])
def api_check1():
    url = request.args.get('url')
    if not url:
        return jsonify({"error": "url parameter is missing"}), 400
    channel_type = request.args.get('channel_type')
    if not channel_type:
        return jsonify({"error": "channel_type parameter is missing"}), 400
    dir_checker()
    checker()
    create_youtube_video(INTERVAL, MAXTOKENS, url, channel_type)
    return jsonify({"url": url}), 200

@app.route('/api/create/2', methods=['GET'])
def api_check2():
    channel_type = request.args.get('channel_type')
    if not channel_type:
        return jsonify({"error": "channel_type parameter is missing"}), 400
    image_amount = request.args.get('image_amount')
    if not image_amount:
        return jsonify({"error": "image_amount parameter is missing"}), 400
    image_generator_custom(channel_type, int(image_amount)) 
    return jsonify({"message": "Method 2"}), 200

@app.route('/api/create/3', methods=['GET'])
def api_check3():
    channel_type = request.args.get('channel_type')
    if not channel_type:
        return jsonify({"error": "channel_type parameter is missing"}), 400
    add_image_to_production(channel_type, "generated-images")
    return jsonify({"message": "Method 3"}), 200

@app.route('/api/method/4', methods=['GET'])
def api_check4():
    dir_checker()
    checker()
    return jsonify({"message": "Method 4"}), 200

if __name__ == "__main__":
   
    app.run(debug=True)