illuStrator / app.py
5m4ck3r's picture
Update app.py
3b189fb verified
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")
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"
}
)
@app.route("/solution")
def getSolution_from_query():
query = request.args.get("query")
try:
cl = getSolution()
out = cl.get_solution(query)
return jsonify(
{
"status" : True,
"msg" : out
}
)
except:
print(f"ERROR : {traceback.format_exc()}")
return jsonify(
{
"status" : False,
"msg" : "Something error has occur"
}
)
@app.route("/convert", methods=["POST"])
def convert_to_readable():
try:
xml = request.data.decode("utf-8")
if not xml:
return jsonify({"status": False, "msg": "No XML provided"}), 400
conv = convertBlock()
txt = conv.convert(xml)
return jsonify({"status": True, "msg": txt})
except Exception:
return jsonify({"status": False, "msg": f"Error: {traceback.format_exc()}"}), 500