edit lftk format

app.py

@@ -5,6 +5,7 @@ import spacy
 import time
 import os
 import openai
+import json
 
 # Load the Vicuna 7B model and tokenizer
 vicuna_tokenizer = AutoTokenizer.from_pretrained("lmsys/vicuna-7b-v1.3")
@@ -30,9 +31,11 @@ def linguistic_features_fn(message):
     # Use LFTK to dynamically extract handcrafted linguistic features
     extracted_features = LFTK.extract(features = ["a_word_ps", "a_kup_pw", "n_noun"])
 
-    print(extracted_features)
+    formatted_output = json.dumps(extracted_features, indent=2)
 
-    return extracted_features
+    print(formatted_output)
+
+    return formatted_output
 
 def update_api_key(new_key):
     print("update_api_key ran")
@@ -262,6 +265,7 @@ def interface():
             
             ling_ents_btn.click(fn=update_textbox, inputs=ling_ents_prompt, outputs=user_prompt_1, api_name="ling_ents_btn")
 
+            # Show features from LFTK 
             ling_ents_btn.click(linguistic_features_fn, inputs=[ling_ents_prompt], outputs=[linguistic_features_textbox])
 
             # Event Handler for GPT 3.5 Chatbot
@@ -320,6 +324,8 @@ def interface():
 
             # Outputs
             user_prompt_2 = gr.Textbox(label="Original prompt", )
+            linguistic_features_textbox_2 = gr.Textbox(label="Linguistic Features", disabled=True)
+
             gr.Markdown("### Strategy 1 - QA-Based Prompting")
             strategy1 = gr.Markdown("S1", visible=False)
             with gr.Row():
@@ -352,6 +358,9 @@ def interface():
             
             task_btn.click(fn=update_textbox, inputs=task_prompt, outputs=user_prompt_2, api_name="task_btn")
 
+            # Show features from LFTK 
+            task_btn.click(linguistic_features_fn, inputs=[task_prompt], outputs=[linguistic_features_textbox_2])
+
             # Event Handler for GPT 3.5 Chatbot POS/Chunk, user must submit api key before submitting the prompt
             # Will activate after getting API key
             # task_apikey_btn.click(update_api_key, inputs=ling_ents_apikey_input)

requirements.txt

@@ -8,4 +8,5 @@ gpt2
 nltk
 lftk
 spacy
+json
 https://huggingface.co/spacy/en_core_web_sm/resolve/main/en_core_web_sm-any-py3-none-any.whl