Update app.py

app.py

@@ -1,22 +1,72 @@
-FROM python:3.9-slim
+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)
 
-# Set working directory
-WORKDIR /app
+def get_client_ip():
+    return request.headers.get(os.getenv('head'), get_remote_address())
 
-# Copy the current directory contents into the container at /app
-COPY . /app
+limiter = Limiter(
+    key_func=get_client_ip,
+    app=app,
+    default_limits=["8 per minute"]
+)
 
-# Install Gunicorn and your application dependencies
-RUN pip install --no-cache-dir gunicorn && \
-    pip install --no-cache-dir -r requirements.txt
 
-# Expose the port Gunicorn will listen on
-EXPOSE 7860
+users = 0
+userslist = []
 
-# Define environment variables
-ENV FLASK_APP=app.py
-ENV FLASK_RUN_HOST=0.0.0.0
-ENV FLASK_RUN_PORT=7860
+@app.route("/")
+def index():
+    global users 
+    if(request.remote_addr not in userslist):
+        userslist.append(request.remote_addr)
+    users = len(userslist)
+    return render_template('models.html')
 
-# Run the Gunicorn server
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]
+    
+@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)