Update app.py

app.py

@@ -32,7 +32,7 @@ parsed_descriptions_queue = deque()
 
 # Usage limits
 MAX_DESCRIPTIONS = 30
-MAX_IMAGES = 10
+MAX_IMAGES = 3
 
 def initialize_diffusers():
     from optimum.quanto import freeze, qfloat8, quantize
@@ -145,14 +145,25 @@ 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 parsed_descriptions, images
+    return images
 
 if __name__ == '__main__':
-    interface = gr.Interface(
+    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"), gr.Gallery(label="Generated Images")]
+        outputs=gr.Textbox(label="Generated Descriptions")
     )
 
-    interface.launch()
+    image_interface = gr.Interface(
+        fn=generate_images_from_descriptions,
+        inputs=gr.Textbox(label="Generated Descriptions"),
+        outputs=gr.Gallery(label="Generated Images")
+    )
+
+    interface = gr.TabbedInterface([description_interface, image_interface], ["Generate Descriptions", "Generate Images"])
+
+    interface.launch(share=True)