charliealonso commited on
Commit
fb243a4
1 Parent(s): ee788fc

Generar pormpts de cada TAB

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -15,12 +15,14 @@ def parametroAleatorio(lista):
15
  return res
16
 
17
 
18
- def concat_prompt(csv_reader, valoresUnicos, *args):
19
- campos = {campo: parametroAleatorio(valoresUnicos[campo]) for campo in csv_reader.fieldnames}
20
- campos.update({campo: valor for campo, valor in zip(csv_reader.fieldnames, args) if valor})
 
21
 
22
- prompt = ', '.join([campos[campo] for campo in csv_reader.fieldnames])
23
- return prompt
 
24
 
25
 
26
  csv_files = [f for f in os.listdir() if os.path.isfile(f) and f.endswith('.csv')]
@@ -36,9 +38,6 @@ for file_name in csv_files:
36
  valoresUnicos[campo] = set()
37
  for row in csv_reader:
38
  for campo, valor in row.items():
39
- if campo not in valoresUnicos:
40
- valoresUnicos[campo] = set()
41
- print(f"Se encontró un nuevo campo {campo} en el archivo {file_name}")
42
  valoresUnicos[campo].add(valor)
43
 
44
  inputs = [Dropdown(list(valoresUnicos[campo]), label=campo.capitalize()) for campo in valoresUnicos.keys()]
@@ -49,7 +48,7 @@ for file_name in csv_files:
49
 
50
  title = f"Generador Prompts Tecnolitas"
51
  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>'
52
- fn = lambda *args: concat_prompt(csv_reader, valoresUnicos, *args)
53
 
54
 
55
  interface = Interface(fn=fn, inputs=inputs, outputs=output_text, title=title, description=description)
 
15
  return res
16
 
17
 
18
+ def make_concat_prompt(csv_reader, valoresUnicos):
19
+ def concat_prompt(*args):
20
+ campos = {campo: parametroAleatorio(valoresUnicos[campo]) for campo in csv_reader.fieldnames}
21
+ campos.update({campo: valor for campo, valor in zip(csv_reader.fieldnames, args) if valor})
22
 
23
+ prompt = ', '.join([campos[campo] for campo in csv_reader.fieldnames])
24
+ return prompt
25
+ return concat_prompt
26
 
27
 
28
  csv_files = [f for f in os.listdir() if os.path.isfile(f) and f.endswith('.csv')]
 
38
  valoresUnicos[campo] = set()
39
  for row in csv_reader:
40
  for campo, valor in row.items():
 
 
 
41
  valoresUnicos[campo].add(valor)
42
 
43
  inputs = [Dropdown(list(valoresUnicos[campo]), label=campo.capitalize()) for campo in valoresUnicos.keys()]
 
48
 
49
  title = f"Generador Prompts Tecnolitas"
50
  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>'
51
+ fn = make_concat_prompt(csv_reader, valoresUnicos)
52
 
53
 
54
  interface = Interface(fn=fn, inputs=inputs, outputs=output_text, title=title, description=description)