Update app.py

app.py

@@ -19,8 +19,8 @@ print (loaded_model)
 
 
 
-def run_dif_color(out_prompt,model_drop,cnt,color):
-
+def run_dif_color(out_prompt,model_drop,cnt,color,tint):
+    print (color)
     p_seed=""
     out_box=[]
     out_html=""
@@ -35,7 +35,7 @@ def run_dif_color(out_prompt,model_drop,cnt,color):
             raw=Image.open(out_img)
             raw=raw.convert('RGB')
 
-            colorize = RGBTransform().mix_with((255, 0, 0),factor=.30).applied_to(raw)    
+            colorize = RGBTransform().mix_with(color,factor=float(tint)).applied_to(raw)    
             
             out_box.append(colorize)           
         except Exception as e:
@@ -158,13 +158,17 @@ min-width:200px;
 """
 with gr.Blocks(css=css) as app:
     with gr.Row():
-        inp=gr.Textbox(label="Prompt")
-        btn=gr.Button()
+        with gr.Column():
+            inp=gr.Textbox(label="Prompt")
+            btn=gr.Button()
+        with gr.Column():
+            col = gr.ColorPicker(label="Color Tint")
+            tint = gr.Slider(label="Tint Strength", minimum=0, maximum=1, step=0.01, value=0.30)
         
     with gr.Row():
         model_drop=gr.Dropdown(label="Models", choices=models, type='index', value=models[0])
         cnt = gr.Number(value=1)
     out_html=gr.HTML()
     outp=gr.Gallery()
-    btn.click(run_dif_color,[inp,model_drop,cnt],[outp,out_html])
+    btn.click(run_dif_color,[inp,model_drop,cnt,col,tint],[outp,out_html])
 app.launch()