Update app.py

app.py

@@ -1,73 +1,22 @@
-from flask import Flask, render_template,request,jsonify,Response  
-import os
-import json
-from flask_cors import CORS
-import time
-from flask_limiter import Limiter
-from flask_limiter.util import get_remote_address
-import requests
-app = Flask(__name__)
-CORS(app)
+FROM python:3.9-slim
 
-def get_client_ip():
-    return request.headers.get(os.getenv('head'), get_remote_address())
+# Set working directory
+WORKDIR /app
 
-limiter = Limiter(
-    key_func=get_client_ip,
-    app=app,
-    default_limits=["8 per minute"]
-)
+# Copy the current directory contents into the container at /app
+COPY . /app
 
+# Install Gunicorn and your application dependencies
+RUN pip install --no-cache-dir gunicorn && \
+    pip install --no-cache-dir -r requirements.txt
 
-users = 0
-userslist = []
+# Expose the port Gunicorn will listen on
+EXPOSE 7860
 
-@app.route("/")
-def index():
-    global users 
-    global userslist
-    if(request.remote_addr not in userslist):
-        userslist.append(request.remote_addr)
-    users = len(userslist)
-    return render_template('models.html')
+# Define environment variables
+ENV FLASK_APP=app.py
+ENV FLASK_RUN_HOST=0.0.0.0
+ENV FLASK_RUN_PORT=7860
 
-    
-@app.route("/users")
-def users():
-    return {"total users : " : users}
-
-@app.route('/gen', methods=['POST'])
-@limiter.limit("8 per minute")
-def Hf():
-    
-    prompt = request.json.get('prompt', '')
-    negative = request.json.get('negative', '')
-    steps = request.json.get('steps', 20)
-    width = request.json.get('width',1024)
-    height = request.json.get('height',1024)
-    scale = request.json.get('scale',7)
-    model = request.json.get('model','sd3')
-    style = request.json.get('style', 'Cinematic')
-    def Gen(prompt,negative,steps,width,height,scale,style,model):
-        req = requests.post('https://xyplon-server2.hf.space/hf/img/gen',headers={
-            'Authorization' : os.getenv('auth')
-        },json={
-            'prompt': prompt,
-            'negative': negative,
-            'steps': steps,
-            'width': width,
-            'height': height,
-            'scale': scale,
-            'model' : model,
-            'style': style
-        }, stream=True)
-        if(req.status_code!=200):
-            return "an error occurred! ", 500
-        for chunk in req.iter_lines():
-            yield f'{chunk.decode()}\n'
-        
-    return Response(Gen(prompt=prompt,negative=negative,steps=steps,width=width,height=height,scale=scale,style=style,model=model), mimetype="text/event-stream")
-
-
-if __name__ == "__main__":
-    app.run(debug=True, host='0.0.0.0', port=7860, threaded=True)
+# Run the Gunicorn server
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]