Update app.py

app.py

@@ -40,7 +40,7 @@ def chat_inf(system_prompt,prompt,history):
 
 
         
-def get_screenshot(chat: list,height=5000,width=600,chatblock=[1],header=True,theme="light",wait=3000):
+def get_screenshot(chat: list,height=5000,width=600,chatblock=[1],theme="light",wait=3000,header=True):
     result = ss_client.predict(chat,height,width,chatblock,header,theme,wait,api_name="/run_script")
 		# str  in 'Chat: [('user','bot'),('user','bot')]' Textbox component
 		# float  in 'Height' Number component
@@ -79,10 +79,17 @@ with gr.Blocks() as app:
                             stop_btn=gr.Button("Stop")
                             clear_btn=gr.Button("Clear")
         with gr.Column(scale=1):
-            with gr.Group():    
+            with gr.Group():
+                with gr.Row():
+                    im_height=gr.Number(label="Height",value=5000)
+                    im_width=gr.Number(label="Width",value=500)
+                wait_time=gr.Number(label="Wait Time",value=3000)
+                theme=gr.Radio(label="Theme", choices=["light","dark"])
+                chatblock=gr.Dropdown(label="Chatblocks",choices=[c for c in range(1,20)])
+                
                 im_btn=gr.Button("Screenshot")
                 img=gr.Image(type='filepath')
     btn.click(chat_inf,[sys_inp,inp,chat_b],chat_b)
-    im_btn.click(get_screenshot,chatbot,img)
+    im_btn.click(get_screenshot,[chat_b,im_height,im_width,chatblock,theme,wait_time],img)
     #app.load(get_screenshot,inp,img)
 app.launch()