tywei08 commited on
Commit
cd5381e
·
verified ·
1 Parent(s): 8f2f745

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -102,14 +102,19 @@ with gr.Blocks(css=css_styles) as demo:
102
 
103
  output_recipe = gr.HTML(label="Your Cocktail Recipe")
104
 
105
- play_button = gr.Button("Play Music", visible=False, elem_classes=["generate-button", "right-aligned-small"]) # Initially not visible
 
 
 
106
  background_music = gr.Audio(label="Background Music", autoplay=True, visible=False) # Initially not visible
107
 
108
  def on_generate_click(*args):
109
  recipe, show_play_button = generate_cocktail(*args)
110
  return recipe, gr.update(visible=show_play_button)
111
-
112
 
 
 
 
113
  generate_button.click(
114
  fn=on_generate_click,
115
  inputs=[mood, sweetness, sour, savory, bitter, flavor_association, drinking_experience, soberness_level, allergies, additional_requests],
@@ -118,7 +123,7 @@ with gr.Blocks(css=css_styles) as demo:
118
 
119
  play_button.click(fn=play_music, inputs=[], outputs=[background_music, background_music])
120
 
121
-
122
 
123
  if __name__ == "__main__":
124
  demo.launch(#enable_queue=False,
 
102
 
103
  output_recipe = gr.HTML(label="Your Cocktail Recipe")
104
 
105
+ with gr.Row():
106
+ play_button = gr.Button("Play Music", visible=False, elem_classes=["generate-button"], scale=4) # Initially not visible
107
+ clear_button = gr.Button("Clear", scale=1)
108
+
109
  background_music = gr.Audio(label="Background Music", autoplay=True, visible=False) # Initially not visible
110
 
111
  def on_generate_click(*args):
112
  recipe, show_play_button = generate_cocktail(*args)
113
  return recipe, gr.update(visible=show_play_button)
 
114
 
115
+ def reset(mood, sweetness, sour, savory, bitter, flavor_association, drinking_experience, soberness_level, allergies, additional_requests):
116
+ return "", 0, 0, 0, 0, [], [], 10, "", ""
117
+
118
  generate_button.click(
119
  fn=on_generate_click,
120
  inputs=[mood, sweetness, sour, savory, bitter, flavor_association, drinking_experience, soberness_level, allergies, additional_requests],
 
123
 
124
  play_button.click(fn=play_music, inputs=[], outputs=[background_music, background_music])
125
 
126
+ clear_button.click(fn=reset, inputs=[], outputs=[mood, sweetness, sour, savory, bitter, flavor_association, drinking_experience, soberness_level, allergies, additional_requests])
127
 
128
  if __name__ == "__main__":
129
  demo.launch(#enable_queue=False,