Update app.py

app.py

@@ -23,12 +23,7 @@ def textover(im,text):
     
     font = ImageFont.truetype("./fonts/unifont-15.0.01.ttf", int(font_size))
     draw.text((x, y),text, font = font, fill=t_fill)
-
-
-
-
-
-
+    return im
 def get_concat_h_cut(in1, in2):
     im1=Image.open(in1)
     im2=Image.open(in2)
@@ -46,7 +41,7 @@ def get_concat_v_cut(im1,theme='light'):
     if theme=='light':
         color=(255,255,255)
     dst = Image.new('RGB', (im1.width, im1.height +200),color=color)
-    dst.paste(im1, (0, 0))
+    dst.paste(im1, (0, 200))
     return dst
 
 
@@ -101,6 +96,7 @@ def generate(prompt, history):
     
 def run(inp,model_drop):
     prompt=inp['PROMPT']
+    text=inp['MEME_TEXT']
     model=loaded_model[int(model_drop)]
     out_img=model(prompt)
     print(out_img)
@@ -111,6 +107,7 @@ def run(inp,model_drop):
     if r.status_code == 200:
         out = Image.open(io.BytesIO(r.content))
     out=get_concat_v_cut(out)
+    out=textover(out,text)
     return out
 
 def run_gpt(in_prompt,history,):
@@ -157,6 +154,5 @@ with gr.Blocks() as app:
         
     sub_b = submit_b.click(generate, [msg,chatbot],[chatbot,json_out])
     sub_im = submit_im.click(run, [json_out,model_drop],[im_out])
-    sub_e = msg.submit(run, [msg, chatbot,model_drop], [chatbot,im_out])
     stop_b.click(None,None,None, cancels=[sub_b,sub_e])
 app.launch()