charliealonso commited on
Commit
7dc402a
1 Parent(s): e971215

Concatenar orden correcto

Browse files
Files changed (1) hide show
  1. app.py +5 -13
app.py CHANGED
@@ -1,6 +1,4 @@
1
-
2
  from contextlib import nullcontext
3
-
4
  import csv
5
  import random
6
  import gradio as gr
@@ -13,18 +11,12 @@ def parametroAleatorio(lista):
13
  return res
14
 
15
  def concat_prompt(*args):
16
- campos = {arg: valor for arg, valor in zip(csv_reader.fieldnames, args)}
17
- choices = [campo.strip() for campo in campos.values() if campo]
18
-
19
- for campo, valor in campos.items():
20
- if not valor:
21
- # Si el campo está vacío, se selecciona un valor aleatorio de la lista correspondiente
22
- campos[campo] = parametroAleatorio(valoresUnicos[campo])
23
- choices.append(campos[campo])
24
-
25
- prompt = ', '.join(choices)
26
  return prompt
27
 
 
28
  with open('diseño.csv', 'r') as csv_file:
29
  csv_reader = csv.DictReader(csv_file)
30
  valoresUnicos = {}
@@ -43,7 +35,7 @@ output_text = Textbox(label='Prompt')
43
 
44
  title = "Generador Prompts Midjourney"
45
  description = '<div align="center"> Selecciona entre los distintos menús desplegables para generar un Prompt creativo, los que dejes en blanco serán elegidos aleatoriamente.</div>'
46
- inputs = [Dropdown(list(valoresUnicos[campo]), label=campo.capitalize()) for campo in valoresUnicos.keys()]
47
  outputs = output_text
48
 
49
  gr.Interface(fn=concat_prompt, inputs=inputs, outputs=outputs, title=title, description=description).launch()
 
 
1
  from contextlib import nullcontext
 
2
  import csv
3
  import random
4
  import gradio as gr
 
11
  return res
12
 
13
  def concat_prompt(*args):
14
+ campos = {campo: parametroAleatorio(valoresUnicos[campo]) for campo in csv_reader.fieldnames}
15
+ campos.update({campo: valor for campo, valor in zip(csv_reader.fieldnames, args) if valor})
16
+ prompt = ', '.join([campos[campo] for campo in csv_reader.fieldnames])
 
 
 
 
 
 
 
17
  return prompt
18
 
19
+
20
  with open('diseño.csv', 'r') as csv_file:
21
  csv_reader = csv.DictReader(csv_file)
22
  valoresUnicos = {}
 
35
 
36
  title = "Generador Prompts Midjourney"
37
  description = '<div align="center"> Selecciona entre los distintos menús desplegables para generar un Prompt creativo, los que dejes en blanco serán elegidos aleatoriamente.</div>'
38
+ inputs = [Dropdown(list(valoresUnicos[campo]), label=campo.capitalize()) for campo in csv_reader.fieldnames]
39
  outputs = output_text
40
 
41
  gr.Interface(fn=concat_prompt, inputs=inputs, outputs=outputs, title=title, description=description).launch()