Update app.py

app.py

@@ -84,7 +84,12 @@ def index():
 
 @app.route('/generate', methods=['GET'])
 def generate_image():
-    print(request.remote_addr);
+    if request.headers.getlist("X-Forwarded-For"):
+        client_ip = request.headers.getlist("X-Forwarded-For")[0]
+    else:
+        client_ip = request.remote_addr
+
+    print(f"Client IP: {client_ip}")
     prompt = request.args.get("prompt", "")
     negative_prompt = request.args.get("negative_prompt", "")
     steps = int(request.args.get("steps", 35))