add feature for custom fields

app.py

@@ -5,7 +5,8 @@ import yake
 
 demo = gr.Blocks()
 
-def generate(input, attempted_items, offset):
+def generate(input, attempted_items, offset, custom):
+
     # Step 1: Import
 
     pdfFileObj = open(input.name, 'rb')
@@ -34,6 +35,11 @@ def generate(input, attempted_items, offset):
     for kw in keywords:
         kw_list.append(kw[0])
 
+    if(len(custom)):
+        split_custom = custom.split(';')
+        for kw in split_custom:
+            kw_list.append(kw)
+
     # Step 3: Process for Assignment
     output = {}
     for kw in kw_list:
@@ -110,10 +116,16 @@ def offset_changer(offset_input):
         offset: int(offset_input)
     }
 
+def custom_changer (custom_input):
+    return {
+        custom: custom_input
+    }
+
 with demo:
 
     attempted_items = gr.State(50)
     offset = gr.State(0)
+    custom = gr.State("")
 
     gr.Markdown("# PDF to Index")
 
@@ -133,7 +145,14 @@ with demo:
             attempted_items_input.change(attempted_items_changer, [attempted_items_input], [attempted_items])
             offset_input.change(offset_changer, [offset_input], [offset])
 
-        gr.HTML("<p><em>Attempted Items is the number of terms intended for index (output may be slightly lower), while Page Offset is a value added to each page number found in the file. In the case of invalid values, Attempted Items will default to 50 and Page Offset will default to 0.</em></p>")
+        gr.HTML("<p><em>Attempted Items is the number of terms intended for index (output may be slightly lower), while Page Offset is a value added to each page number found in the file. In the case of invalid values, Attempted Items will default to 50 and Page Offset will default to 0. If the fields do not produce expected values, you may be clicking too quickly -- please try again.</em></p>")
+
+        with gr.Row():
+            custom_input = gr.Textbox(value="", show_label=True, label="Custom")
+            custom_input.change(custom_changer, [custom_input], [custom])
+
+        gr.HTML("<p><em>You can add semicolon-separated values in Custom to add custom fields to index. If the fields do not produce expected values, you may be clicking too quickly -- please try again.</em></p>")
+
 
         gr.Markdown("---")
 
@@ -148,7 +167,7 @@ with demo:
 
     convert_button.click(
         fn=generate,
-        inputs=[uploaded_file, attempted_items, offset],
+        inputs=[uploaded_file, attempted_items, offset, custom],
         outputs=[index],
     )