Update app.py

app.py

@@ -47,9 +47,9 @@ def generate_cocktail(mood, sweetness, sour, savory, bitter, flavor_association,
             messages=messages,
             max_tokens=1024)
         name, quote, ingredients, instruction, notes = extract_info(response.choices[0].message.content)
-        return format_cocktail_output(name, quote, ingredients, instruction, notes), True, True
+        return format_cocktail_output(name, quote, ingredients, instruction, notes), True
     except Exception as e:
-        return f'<p style="color: white; font-size: 20px;">{str(e)}</p>', True, True
+        return f'<p style="color: white; font-size: 20px;">{str(e)}</p>', True
 
 # Extract information from the response generated by OpenAI
 def extract_info(output_text):
@@ -85,14 +85,6 @@ def format_cocktail_output(name, quote, ingredients, instruction, notes):
     '''
     return html_output
 
-def save_as_png(html_content):
-    """Converts HTML content to PDF, encodes it in base64, and returns a download link."""
-    hti = Html2Image()
-    css = 'body {background: url("https://images.unsplash.com/photo-1514361726087-38371321b5cd?q=80&w=2370&auto=format&fit=crop&ixlib=rb-4.0.3&ixid=M3wxMjA3fDB8MHxwaG90by1wYWdlfHx8fGVufDB8fHx8fA%3D%3D");}'
-
-    # screenshot an HTML string (css is optional)
-    hti.screenshot(html_str=html_content, css_str=css, save_as="CocktailRecipe.png")
-
         
 with open('style.css', 'r') as file:
     css_styles = file.read()
@@ -129,22 +121,18 @@ with gr.Blocks(css=css_styles, theme='xiaobaiyuan/theme_brief@>=0.0.2,<0.0.3') a
         
     play_button = gr.Button("Play Music", visible=False, elem_classes=["generate-button"], scale=1)  # Initially not visible
     background_music = gr.Audio(label="Background Music", autoplay=True, visible=False, scale=4)  # Initially not visible
-    pdf_download_link = gr.HTML(visible=False)  # For displaying the PDF download link
-
-    with gr.Row():
-        save_png_button = gr.Button("Download Recipe as PNG", visible=False)
 
     def on_generate_click(*args):
-        recipe, show_play_button, show_save_button = generate_cocktail(*args)
-        return recipe, gr.update(visible=show_play_button), gr.update(visible=show_save_button)
+        recipe, show_play_button = generate_cocktail(*args)
+        return recipe, gr.update(visible=show_play_button)
 
     def reset():
-        return "", 0, 0, 0, 0, [], [], 10, "", "", "", gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        return "", 0, 0, 0, 0, [], [], 10, "", "", "", gr.update(visible=False), gr.update(visible=False)
         
     generate_button.click(
         fn=on_generate_click,
         inputs=[mood, sweetness, sour, savory, bitter, flavor_association, drinking_experience, soberness_level, allergies, additional_requests],
-        outputs=[output_recipe, play_button, save_png_button]
+        outputs=[output_recipe, play_button]
     )
 
     with gr.Row():
@@ -155,10 +143,8 @@ with gr.Blocks(css=css_styles, theme='xiaobaiyuan/theme_brief@>=0.0.2,<0.0.3') a
                     outputs=[output_recipe, play_button], fn=on_generate_click, elem_id="example-box")
     
     play_button.click(fn=play_music, inputs=[], outputs=[background_music, background_music])
-
-    save_png_button.click(fn=save_as_png, inputs=[output_recipe], outputs=[pdf_download_link])
     
-    clear_button.click(fn=reset, inputs=[], outputs=[mood, sweetness, sour, savory, bitter, flavor_association, drinking_experience, soberness_level, allergies, additional_requests, output_recipe, play_button, background_music, save_png_button])
+    clear_button.click(fn=reset, inputs=[], outputs=[mood, sweetness, sour, savory, bitter, flavor_association, drinking_experience, soberness_level, allergies, additional_requests, output_recipe, play_button, background_music])
         
 if __name__ == "__main__":
     MoodShaker.launch(#enable_queue=False,