Create app.py

app.py

@@ -0,0 +1,58 @@
+from contextlib import nullcontext
+
+import csv
+import os.path
+import random
+import gradio as gr
+from gradio.components import Textbox, Dropdown
+from gradio import Interface, TabbedInterface
+
+
+def parametroAleatorio(lista):
+    res = random.choice(list(lista))
+    while res == '':
+        res = random.choice(list(lista))
+    return res
+
+
+def make_concat_prompt(csv_reader, valoresUnicos):
+    def concat_prompt(*args):
+        campos = {campo: parametroAleatorio(valoresUnicos[campo]) for campo in csv_reader.fieldnames}
+        campos.update({campo: valor for campo, valor in zip(csv_reader.fieldnames, args) if valor})
+
+        prompt = ', '.join([campos[campo] for campo in csv_reader.fieldnames])
+        return prompt
+    return concat_prompt
+
+
+csv_files = [f for f in os.listdir() if os.path.isfile(f) and f.endswith('.csv')]
+csv_files.sort() # Ordenar alfabéticamente
+
+interfaces = []
+tab_labels = []
+for file_name in csv_files:
+    with open(file_name, 'r') as csv_file:
+        csv_reader = csv.DictReader(csv_file)
+        valoresUnicos = {}
+        for campo in csv_reader.fieldnames:
+            valoresUnicos[campo] = set()
+        for row in csv_reader:
+            for campo, valor in row.items():
+                valoresUnicos[campo].add(valor)
+
+        inputs = [Dropdown(list(valoresUnicos[campo]), label=campo.capitalize()) for campo in valoresUnicos.keys()]
+        output_text = Textbox(label='Prompt')
+
+        tab_label = os.path.splitext(file_name)[0].title()
+        tab_labels.append(tab_label)
+        
+        title = f"Generador Prompts Tecnolitas"
+        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>'
+        fn = make_concat_prompt(csv_reader, valoresUnicos)
+
+       
+        interface = Interface(fn=fn, inputs=inputs, outputs=output_text, title=title, description=description)
+        interfaces.append(interface)
+
+tabbed_interface = TabbedInterface(interfaces, tab_labels)
+tabbed_interface.launch()