Update app.py

app.py

@@ -30,35 +30,52 @@ autocomplete_data = [str(item).strip('"') for item in autocomplete_data]  # Clea
 # Encode all items in the dataset into embeddings
 autocomplete_embeddings = model.encode(autocomplete_data, convert_to_tensor=True)
 
-# Synonym Expansion Function
+# Function to find synonyms dynamically
 def find_synonym(word, top_n=1):
-    """Finds the most similar word in the dataset based on embeddings."""
     query_embedding = model.encode(word, convert_to_tensor=True)
     results = util.semantic_search(query_embedding, autocomplete_embeddings, top_k=top_n)
     return [autocomplete_data[result['corpus_id']] for result in results[0]]
 
-# Autocomplete function with synonym handling
+# Function to correct spellings
+def correct_spelling(word):
+    matches = process.extract(word, autocomplete_data, scorer=fuzz.ratio, limit=1)
+    best_match, score = matches[0]
+    return best_match if score > 80 else word  # Only correct if confidence > 80%
+
+# Autocomplete function with tracking
 def autocomplete(query):
     if not query.strip():
-        return []  # Avoid empty queries
+        return {"Query": query, "Correction": "None", "Synonym": "None", "Suggestions": []}
+
+    original_query = query.strip()
+    corrected_query = correct_spelling(original_query)
+    synonym_query = find_synonym(corrected_query, top_n=1)[0] if corrected_query != original_query else corrected_query
 
-    # Find synonyms dynamically
-    synonyms = find_synonym(query, top_n=3)
-    
     # Perform fuzzy matching with synonyms included
-    matches = process.extract(query, autocomplete_data + synonyms, scorer=fuzz.partial_ratio, limit=5)
+    matches = process.extract(synonym_query, autocomplete_data, scorer=fuzz.partial_ratio, limit=5)
+    suggestions = [match[0] for match in matches]
+
+    # Detect if spelling correction or synonym replacement occurred
+    correction_status = f"{original_query} → {corrected_query}" if original_query != corrected_query else "None"
+    synonym_status = f"{corrected_query} → {synonym_query}" if corrected_query != synonym_query else "None"
 
-    # Return list of suggestions
-    return [match[0] for match in matches]
+    return {
+        "Query": original_query,
+        "Correction": correction_status,
+        "Synonym": synonym_status,
+        "Suggestions": "\n".join(suggestions)
+    }
 
-# Gradio interface
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("### AI-Powered Autocomplete with Synonyms")
+    gr.Markdown("### AI-Powered Autocomplete with Spell Correction & Synonyms")
 
     query = gr.Textbox(label="Start typing for autocomplete")
-    autocomplete_output = gr.Textbox(label="Autocomplete Suggestions", lines=5, interactive=False)
+    correction_output = gr.Textbox(label="Spelling Correction Applied", interactive=False)
+    synonym_output = gr.Textbox(label="Synonym Applied", interactive=False)
+    suggestions_output = gr.Textbox(label="Autocomplete Suggestions", lines=5, interactive=False)
 
-    # Trigger autocomplete on change
-    query.change(fn=autocomplete, inputs=query, outputs=autocomplete_output)
+    # Bind function to UI
+    query.change(fn=autocomplete, inputs=query, outputs=[correction_output, synonym_output, suggestions_output])
 
 demo.launch()