fix for fastapi + uvicorn

Dockerfile

@@ -5,4 +5,4 @@ COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
-CMD ["python", "app.py"]
+CMD ["uvicorn","app:app","--host","0.0.0.0","--port","7860"]

app.py

@@ -1,32 +1,22 @@
-from flask import Flask, request, jsonify
+from fastapi import FastAPI, Query, HTTPException
+from typing import Dict, Optional
 import requests
 
-app = Flask(__name__)
+app = FastAPI()
 
-# @app.route('/', methods=['GET'])
-# def hello():
-#     return {'Hello' : 'World!'}
+@app.get("/")
+def read_root():
+    return {"message": "Hello, World!"}
 
-@app.route('/', methods=['GET'])
-def api():
-    print("get", request.args)
+@app.get("/api/")
+def api(endpoint: str = Query(..., alias="endpoint"), other_params: Optional[Dict[str, str]] = None):
     # クライアントからのリクエストデータを取得
-    endpoint = request.args.get('endpoint')
-    other_params = {key: value for key, value in request.args.items() if key != 'endpoint'}
-
     if not endpoint:
-        return jsonify({'error': 'No endpoint specified'}), 400
-
+        raise HTTPException(status_code=400, detail="No endpoint specified")
+    
     # AWS API Gatewayにリクエストを転送
     print(endpoint, other_params)
     response = requests.get(endpoint, params=other_params)
-
+    
     # AWSからのレスポンスをクライアントに返す
-    return jsonify(response.json()), response.status_code
-
-
-
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
-
-
+    return response.json()

requirements.txt

@@ -1,2 +1,3 @@
-Flask
+fastapi
+uvicorn
 requests