Update app.py

Final working version with motivational quote dataset

app.py

@@ -1,43 +1,46 @@
 import gradio as gr
 from datasets import load_dataset
 from itertools import chain
-import random
 from difflib import get_close_matches
+import random
 
-# Load dataset
-raw_dataset = load_dataset("asuender/motivational-quotes", "quotes_extended", split="train")
-dataset = list(raw_dataset)
+# Load and clean dataset
+dataset = load_dataset("asudener/motivational-quotes", "quotes_extended", split="train")
 
-# Extract fields
-quotes = [item["quote"] for item in dataset]
-authors = [item["author"] for item in dataset]
-# 🔁 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)))
+quotes = [item["quote"] for item in dataset if "quote" in item]
+authors = [item["author"] if item.get("author") else "Unknown" for item in dataset]
+tags_list = [item["tags"].split(", ") if item.get("tags") else [] for item in dataset]
 
-# Recommendation logic
+all_tags = sorted(set(chain.from_iterable(tags_list)))
+
+# Quote search logic
 def recommend_quote(user_input):
     if not user_input:
-        return "Please enter a category or theme."
+        return "Please enter a category."
 
-    matched = get_close_matches(user_input, all_categories, n=1, cutoff=0.5)
+    matched = get_close_matches(user_input.lower(), all_tags, n=1, cutoff=0.4)
     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, tags in zip(quotes, authors, tags_list)
+        if selected_tag in [t.lower() for t in tags]
+    ]
 
     if not matches:
-        return f"😕 No quotes found for the tag '{selected_cat}'."
+        return f"No quotes found for the tag '{selected_tag}'."
 
     quote, author = random.choice(matches)
-    return f"“{quote}”\n— {author}"
+    return f"\"{quote}\"\n\n— {author}"
 
-# Launch UI
-gr.Interface(
+# UI
+interface = gr.Interface(
     fn=recommend_quote,
-    inputs=gr.Textbox(label="Type a category like 'hope', 'courage', 'love'...", placeholder="e.g. hope"),
+    inputs=gr.Textbox(label="Type a category like 'hope', 'courage', 'resilience'...", placeholder="e.g. hope"),
     outputs="text",
     title="MoodMatch",
     description="Get inspiring quotes by typing any theme or category — even partial matches work."
-).launch()
+)
+
+interface.launch()