update

Dockerfile

@@ -22,10 +22,5 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 # Copy the rest of the application code into the container, changing ownership to 'user'
 COPY --chown=user . /app
 
-# Set the Flask environment variables
-ENV FLASK_APP=app.py
-ENV FLASK_RUN_HOST=0.0.0.0
-ENV FLASK_RUN_PORT=7860
-
-# Specify the command to run the Flask application
-CMD ["flask", "run"]
+# Specify the command to run the application
+CMD ["python", "app.py"]

app.py

@@ -1,6 +1,7 @@
 from flask import Flask
 from fastapi_app import fastapi_app
-from fastapi.middleware.wsgi import WSGIMiddleware
+from werkzeug.middleware.dispatcher import DispatcherMiddleware
+from werkzeug.serving import run_simple
 
 # Initialize Flask app
 flask_app = Flask(__name__)
@@ -10,8 +11,10 @@ flask_app = Flask(__name__)
 def flask_hello():
     return 'Hello from Flask!'
 
-# Mount FastAPI on Flask using WSGIMiddleware
-flask_app.wsgi_app = WSGIMiddleware(fastapi_app)
+# Combine Flask and FastAPI using DispatcherMiddleware
+application = DispatcherMiddleware(flask_app, {
+    '/fastapi': fastapi_app
+})
 
 if __name__ == '__main__':
-    flask_app.run()
+    run_simple('0.0.0.0', 7860, application)

requirements.txt

@@ -4,4 +4,5 @@ datasets==2.21.0
 torch==2.3.1
 flask==3.0.3
 fastapi
-uvicorn[standard]
+uvicorn[standard]
+werkzeug