Update app.py

app.py

@@ -33,7 +33,7 @@ parsed_descriptions_queue = deque()
 
 # Usage limits
 MAX_DESCRIPTIONS = 30
-MAX_IMAGES = 2  # Limit to 2 images
+MAX_IMAGES = 3  # Limit to 3 images
 
 # Preload models and checkpoints
 print("Preloading models and checkpoints...")
@@ -98,6 +98,10 @@ def parse_descriptions(text):
     descriptions = [desc.strip() for desc in descriptions if len(desc.split()) >= 3]
     return descriptions
 
+def format_descriptions(descriptions):
+    formatted_descriptions = "\n".join(descriptions)
+    return formatted_descriptions
+
 @spaces.GPU
 def generate_descriptions(user_prompt, seed_words_input, batch_size=100, max_iterations=1):  # Set max_iterations to 1
     descriptions = []
@@ -133,7 +137,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=2):  # Set max_iterations to 1
+def generate_images(parsed_descriptions, max_iterations=3):  # Set max_iterations to 3
     # Limit the number of descriptions passed to the image generator to 2
     if len(parsed_descriptions) > MAX_IMAGES:
         parsed_descriptions = parsed_descriptions[:MAX_IMAGES]
@@ -146,13 +150,14 @@ def generate_images(parsed_descriptions, max_iterations=2):  # Set max_iteration
 
 def combined_function(user_prompt, seed_words_input):
     parsed_descriptions = generate_descriptions(user_prompt, seed_words_input)
+    formatted_descriptions = format_descriptions(parsed_descriptions)
     images = generate_images(parsed_descriptions)
-    return parsed_descriptions, images
+    return formatted_descriptions, images
 
 if __name__ == '__main__':
     def generate_and_display(user_prompt, seed_words_input):
-        parsed_descriptions, images = combined_function(user_prompt, seed_words_input)
-        return parsed_descriptions, images
+        formatted_descriptions, images = combined_function(user_prompt, seed_words_input)
+        return formatted_descriptions, images
 
     interface = gr.Interface(
         fn=generate_and_display,
@@ -162,4 +167,4 @@ if __name__ == '__main__':
         allow_flagging='never'  # Disable flagging
     )
 
-    interface.launch(share=True)
+    interface.launch(share=True)