from flask import Flask, request, jsonify
from asgiref.sync import async_to_sync
from hiou import *
import traceback

app = Flask(__name__)

async def generate_illu(subject, out1, out2, expression, bgdesc, bgcol):
    gene = gen()
    result = await gene.generate_illustrator(subject, out1, out2, expression, bgdesc, bgcol)
    return result

async def generate_image(prompt, negative):
    gene = gen()
    result = await gene.generete_image(prompt, negative)
    return result

@app.route('/illu')
def gettans_task():
    query_string = request.args.get('s')
    tore = request.args.get('t', 4)
    try:
        tore = int(tore)
    except:
        tore = 4
    ser = search()
    result = ser.search(query_string, tore)
    return jsonify(
        result
    )

@app.route('/genillu')
def generateIllus():
    subject = request.args.get('subject')
    out1 = request.args.get('out1')
    out2 = request.args.get('out2')
    expression = request.args.get('exp')
    bgdesc = request.args.get('desc')
    bgcol = request.args.get('bgcol')
    try:
        resul = async_to_sync(generate_illu)(subject, out1, out2, expression, bgdesc, bgcol)
    except Exception as e:
        print(f"ERROR : {e}, {traceback.format_exc()}")
        return jsonify([])
    return jsonify(resul)

@app.route('/genimg')
def generate_ImageS():
    prompt = request.args.get("prompt")
    negative = request.args.get("negative")
    print(f"Image prompt : {prompt} - {negative}")
    try:
        resul = async_to_sync(generate_image)(prompt, negative)
    except Exception as e:
        print(f"ERROR : {e}, {traceback.format_exc()}")
        return jsonify([])
    return jsonify(resul)

@app.route('/')
def root():
    return jsonify(
        {
            "status" : True,
            "message" : "Nothing here"
        }
    )