import gradio as gr
from PIL import Image

demo_inf = gr.Blocks.load(name="spaces/freddyaboulton/blocks_inputs")
#assert demo_inf("Foo", "bar") == "Foo bar"

def txt_fun(txt1, txt2):
  return demo_inf(txt1, txt2)

#original = Image.open("./c4.jpg")
#output_img_path = demo_inf("./c4.jpg", fn_index=1)
#output = Image.open(output_img_path)
#assert original.size == output.size

def img_fun1(img_in):
  return demo_inf(img_in) #, fn_index=1)

#original = Image.open("./c4.jpg")
#output_img_path = demo_inf("./c4.jpg", fn_index=1)
#output = Image.open(output_img_path)
#assert original.size == output.size

def img_fun(img_in):
  original = Image.open("./c4.jpg")
  print("after line1")
  output_img_path = demo_inf("./c4.jpg", fn_index=1)
  print("after line1")
  output = Image.open(output_img_path)
  print("after line1")
  return output
  
demo = gr.Blocks()

with demo:
  gr.Markdown("<h1><center>Testing</center></h1>")
  gr.Markdown(
        """Testing Blocks Inference"""
        )
  with gr.Row():
    in_image = gr.Image(visible=False)
    out_image = gr.Image()
    
    in_text1 = gr.Textbox()
    in_text2 = gr.Textbox()
    out_text = gr.Textbox()
    
  b1 = gr.Button("Image Button")
  b2 = gr.Button("Text Button") 

  b1.click(img_fun,inputs=in_image, outputs=out_image) 
  b2.click(txt_fun, inputs=[in_text1, in_text2], outputs=out_text)
  #examples=examples 

demo.launch(enable_queue=True, debug=True)