Upload 2 files

Dockerfile

@@ -0,0 +1,28 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN apt-get update && apt-get -y install libgl1-mesa-dev
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+RUN useradd -m -u 58241 user
+
+USER user
+
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+
+COPY --chown=user . $HOME/app
+
+RUN mkdir ~/.cache
+
+RUN chmod -R 777 ~/.cache
+
+COPY . .
+
+CMD [ "gunicorn", "-b", "0.0.0.0:7860", "--timeout", "300", "main:app"]

main.py

@@ -1,9 +1,113 @@
-import os
-import gradio as gr
+from flask import Flask, request, send_file, jsonify, render_template, send_from_directory, Response
+import io
+import base64
+import random
+import torch
+from models.inference import image_gen
 
-access_token = os.environ.get('HF_TOKEN',None)
+app = Flask(__name__)
 
-with gr.Blocks() as demo:
-    gr.load("sun000090/gradio_ui_test", hf_token=access_token, src="spaces")
+@app.route("/")
+def index():
+    return render_template("index.html")
 
-demo.launch()
+@app.route("/image_gen", methods=['POST'])
+def gen():
+    image_path = None
+    image_bytes = None
+    prompts_out = request.form['prompts_out']
+    neg_prompts_out = request.form['neg_prompts_out']
+    steps_out = int(request.form['steps_out'])
+    cfg_out = int(request.form['cfg_out'])
+    num_images_out = int(request.form['num_images_out'])
+    height_out = int(request.form['height_out'])
+    width_out = int(request.form['width_out'])
+    clip_skip_out = int(request.form['clip_skip_out'])
+
+    print("---------------------")
+    print("prompts out ", prompts_out)
+    print("neg prompt", neg_prompts_out)
+    print("---------------------")
+
+    image = image_gen(prompts_out, neg_prompts_out, steps_out, cfg_out, num_images_out,
+                    height_out, width_out, clip_skip_out)
+
+    # print(image)
+    # print("---------------------")
+    
+    image_bytes = io.BytesIO()
+    image.save(image_bytes, format="JPEG", quality=67)
+    
+    image_bytes = base64.b64encode(image_bytes.getvalue()).decode('utf-8')
+    print(f"Generated image of size {len(image_bytes)} bytes.")
+        
+    return render_template('index.html', image_path=image_bytes)
+    
+
+'''
+@app.route('/', methods=['POST'])
+def myapp():
+    # prompts_out = request.args["Prompt"]
+    # neg_prompts_out = request.args["Negative Prompt"]
+    # steps_out = request.args["Steps"]
+    # cfg_out = request.args["CFG"]
+    # num_images_out = request.args["Number of Images"]
+    # height_out = request.args["Height"]
+    # width_out = request.args["Width"]
+    # clip_skip_out = request.args["Clip Skip"]
+    # img_data = image_gen(prompts_out, neg_prompts_out, steps_out, cfg_out, num_images_out,
+    #                     height_out, width_out, clip_skip_out)
+    # return send_file(img_data, mimetype='image/png')
+    return "Hello World!"
+'''
+
+
+'''
+@app.route('/', methods=['GET', 'POST'])
+def index():
+    image_path = None
+    image_bytes =None
+    if request.method == 'POST':
+        prompts_out = request.form['prompts_out']
+        neg_prompts_out = request.form['neg_prompts_out']
+        steps_out = int(request.form['steps_out'])
+        cfg_out = int(request.form['cfg_out'])
+        num_images_out = int(request.form['num_images_out'])
+        height_out = int(request.form['height_out'])
+        width_out = int(request.form['width_out'])
+        clip_skip_out = int(request.form['clip_skip_out'])
+
+        ## seed = random.randrange(1, 100)
+
+        print("---------------------")
+        print("prompts out ", prompts_out)
+        print("neg prompt", neg_prompts_out)
+        print("---------------------")
+        
+        image = image_gen(prompts_out, neg_prompts_out, steps_out, cfg_out, num_images_out,
+                        height_out, width_out, clip_skip_out)
+
+        print(image)
+        print("---------------------")
+        
+        image_bytes = io.BytesIO()
+        image.save(image_bytes, format="JPEG", quality=67)
+        
+        image_bytes = base64.b64encode(image_bytes.getvalue()).decode('utf-8')
+        print(f"Generated image of size {len(image_bytes)} bytes.")
+        
+        # img_str = base64.b64encode(image_bytes.getvalue())
+        # img_obj = "data:image/png;base64," + str(img_str)[2:-1]
+        
+        # return Response(image_bytes, mimetype="image/jpeg")
+        
+        # return render_template('result.html',image=img_obj)
+        
+    return render_template('index.html', image_path=image_bytes)
+'''
+
+"""
+@app.route('/')
+def hello():
+    return {"test": "Hello world!"}
+"""