Update app.py

app.py

@@ -91,16 +91,6 @@ def chat_inf(system_prompt,prompt,history,memory,client_choice,seed,temp,tokens,
         print("\n######### HIST "+str(in_len))
         print("\n######### TOKENS "+str(tokens))        
 
-def get_screenshot(chat: list,height=5000,width=600,chatblock=[],theme="light",wait=3000,header=True):
-    print(chatblock)
-    tog = 0
-    if chatblock:
-        tog = 3
-    result = ss_client.predict(str(chat),height,width,chatblock,header,theme,wait,api_name="/run_script")
-    out = f'https://omnibus-html-image-current-tab.hf.space/file={result[tog]}'
-    print(out)
-    return out
-
 def clear_fn():
     return None,None,None,None
 rand_val=random.randint(1,1111111111111111)
@@ -143,8 +133,6 @@ with gr.Blocks() as app:
     client_choice.change(load_models,client_choice,[chat_b])
     app.load(load_models,client_choice,[chat_b])
     
-    im_go=im_btn.click(get_screenshot,[chat_b,im_height,im_width,chatblock,theme,wait_time],img)
-    
     chat_sub=inp.submit(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt],[chat_b,memory])
     go=btn.click(check_rand,[rand,seed],seed).then(chat_inf,[sys_inp,inp,chat_b,memory,client_choice,seed,temp,tokens,top_p,rep_p,chat_mem,custom_prompt],[chat_b,memory])