Update app.py

app.py

@@ -1,11 +1,23 @@
 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', "smiling")
+    query_string = request.args.get('s')
     tore = request.args.get('t', 4)
     try:
         tore = int(tore)
@@ -17,6 +29,32 @@ def gettans_task():
         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(