Update app.py

app.py

@@ -6,24 +6,22 @@ import re
 model = SentenceTransformer('all-MiniLM-L6-v2')
 
 def extract_lists(text):
-    # Use regex to find all lists in the format ['item1', 'item2', ...]
-    pattern = r'\[([^\]]+)\]'
-    matches = re.findall(pattern, text)
-    
-    # Process each match into a list of keywords
+    # Split the input text by newlines and process each line
+    lines = text.split('\n')
     lists = []
-    for match in matches:
-        # Split by comma, strip whitespace and quotes, and filter out empty strings
-        keywords = [item.strip().strip("'\"") for item in match.split(',') if item.strip()]
-        if keywords:  # Only add non-empty lists
-            lists.append(keywords)
-    
+    for line in lines:
+        # Strip whitespace and check if the line is not empty
+        line = line.strip()
+        if line:
+            # Split the line by spaces and strip any remaining whitespace
+            keywords = [item.strip() for item in line.split() if item.strip()]
+            if keywords:
+                lists.append(keywords)
     return lists
 
 def compare_embeddings(query, lists_text):
     # Extract lists from the input text
     keyword_lists = extract_lists(lists_text)
-    
     if not keyword_lists:
         return "No valid lists found in the input. Please check the format."
 
@@ -40,7 +38,6 @@ def compare_embeddings(query, lists_text):
 
         # Calculate average similarity for the list
         avg_similarity = similarities.mean().item()
-
         results.append((i, avg_similarity, keywords))
 
     # Sort results by similarity score (descending)
@@ -50,7 +47,7 @@ def compare_embeddings(query, lists_text):
     output = ""
     for i, score, keywords in results:
         output += f"List {i}: Similarity score {score:.4f}\n"
-        output += f"  Keywords: {', '.join(keywords)}\n\n"
+        output += f" Keywords: {' '.join(keywords)}\n\n"
 
     return output
 
@@ -61,13 +58,13 @@ iface = gr.Interface(
         gr.Textbox(label="Query"),
         gr.Textbox(
             label="Lists of keywords",
-            placeholder="Enter lists in the format: ['keyword1', 'keyword2', ...] ['keyword3', 'keyword4', ...]",
+            placeholder="Enter each list of keywords on a new line, with keywords separated by spaces.",
             lines=5
         ),
     ],
     outputs=gr.Textbox(label="Results"),
     title="Keyword Lists Comparison App",
-    description="Compare a query with multiple lists of keywords and find the most relevant lists. Enter each list in square brackets, separated by commas."
+    description="Compare a query with multiple lists of keywords and find the most relevant lists. Enter each list on a new line, with keywords separated by spaces."
 )
 
 # Launch the app