Update test.py

test.py

@@ -10,7 +10,7 @@ def index():
     return render_template('index.html')
 
 @app.route('/generate', methods=['POST'])
-def generate_image():
+async def generate_image():
     prompt = request.form['prompt']
     style = request.form['style']
     ratio = request.form['ratio']
@@ -18,11 +18,11 @@ def generate_image():
     imagine = AsyncImagine()
 
     try:
-        img_data = asyncio.run(imagine.sdprem(
+        img_data = await imagine.sdprem(
             prompt=prompt,
             style=Style[style],
             ratio=Ratio[ratio]
-        ))
+        )
     except Exception as e:
         return f"An error occurred while generating the image: {e}", 500
 
@@ -44,7 +44,7 @@ def generate_image():
     return render_template('output.html')
 
 @app.route('/api/generate', methods=['POST'])
-def api_generate_image():
+async def api_generate_image():
     data = request.get_json()
     prompt = data['prompt']
     style = data['style']
@@ -53,11 +53,11 @@ def api_generate_image():
     imagine = AsyncImagine()
     
     try:
-        img_data = asyncio.run(imagine.sdprem(
+        img_data = await imagine.sdprem(
             prompt=prompt,
             style=Style[style],
             ratio=Ratio[ratio]
-        ))
+        )
     except Exception as e:
         return jsonify({'error': f"An error occurred while generating the image: {e}"}), 500
 
@@ -80,4 +80,4 @@ def api_generate_image():
     return send_file(image_path, mimetype='image/jpeg', as_attachment=True)
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860)