Update app.py

app.py

@@ -44,7 +44,7 @@ def generateStory(theme1, theme2):
     return story
 
 
-def illustratedStory(theme1, theme2):
+def illustratedStory(story, theme1, theme2):
 
     if story != 'Please generate again':
       #image_prompt = illustration_response["choices"][0]["text"]
@@ -175,7 +175,7 @@ with gr.Blocks(css='''
     
     
     b1.click(generateStory, inputs=[theme1,theme2], outputs=[story_output])
-    b2.click(illustratedStory, inputs=[theme1, theme2], outputs=[illustration])
+    b2.click(illustratedStory, inputs=[story, theme1, theme2], outputs=[illustration])