Allow users to specify citation style

app.py

@@ -53,6 +53,11 @@ with gr.Blocks(css=css_style) as demo:
                                         "User", "Group"], label="Zotero Library Type", value="User", elem_id="zotero-library-type")
             zot_library_id = gr.Textbox(
                 label="Zotero User/Group ID", value=os.getenv('ZOTERO_LIBRARY_ID'))
+            
+            zot_citation_style = gr.Textbox(
+                label="Zotero Citation Style",
+                value='nature',
+            )
 
             zot_selected_col = gr.Radio(
                 [], label="Zotero Collection", elem_id="zotero-collection", visible=False)
@@ -108,6 +113,13 @@ with gr.Blocks(css=css_style) as demo:
                  question, answer, messages, msg_board, zot_selected_col],
         show_progress=False
     )
+    zot_citation_style.change(
+        reset_collection,
+        inputs=[messages],
+        outputs=[zot_selected_col, zot_fetch_col_btn,
+                 question, answer, messages, msg_board, zot_selected_col],
+        show_progress=False
+    )
 
     zot_fetch_col_btn.click(
         fn=fetch_collections,
@@ -126,7 +138,7 @@ with gr.Blocks(css=css_style) as demo:
 
     question.submit(
         fn=handle_submit,
-        inputs=[zot, zot_selected_col, zot_collections, question, messages],
+        inputs=[zot, zot_selected_col, zot_collections, zot_citation_style, question, messages],
         outputs=[messages, msg_board, answer],
         show_progress=False
     )

functions.py

@@ -161,7 +161,7 @@ def reset_collection(messages):
     )
 
 
-def handle_submit(zot, collection_name, collections, question, messages):
+def handle_submit(zot, collection_name, collections, style, question, messages):
     collection_name_only = re.sub('\s\(\d+\)$', '', collection_name)
     messages.set([Message(
         Icons.OK, f"Selected collection: <span style='font-weight: bold'>{collection_name_only}</span>.")])
@@ -257,9 +257,9 @@ def handle_submit(zot, collection_name, collections, question, messages):
                 citation_dict[attachment["key"]] = parents[parent_id]
             else:
                 parent = zot.item(
-                    parent_id, content="bib", style="nature")[0]
+                    parent_id, content="bib", style=style)[0]
                 bib = f"""
-                {html.fragment_fromstring(parent).xpath("normalize-space(div[2])")}
+                {html.fragment_fromstring(parent).xpath("normalize-space(//*)")}
                 <a href="{attachment['links']['alternate']['href']}" target="_blank" class="zotero-link">Open in Zotero</a>
                 """
                 parents[parent_id] = bib