Update main.py

main.py

@@ -1,4 +1,3 @@
-# Fichier: main.py
 import os
 import io
 from flask import Flask, request, send_file, jsonify
@@ -6,14 +5,12 @@ from flask_cors import CORS
 from src.solle_img_generation import imggenrator
 from asgiref.wsgi import WsgiToAsgi
 
-# L'instance de l'application doit s'appeler 'app'
-app = Flask(__name__)
-CORS(app)
+flask_app = Flask(__name__)
+CORS(flask_app)
 
-# Crée une instance de votre classe de génération d'images
 image_generator = imggenrator()
 
-@app.route('/generate_image', methods=['POST'])
+@flask_app.route('/generate_image', methods=['POST'])
 def generate_image_api():
     try:
         data = request.get_json()
@@ -34,15 +31,12 @@ def generate_image_api():
     
     return jsonify({"message": "No image was generated."}), 500
 
-# Endpoint de santé pour vérifier si l'API est en cours d'exécution
-@app.route('/', methods=['GET'])
+@flask_app.route('/', methods=['GET'])
 def health_check():
     return jsonify({"status": "running"}), 200
 
-# Adapte l'application Flask (WSGI) pour qu'elle puisse être servie par Uvicorn (ASGI)
-asgi_app = WsgiToAsgi(app)
+app = WsgiToAsgi(flask_app)
 
 if __name__ == '__main__':
-    # Le lancement local reste inchangé pour les tests
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(flask_app, host="0.0.0.0", port=7860)