davidizzle commited on
Commit
ddf3c8c
·
1 Parent(s): 8531ee6
Files changed (1) hide show
  1. app.py +18 -8
app.py CHANGED
@@ -58,20 +58,30 @@ with gr.Blocks() as demo:
58
 
59
  generate_btn = gr.Button("Generate")
60
 
61
- # Spinner with your custom GIF
62
  spinner = gr.HTML(
63
  "<div style='text-align:center'><img src='https://media2.giphy.com/media/v1.Y2lkPTc5MGI3NjExMXViMm02MnR6bGJ4c2h3ajYzdWNtNXNtYnNic3lnN2xyZzlzbm9seSZlcD12MV9pbnRlcm5hbF9naWZfYnlfaWQmY3Q9cw/k32ddF9WVs44OUaZAm/giphy.gif' width='180'></div>",
64
- visible=False # hidden by default
65
  )
66
 
67
  output = gr.Code(label="🧾 Leonardo's Work")
68
 
69
  def wrapped_generate(prompt, style):
70
- spinner.visible = True
71
- result = generate_code(prompt, style)
72
- spinner.visible = False
73
- return result
74
 
75
- generate_btn.click(fn=wrapped_generate, inputs=[prompt, style], outputs=output)
 
 
 
 
 
 
 
 
 
 
 
 
 
76
 
77
- demo.launch()
 
58
 
59
  generate_btn = gr.Button("Generate")
60
 
 
61
  spinner = gr.HTML(
62
  "<div style='text-align:center'><img src='https://media2.giphy.com/media/v1.Y2lkPTc5MGI3NjExMXViMm02MnR6bGJ4c2h3ajYzdWNtNXNtYnNic3lnN2xyZzlzbm9seSZlcD12MV9pbnRlcm5hbF9naWZfYnlfaWQmY3Q9cw/k32ddF9WVs44OUaZAm/giphy.gif' width='180'></div>",
63
+ visible=False
64
  )
65
 
66
  output = gr.Code(label="🧾 Leonardo's Work")
67
 
68
  def wrapped_generate(prompt, style):
69
+ code = generate_code(prompt, style)
70
+ return gr.update(visible=False), code
 
 
71
 
72
+ generate_btn.click(
73
+ fn=wrapped_generate,
74
+ inputs=[prompt, style],
75
+ outputs=[spinner, output],
76
+ show_progress=True,
77
+ preprocess=True
78
+ )
79
+
80
+ generate_btn.click(
81
+ fn=lambda: gr.update(visible=True),
82
+ inputs=[],
83
+ outputs=[spinner],
84
+ queue=False
85
+ )
86
 
87
+ demo.launch()