Changes

app.py

@@ -50,11 +50,11 @@ def classify_prompt(category,conv_prefix, response):
     #     return 'non refusal' if category=='refusal' else 'unsafe'
     return 'a'
 
-async def add_to_dataset(safe_text,unsafe_text,conv_prefix, response,llama_resp,collinear_resp):
+async def add_to_dataset(conv_prefix, response,llama_resp,collinear_resp):
     conv_prefix = json.loads(conv_prefix)
     dataset = load_dataset("collinear-ai/collinear-guard-demo")
-    new_row = {'safe_text':safe_text,
-            'unsafe_text':unsafe_text,
+    new_row = {'safe_text':"",
+            'unsafe_text':"",
             'conv_prefix':conv_prefix,
             'response':response,
             'llama_resp':{"output":llama_resp},
@@ -76,6 +76,7 @@ async def add_to_dataset(safe_text,unsafe_text,conv_prefix, response,llama_resp,
 
 
 def process_inputs(category,conv_prefix, response_content):
+    print('process inputs')
     response = {"role": "assistant", "content": response_content}
     conv_prefix = json.loads(conv_prefix)
     output = classify_prompt(category,conv_prefix, response)
@@ -148,7 +149,7 @@ with gr.Blocks() as demo:
             outputs=[collinear_output,llama_output]
         ).then(
             fn=add_to_dataset, 
-            inputs=["", "", conv_prefix, response, llama_output, collinear_output],
+            inputs=[conv_prefix, response, llama_output, collinear_output],
             outputs=[]
         )