Update api.py

api.py

@@ -42,10 +42,10 @@ def enhance(prompt):
 print("Loading AI model...")
 
 pipe=StableDiffusionPipeline.from_pretrained(
-"runwayml/stable-diffusion-v1-5",
-torch_dtype=torch.float32
+"runwayml/stable-diffusion-v1-5"
 )
 
+pipe=pipe.to("cpu")
 pipe.enable_attention_slicing()
 
 print("MODEL READY")
@@ -139,96 +139,185 @@ margin-top:20px;
 <script>
 
 function gen(){
-fetch("/generate",{method:"POST",headers:{"Content-Type":"application/json"},
-body:JSON.stringify({prompt:prompt.value})})
+
+fetch("/generate",{
+method:"POST",
+headers:{"Content-Type":"application/json"},
+body:JSON.stringify({
+prompt:document.getElementById("prompt").value
+})
+})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function product(){
-fetch("/product",{method:"POST",headers:{"Content-Type":"application/json"},
-body:JSON.stringify({prompt:prompt.value})})
+
+fetch("/product",{
+method:"POST",
+headers:{"Content-Type":"application/json"},
+body:JSON.stringify({
+prompt:document.getElementById("prompt").value
+})
+})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function logo(){
-fetch("/logo",{method:"POST",headers:{"Content-Type":"application/json"},
-body:JSON.stringify({prompt:prompt.value})})
+
+fetch("/logo",{
+method:"POST",
+headers:{"Content-Type":"application/json"},
+body:JSON.stringify({
+prompt:document.getElementById("prompt").value
+})
+})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function banner(){
-fetch("/banner",{method:"POST",headers:{"Content-Type":"application/json"},
-body:JSON.stringify({prompt:prompt.value})})
+
+fetch("/banner",{
+method:"POST",
+headers:{"Content-Type":"application/json"},
+body:JSON.stringify({
+prompt:document.getElementById("prompt").value
+})
+})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function social(){
-fetch("/social",{method:"POST",headers:{"Content-Type":"application/json"},
-body:JSON.stringify({prompt:prompt.value})})
+
+fetch("/social",{
+method:"POST",
+headers:{"Content-Type":"application/json"},
+body:JSON.stringify({
+prompt:document.getElementById("prompt").value
+})
+})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function batch(){
-fetch("/batch",{method:"POST",headers:{"Content-Type":"application/json"},
-body:JSON.stringify({prompt:prompt.value})})
+
+fetch("/batch",{
+method:"POST",
+headers:{"Content-Type":"application/json"},
+body:JSON.stringify({
+prompt:document.getElementById("prompt").value
+})
+})
 .then(r=>r.json())
-.then(data=>alert("Generated "+data.length+" images"))
+.then(data=>{
+alert("Generated "+data.length+" images")
+})
+
 }
 
 function variations(){
-fetch("/variations",{method:"POST",headers:{"Content-Type":"application/json"},
-body:JSON.stringify({prompt:prompt.value})})
+
+fetch("/variations",{
+method:"POST",
+headers:{"Content-Type":"application/json"},
+body:JSON.stringify({
+prompt:document.getElementById("prompt").value
+})
+})
 .then(r=>r.json())
-.then(data=>alert("Created "+data.length+" variations"))
+.then(data=>{
+alert("Created "+data.length+" variations")
+})
+
 }
 
 function restore(){
-var f=file.files[0]
+
+var f=document.getElementById("file").files[0]
+
 var d=new FormData()
+
 d.append("image",f)
 
 fetch("/restore",{method:"POST",body:d})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function upscale(){
-var f=file.files[0]
+
+var f=document.getElementById("file").files[0]
+
 var d=new FormData()
+
 d.append("image",f)
 
 fetch("/upscale",{method:"POST",body:d})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function color(){
-var f=file.files[0]
+
+var f=document.getElementById("file").files[0]
+
 var d=new FormData()
+
 d.append("image",f)
 
 fetch("/colorize",{method:"POST",body:d})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function removebg(){
-var f=file.files[0]
+
+var f=document.getElementById("file").files[0]
+
 var d=new FormData()
+
 d.append("image",f)
 
 fetch("/removebg",{method:"POST",body:d})
 .then(r=>r.blob())
-.then(b=>result.src=URL.createObjectURL(b))
+.then(b=>{
+document.getElementById("result").src=URL.createObjectURL(b)
+})
+
 }
 
 function gallery(){
+
 window.location="/gallery"
+
 }
 
 </script>
@@ -249,14 +338,22 @@ def home():
 @app.route("/generate",methods=["POST"])
 def generate():
 
+    print("GENERATE START")
+
     prompt=request.json["prompt"]
+
     prompt=enhance(translate(prompt))
 
+    print("PROMPT:",prompt)
+
     img=pipe(prompt,num_inference_steps=20).images[0]
 
     path=OUTPUT_DIR+"/image.png"
+
     img.save(path)
 
+    print("IMAGE SAVED")
+
     return send_file(path,mimetype="image/png")
 
 # =============================
@@ -267,11 +364,13 @@ def generate():
 def product():
 
     prompt=request.json["prompt"]
+
     prompt=translate(prompt)+", luxury product photography"
 
     img=pipe(prompt).images[0]
 
     path=OUTPUT_DIR+"/product.png"
+
     img.save(path)
 
     return send_file(path,mimetype="image/png")
@@ -284,11 +383,13 @@ def product():
 def logo():
 
     prompt=request.json["prompt"]
+
     prompt=translate(prompt)+", minimalist vector logo"
 
     img=pipe(prompt).images[0]
 
     path=OUTPUT_DIR+"/logo.png"
+
     img.save(path)
 
     return send_file(path,mimetype="image/png")
@@ -301,11 +402,13 @@ def logo():
 def banner():
 
     prompt=request.json["prompt"]
+
     prompt=translate(prompt)+", website banner design"
 
     img=pipe(prompt).images[0]
 
     path=OUTPUT_DIR+"/banner.png"
+
     img.save(path)
 
     return send_file(path,mimetype="image/png")
@@ -318,11 +421,13 @@ def banner():
 def social():
 
     prompt=request.json["prompt"]
+
     prompt=translate(prompt)+", instagram advertising post"
 
     img=pipe(prompt).images[0]
 
     path=OUTPUT_DIR+"/social.png"
+
     img.save(path)
 
     return send_file(path,mimetype="image/png")