heaversm commited on
Commit
7b994e6
1 Parent(s): 15137f1

add numbering to images

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -110,6 +110,7 @@ def download_all_images():
110
 
111
  def generate_images(prompts, pw, model):
112
  # Check for a valid password
 
113
  if pw != os.getenv("PW"):
114
  raise gr.Error("Invalid password. Please try again.")
115
 
@@ -120,7 +121,8 @@ def generate_images(prompts, pw, model):
120
  # Split the prompts string into individual prompts based on semicolon separation
121
  prompts_list = [prompt for prompt in prompts.split(';') if prompt]
122
 
123
- for entry in prompts_list:
 
124
  entry_parts = entry.split('-', 1) # Split by the first dash found
125
  if len(entry_parts) == 2:
126
  #raise gr.Error("Invalid prompt format. Please ensure it is in 'initials-prompt' format.")
@@ -150,7 +152,7 @@ def generate_images(prompts, pw, model):
150
 
151
  image_url = response.data[0].url
152
  # conditionally render the user to the label with the prompt
153
- image_label = f"Prompt: {text}" if user_initials == "" else f"User: {user_initials}, Prompt: {text}"
154
 
155
  try:
156
  # Save the prompt, model, image URL, generation time and creation timestamp to the database
 
110
 
111
  def generate_images(prompts, pw, model):
112
  # Check for a valid password
113
+
114
  if pw != os.getenv("PW"):
115
  raise gr.Error("Invalid password. Please try again.")
116
 
 
121
  # Split the prompts string into individual prompts based on semicolon separation
122
  prompts_list = [prompt for prompt in prompts.split(';') if prompt]
123
 
124
+ for i, entry in enumerate(prompts_list):
125
+ print(i)
126
  entry_parts = entry.split('-', 1) # Split by the first dash found
127
  if len(entry_parts) == 2:
128
  #raise gr.Error("Invalid prompt format. Please ensure it is in 'initials-prompt' format.")
 
152
 
153
  image_url = response.data[0].url
154
  # conditionally render the user to the label with the prompt
155
+ image_label = f"Prompt {i}: {text}" if user_initials == "" else f"Prompt {i}: {text}, User: {user_initials}"
156
 
157
  try:
158
  # Save the prompt, model, image URL, generation time and creation timestamp to the database