Update app.py

Final working version with motivational quote dataset

app.py

@@ -4,31 +4,36 @@ from itertools import chain
 import random
 from difflib import get_close_matches
 
+# Load dataset
 raw_dataset = load_dataset("asuender/motivational-quotes", "quotes_extended", split="train")
 dataset = list(raw_dataset)
 
+# Extract fields
 quotes = [item["quote"] for item in dataset]
 authors = [item["author"] for item in dataset]
-tags_list = [item["tags"].split(", ") if item["tags"] else [] for item in dataset]
-all_tags = sorted(set(chain.from_iterable(tags_list)))
+# 🔁 Try changing "category" to the correct field name if needed
+categories_list = [item["category"].split(", ") if "category" in item else [] for item in dataset]
+all_categories = sorted(set(chain.from_iterable(categories_list)))
 
+# Recommendation logic
 def recommend_quote(user_input):
     if not user_input:
-        return "Please enter a category."
-    
-    matched = get_close_matches(user_input, all_tags, n=1, cutoff=0.5)
+        return "Please enter a category or theme."
+
+    matched = get_close_matches(user_input, all_categories, n=1, cutoff=0.5)
     if not matched:
-        return f"😔 No quotes found for a category like '{user_input}'. Try something else."
+        return f"😕 No quotes found for a category like '{user_input}'. Try something else."
+
+    selected_cat = matched[0]
+    matches = [(q, a) for q, a, c in zip(quotes, authors, categories_list) if selected_cat in c]
 
-    selected_tag = matched[0]
-    matches = [(q, a) for q, a, t in zip(quotes, authors, tags_list) if selected_tag in t]
-    
     if not matches:
-        return f"😔 No quotes found for the tag '{selected_tag}'."
-    
+        return f"😕 No quotes found for the tag '{selected_cat}'."
+
     quote, author = random.choice(matches)
-    return f"{quote}\n\n— {author}"
+    return f"“{quote}”\n— {author}"
 
+# Launch UI
 gr.Interface(
     fn=recommend_quote,
     inputs=gr.Textbox(label="Type a category like 'hope', 'courage', 'love'...", placeholder="e.g. hope"),