Update app.py

app.py

@@ -16,12 +16,21 @@ def parse_fashion_query(query):
     result = nlp_ner(query)
 
     structured_output = {}
+    prev_entity = None
+
     for label in result:
         entity = label["entity"]
         word = label["word"]
-        structured_output.setdefault(entity, []).append(word)
-    
-    return {"structured_output": structured_output, "raw_output": result}
+
+        # Merge subwords (handling "##tokens")
+        if word.startswith("##") and prev_entity == entity:
+            structured_output[entity][-1] += word[2:]
+        else:
+            structured_output.setdefault(entity, []).append(word)
+
+        prev_entity = entity
+
+    return structured_output, result  # Now returns two separate outputs
 
 # Gradio UI
 with gr.Blocks() as demo: