Update main.py

main.py

@@ -8,7 +8,7 @@ app = Flask(__name__)
 # Инициализация клиента OpenAI
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-@app.route('/generate-image', methods=['POST'])
+@app.route('/', methods=["GET", "POST"])
 def generate_image_api():
     # Проверка авторизации
     auth_key = request.headers.get('Authorization')
@@ -23,8 +23,9 @@ def generate_image_api():
     # Генерация изображения с помощью DALL-E
     response = openai.Image.create(prompt=description, n=1)
     image_url = response['data'][0]['url']  # Получение URL изображения
-
-    return jsonify({"image_url": image_url})
+    if request.method == "POST":
+        return jsonify({"image_url": image_url})
+    return render_template("app.py", user_input=prompt, response=response)
 
 if __name__ == '__main__':
     app.run(debug=True)