Dockerfile(redis), model.py(dir), requirements.txt(redis,flask-limiter), app.py(bug fix)

Dockerfile

@@ -16,6 +16,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.10 \
     python3.10-dev \
     python3.10-distutils \
+    redis-server \
     && ln -sf /usr/share/zoneinfo/Asia/Tokyo /etc/localtime \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
@@ -37,11 +38,15 @@ RUN pip install --no-cache-dir -r requirements.txt \
 # アプリケーションのコピー
 COPY . /app
 
+# Redis の起動スクリプトを作成
+RUN echo '#!/bin/bash\nredis-server --daemonize yes\npython app.py' > /app/start.sh \
+    && chmod +x /app/start.sh
+
 # 非rootユーザーの作成と切り替え
 RUN useradd -m appuser && chown -R appuser:appuser /app
 USER appuser
 
 EXPOSE 80
 
-# CMD ["python", "app.py", "--use_gpu"]
-CMD ["python", "app.py"]
+# 起動コマンドを変更
+CMD ["/app/start.sh"]

app.py

@@ -30,8 +30,8 @@ redis_client = redis.Redis(host='localhost', port=6379, db=0)
 
 # レート制限の設定
 limiter = Limiter(
-    app,
-    key_func=get_remote_address,
+    get_remote_address,
+    app=app,
     default_limits=["200 per day", "50 per hour"]
 )
 
@@ -189,7 +189,7 @@ def handle_get_task_order(task_id):
     return jsonify({'task_order': task_order})
 
 @socketio.on('connect')
-def handle_connect():
+def handle_connect(auth):
     # クライアント接続数の制限
     if redis_client.get('connected_clients') and int(redis_client.get('connected_clients')) > 100:
         return False  # 接続を拒否

requirements.txt

@@ -19,3 +19,5 @@ tokenizers
 pytorch_lightning
 python-dotenv
 peft==0.11.1
+flask_limiter==3.7.0
+redis==5.0.7

scripts/model.py

@@ -153,8 +153,6 @@ def create_model(model, use_local):
     net = UnetGenerator(3, 1, 8, 64, norm_layer=norm_layer, use_dropout=False)
 
     import os
-    cwd = os.getcwd() # 現在のディレクトリを保存
-    os.chdir(os.path.dirname(__file__)) # このファイルのディレクトリに移動
     if model == 'default':
         model_path = (lambda filename, subfolder: os.path.join(subfolder, filename) if use_local else download_file(filename, subfolder)) \
                         ("netG.pth", "models/Anime2Sketch")
@@ -167,8 +165,6 @@ def create_model(model, use_local):
                 del ckpt[key]
         net.load_state_dict(ckpt)
 
-        os.chdir(cwd) # 元のディレクトリに戻る
-
     elif model == 'improved':
         ckpt = torch.load('weights/improved.bin', map_location=torch.device('cpu'))
         base = net.model.model[1]