Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 from overlay import txt_on_trans as tt 
 
-def process(txt,font_size,font_x,font_y,font_col,font_op):
-    return tt(txt,font_size,font_x,font_y,font_col,font_op)
+def process(txt,font_size,font_x,font_y,font_col,bg_col,bg_op,im_size_x,im_size_y):
+    return tt(txt,font_size,font_x,font_y,font_col,bg_col,bg_op,im_size_x,im_size_y)
 
 
 with gr.Blocks() as app:
@@ -11,12 +11,19 @@ with gr.Blocks() as app:
         with gr.Column():
             inp = gr.Textbox(lines = 6,min_width=100)
             btn = gr.Button()
-            f_x=gr.Slider(maximum=512, step=1, value=0)
-            f_y=gr.Slider(maximum=512, step=1, value=0)
-            f_size=gr.Number(precision=0, value=16)
-            f_col=gr.ColorPicker()
-            f_op=gr.Slider(maximum=255, step=1, value=255)
+            with gr.Row():
+                f_x=gr.Slider(label='Text X', maximum=512, step=1, value=0)
+                f_y=gr.Slider(label='Text Y', maximum=512, step=1, value=0)
+            with gr.Row():
+                f_size=gr.Number(label='Font Size',precision=0, value=32)
+                b_x=gr.Number(label='Image Width',precision=0, value=512)
+                b_y=gr.Number(label='Image Height',precision=0, value=256)
+                
+            with gr.Row():
+                f_col=gr.ColorPicker(label='Text Color')
+                b_col=gr.ColorPicker(label='Background Color')
+                f_op=gr.Slider(label='Background Opacity', maximum=255, step=1, value=255)
             
         outp = gr.Pil()
-    btn.click(process,[inp,f_size,f_x,f_y,f_col,f_op],outp)
+    btn.click(process,[inp,f_size,f_x,f_y,f_col,b_col,b_op,b_x,b_y],outp)
 app.launch()