Update app.py

app.py

@@ -2,23 +2,32 @@ import gradio as gr
 from sentence_transformers import SentenceTransformer, util
 import networkx as nx
 
-# Load pre-trained model (runs only once at startup)
+# Load the SentenceTransformer model once at startup
 model = SentenceTransformer("sentence-transformers/paraphrase-mpnet-base-v2")
 
 def group_duplicates_api(complaints, threshold=0.7):
     """
-    Input:
-      complaints: list of complaint strings
-      threshold: float between 0-1 (default 0.7)
-    Output:
-      List of groups, where each group is a list of complaint texts
+    Groups similar/duplicate complaints into clusters.
+    complaints: multiline string or list of strings
+    threshold: similarity score between 0 and 1
     """
+    # Handle empty input
     if not complaints:
         return []
 
+    # If using the textbox input, split by newline
+    if isinstance(complaints, str):
+        complaints = [c.strip() for c in complaints.split("\n") if c.strip()]
+
+    # If fewer than 2 complaints, nothing to compare
+    if len(complaints) < 2:
+        return [[c] for c in complaints]
+
+    # Compute embeddings and cosine similarities
     embeddings = model.encode(complaints, convert_to_tensor=True)
     cosine_scores = util.pytorch_cos_sim(embeddings, embeddings)
 
+    # Build similarity graph
     G = nx.Graph()
     G.add_nodes_from(range(len(complaints)))
 
@@ -27,15 +36,12 @@ def group_duplicates_api(complaints, threshold=0.7):
             if cosine_scores[i][j].item() >= threshold:
                 G.add_edge(i, j)
 
+    # Extract connected components as duplicate groups
     duplicate_groups = list(nx.connected_components(G))
-    results = []
-    for group in duplicate_groups:
-        group_texts = [complaints[idx] for idx in group]
-        results.append(group_texts)
-
+    results = [[complaints[idx] for idx in group] for group in duplicate_groups]
     return results
 
-# Gradio interface
+# Create Gradio interface
 demo = gr.Interface(
     fn=group_duplicates_api,
     inputs=[
@@ -44,8 +50,9 @@ demo = gr.Interface(
     ],
     outputs=gr.JSON(label="Duplicate Groups"),
     title="Duplicate Complaint Grouping API",
-    description="Paste multiple complaints (one per line) and get grouped duplicates."
+    description="Paste multiple complaints (one per line) and get grouped duplicates based on semantic similarity."
 )
 
 if __name__ == "__main__":
     demo.launch()
+