Update app.py

app.py

@@ -3,6 +3,7 @@ import random
 import os
 import datetime
 import gradio as gr
+import csv
 
 # -----------------------------
 # Config / data loading
@@ -27,42 +28,32 @@ def tokenize(text):
     return set(text.lower().split())
 
 def best_match_quote(category, user_text):
-    """Find the quote with the most token overlap."""
     if category not in QUOTES:
         return None
     pool = QUOTES[category]
     if not pool:
         return None
-
     q_tokens = tokenize(user_text)
     best_score = -1
     best_quote = None
-
     for entry in pool:
         quote_tokens = tokenize(entry["quote"])
         score = len(q_tokens & quote_tokens)
         if score > best_score:
             best_score = score
             best_quote = entry["quote"]
-
-    # fallback
     if not best_quote:
         best_quote = random.choice(pool)["quote"]
-
     return best_quote
 
 def generate_response(category, user_text):
-    """Produce 3-fold response based on category quotes."""
     if category not in QUOTES:
         return ["No quotes found for this category.", "", ""]
-
     pool = QUOTES[category]
     if not pool:
         return ["No quotes available.", "", ""]
-
-    # For simplicity, pick up to 2-3 quotes for summary
     sampled = random.sample(pool, min(2, len(pool)))
-    summary = " ".join([s["quote"].split(".")[0] + "." for s in sampled])  # first sentence
+    summary = " ".join([s["quote"].split(".")[0] + "." for s in sampled])
     details = " ".join([s["quote"] for s in sampled])
     suggested_url = "https://www.example.com/search?q=" + category.replace(" ", "+")
     return [summary, details, suggested_url]
@@ -73,23 +64,20 @@ def generate_response(category, user_text):
 def respond(message, history, category):
     if not message.strip():
         return "", history
-
     summary, details, suggested_url = generate_response(category, message)
-
-    # Append messages in Gradio "messages" format
     history.append({"role": "user", "content": message})
     history.append({"role": "assistant", "content": f"Summary:\n{summary}"})
     history.append({"role": "assistant", "content": f"What real people say:\n{details}"})
     history.append({"role": "assistant", "content": f"Suggested URL:\n{suggested_url}"})
-
     return "", history
 
 def clear_chat():
     return None
 
+# -----------------------------
+# Below Clear button: Downloads / staging
+# -----------------------------
 def download_current_csv(history):
-    """Export current conversation to CSV."""
-    import csv
     tmp = "conversation_export.csv"
     with open(tmp, "w", newline="", encoding="utf-8") as f:
         writer = csv.writer(f)
@@ -98,14 +86,19 @@ def download_current_csv(history):
             writer.writerow([msg["role"], msg["content"]])
     return tmp
 
+def download_current_json():
+    tmp = "quotes_export.json"
+    with open(tmp, "w", encoding="utf-8") as f:
+        json.dump(QUOTES, f, indent=2, ensure_ascii=False)
+    return tmp
+
 # -----------------------------
-# UI
+# UI (lock everything above Clear button)
 # -----------------------------
 with gr.Blocks() as demo:
     gr.Markdown("## 🎓 College Life Chatbot — 3-Fold Responses")
 
     initial_categories = sorted(list(QUOTES.keys()))
-
     with gr.Row():
         category = gr.Dropdown(
             label="Category",
@@ -118,14 +111,21 @@ with gr.Blocks() as demo:
     send = gr.Button("Send")
     clear = gr.Button("Clear")
 
+    # -----------------------------
+    # Downloads / staged responses
+    # -----------------------------
     with gr.Row():
-        download_csv = gr.File(label="Export conversation to CSV")
+        download_csv_btn = gr.File(label="Export conversation to CSV")
+        download_json_btn = gr.File(label="Download current dataset")
 
-    # Wire events
+    # Wire events (lock above Clear button)
     msg.submit(respond, [msg, chatbot, category], [msg, chatbot])
     send.click(respond, [msg, chatbot, category], [msg, chatbot])
     clear.click(clear_chat, None, chatbot, queue=False)
-    download_csv.download(lambda h: download_current_csv(h), chatbot, download_csv)
+
+    # Downloads wiring
+    download_csv_btn.download(lambda h: download_current_csv(h), chatbot, download_csv_btn)
+    download_json_btn.download(download_current_json, None, download_json_btn)
 
 # -----------------------------
 # Startup log