Update app.py

app.py

@@ -3,10 +3,9 @@ import os
 import re
 import csv
 import tempfile
-from difflib import SequenceMatcher
+from rapidfuzz import fuzz
 import datetime
 import gradio as gr
-from rapidfuzz import fuzz
 
 # -----------------------------
 # Config / data loading
@@ -62,10 +61,8 @@ def infer_sentiment(user_text: str) -> str:
 # Retrieval
 # -----------------------------
 def best_match_quote(user_text: str) -> str:
-    """Search all categories, return best fuzzy match, fallback to random from any category."""
     best_score = 0
     best_quote = None
-
     for cat, quotes_list in QUOTES.items():
         for quote_entry in quotes_list:
             quote = quote_entry.get("quote", "")
@@ -75,8 +72,6 @@ def best_match_quote(user_text: str) -> str:
             if score > best_score:
                 best_score = score
                 best_quote = quote
-
-    # Threshold for unknown
     if best_score < 30 or best_quote is None:
         return f"No data about '{user_text}'"
     return best_quote
@@ -179,8 +174,8 @@ with gr.Blocks() as demo:
     clear.click(clear_chat, None, chatbot, queue=False)
 
     upload_btn.upload(upload_json, upload_btn, [upload_status, category])
-    download_json_btn.download(download_current_json)
-    download_csv_btn.download(lambda: download_conversation_csv(chatbot.value))
+    download_json_btn.download(lambda: download_current_json(), file_types=[".json"])
+    download_csv_btn.download(lambda: download_conversation_csv(chatbot.value), file_types=[".csv"])
 
 # -----------------------------
 # Startup log