Update app.py

app.py

@@ -41,8 +41,8 @@ def process_input(text, rag, prompt_template):
     output = generate(prompt)
 
     # Generamos el archivo JSON
-    #with open('output.json', 'w') as f:
-    #    json.dump(output, f)
+    with open('output.json', 'w') as f:
+        json.dump(output, f)
     
     return output
 
@@ -56,12 +56,13 @@ def create_interface():
 
     # Definimos la función que se ejecutará cuando se envíe la entrada
     def fn(text, rag, prompt_template):
-        output = process_input(text, rag, prompt_template)
+        
         with open('output.json', 'r') as f:
             data = json.load(f)
         classification_types = [item['clasificacion_tipo'] for item in data]
         classification_types_options = [(option, option) for option in classification_types]
         classification_types_checkboxes = gr.CheckboxGroup(label="Clasificacion Tipo", choices=classification_types_options, interactive = True)
+        output = process_input(text, rag, prompt_template)
         return output, classification_types_checkboxes
         
     examples = [