Update app.py

app.py

@@ -161,26 +161,6 @@ def create_interface():
             api_name="generate_random_prompt_with_llm"
         )
 
-        # Add image generation button
-        generate_image_button = gr.Button("Generate Image")
-
-        # Function to generate image
-        def generate_image(text):
-            try:
-                seed = random.randint(0, 1000000)
-                image_path = llm_node.generate_image(text, seed=seed)
-                print(f"Image generated: {image_path}")
-                return image_path
-            except Exception as e:
-                print(f"An error occurred while generating the image: {e}")
-                return None
-
-        # Connect the image generation button
-        generate_image_button.click(
-            generate_image,
-            inputs=[text_output],
-            outputs=[image_output]
-        )
 
     return demo