Update app.py

app.py

@@ -88,7 +88,7 @@ def parse_descriptions(text):
     return descriptions
 
 @spaces.GPU
-def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_iterations=3):
+def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_iterations=2):
     descriptions = []
     description_queue = deque()
     iteration_count = 0
@@ -129,7 +129,7 @@ def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_ite
     return list(parsed_descriptions_queue)
 
 @spaces.GPU(duration=120)
-def generate_images(parsed_descriptions, max_iterations=10):
+def generate_images(parsed_descriptions, max_iterations=3):
     pipe = initialize_diffusers()
 
     if len(parsed_descriptions) < MAX_IMAGES:
@@ -145,25 +145,19 @@ def generate_images(parsed_descriptions, max_iterations=10):
 
 def combined_function(user_prompt, seed_words_input):
     parsed_descriptions = generate_descriptions(user_prompt, seed_words_input)
-    return parsed_descriptions
-
-def generate_images_from_descriptions(parsed_descriptions):
     images = generate_images(parsed_descriptions)
-    return images
+    return parsed_descriptions, images
 
 if __name__ == '__main__':
-    description_interface = gr.Interface(
-        fn=combined_function,
-        inputs=[gr.Textbox(lines=2, placeholder="Enter a prompt for descriptions..."), gr.Textbox(lines=2, placeholder='Enter seed words in quotes, e.g., "cat", "dog", "sunset"...')],
-        outputs=gr.Textbox(label="Generated Descriptions")
-    )
+    def generate_and_display(user_prompt, seed_words_input):
+        parsed_descriptions, images = combined_function(user_prompt, seed_words_input)
+        return gr.update(value=parsed_descriptions, visible=True), gr.update(value=images, visible=True)
 
-    image_interface = gr.Interface(
-        fn=generate_images_from_descriptions,
-        inputs=gr.Textbox(label="Generated Descriptions"),
-        outputs=gr.Gallery(label="Generated Images")
+    interface = gr.Interface(
+        fn=generate_and_display,
+        inputs=[gr.Textbox(lines=2, placeholder="Enter a prompt for descriptions..."), gr.Textbox(lines=2, placeholder='Enter seed words in quotes, e.g., "cat", "dog", "sunset"...')],
+        outputs=[gr.Row([gr.Column([gr.Textbox(label="Generated Descriptions")]), gr.Column([gr.Gallery(label="Generated Images")])])],
+        live=True
     )
 
-    interface = gr.TabbedInterface([description_interface, image_interface], ["Generate Descriptions", "Generate Images"])
-
     interface.launch(share=True)