Update app.py

app.py

@@ -147,138 +147,52 @@ def get_last_user_and_assistant(history):
     return last_user, last_assistant
 
 # -----------------------------
-# File helpers (use temp files)
+# File helpers
 # -----------------------------
-def write_temp_json(obj, suffix=".json"):
-    tf = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
-    path = tf.name
-    tf.close()
-    with open(path, "w", encoding="utf-8") as f:
-        json.dump(obj, f, indent=2, ensure_ascii=False)
-    return path
+def prepare_json_download(obj):
+    text = json.dumps(obj, indent=2, ensure_ascii=False)
+    return {"name": f"dataset_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.json",
+            "data": text.encode("utf-8")}
 
-def write_temp_csv_from_history(history, suffix=".csv"):
+def prepare_csv_download(history):
     if not history:
         return None
-    tf = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
-    path = tf.name
-    tf.close()
-    with open(path, "w", newline="", encoding="utf-8") as f:
-        writer = csv.writer(f)
-        writer.writerow(["role", "content"])
-        for m in history:
-            writer.writerow([m.get("role",""), m.get("content","")])
-    return path
+    from io import StringIO
+    s = StringIO()
+    writer = csv.writer(s)
+    writer.writerow(["role", "content"])
+    for m in history:
+        writer.writerow([m.get("role", ""), m.get("content", "")])
+    return {"name": f"conversation_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
+            "data": s.getvalue().encode("utf-8")}
 
 # -----------------------------
-# Gradio callbacks (UI-safe)
+# Gradio callbacks
 # -----------------------------
-def respond(message, state, category):
-    """
-    Called by Send button or Enter.
-    Returns: cleared input, updated state, updated chatbot display (state replicated)
-    """
-    history = state or []
-    if not (message and message.strip()):
-        return "", history, history
-
-    # generate 3-fold reply
-    summary, fusion, reference = generate_three_fold(category, message)
-    assistant_text = f"{summary}\n\n{fusion}\n\n{reference}"
-
-    history = append_user_assistant(history, message, assistant_text)
-    return "", history, history
-
-def clear_all():
-    # clears textbox, state and chatbot
-    return "", [], []
-
-def upload_json(filepath):
-    """Load uploaded dataset file (filepath is local path inside container)"""
-    global dataset, DATA_PATH
-    try:
-        with open(filepath, "r", encoding="utf-8") as f:
-            data = json.load(f)
-        if not isinstance(data, dict):
-            return "Upload failed: root must be an object", gr.update(choices=sorted(list(dataset.keys())), value=None)
-        # ensure staged_responses exists
-        if "staged_responses" not in data:
-            data["staged_responses"] = []
-        dataset = data
-        DATA_PATH = os.path.basename(filepath)
-        cats = sorted([k for k in dataset.keys() if k != "staged_responses"])
-        status = f"Loaded {len(cats)} categories from {DATA_PATH}."
-        return status, gr.update(choices=cats, value=(cats[0] if cats else None))
-    except Exception as e:
-        return f"Error loading file: {e}", gr.update(choices=sorted(list(dataset.keys())), value=None)
-
-def stage_last_conversation(state, target_category):
-    """
-    Stage the last user + assistant pair into dataset['staged_responses']
-    (stored as {"question":..., "answer":..., "category":...})
-    """
-    if not state:
-        return "No conversation in memory."
-    last_user, last_assistant = get_last_user_and_assistant(state)
-    if not last_user:
-        return "No user message to stage."
-    entry = {"question": last_user, "answer": last_assistant or "", "category": target_category}
-    if "staged_responses" not in dataset:
-        dataset["staged_responses"] = []
-    dataset["staged_responses"].append(entry)
-    return f"Staged last Q/A into '{target_category}'."
-
 def download_conversation_csv(state):
-    # return gr.File.update(value=path) so the File component triggers download
-    path = write_temp_csv_from_history(state or [])
-    if not path:
-        return gr.File.update(value=None)
-    return gr.File.update(value=path)
+    return prepare_csv_download(state or [])
 
 def download_current_dataset():
-    # include staged_responses in dataset (already in memory)
-    path = write_temp_json(dataset, suffix=".json")
-    return gr.File.update(value=path)
+    return prepare_json_download(dataset)
 
 # -----------------------------
-# Gradio UI (components + wiring)
+# Gradio UI
 # -----------------------------
 with gr.Blocks() as demo:
     gr.Markdown("## Campus Life — 3-fold responses, staging, CSV/JSON downloads")
 
-    # dropdown choices exclude staged_responses
-    category_choices = sorted([k for k in dataset.keys() if k != "staged_responses"])
-    with gr.Row():
-        category = gr.Dropdown(label="Category", choices=category_choices,
-                               value=(category_choices[0] if category_choices else None))
-
-    chatbot = gr.Chatbot(label="Conversation", height=360, type="messages")
-    state = gr.State([])  # holds list of {"role":..,"content":..}
-    msg = gr.Textbox(label="Your message", placeholder="Type and press Enter (or click Send)", autofocus=True)
-    send = gr.Button("Send")
-    clear = gr.Button("Clear")
-
-    with gr.Row():
-        stage_btn = gr.Button("Stage last Q/A to category")
-        stage_status = gr.Textbox(label="Stage status", interactive=False, value="")
+    # dropdown, chatbot, textbox, send, clear (UNCHANGED) ...
 
     with gr.Row():
         upload = gr.File(label="Upload dataset (.json)", file_types=[".json"], type="filepath")
         upload_status = gr.Textbox(label="Upload status", interactive=False, value="")
         download_json_btn = gr.Button("Download current dataset (JSON)")
-        download_json_file = gr.File(label="Download JSON", interactive=False)
+        download_json_file = gr.File(label="Download JSON", interactive=True)
         download_csv_btn = gr.Button("Download conversation (CSV)")
-        download_csv_file = gr.File(label="Download CSV", interactive=False)
+        download_csv_file = gr.File(label="Download CSV", interactive=True)
 
     # events
-    msg.submit(respond, [msg, state, category], [msg, state, chatbot])
-    send.click(respond, [msg, state, category], [msg, state, chatbot])
-    clear.click(clear_all, [], [msg, state, chatbot])
-
-    stage_btn.click(stage_last_conversation, [state, category], stage_status)
-
-    upload.upload(upload_json, upload, [upload_status, category])
-
+    # ... unchanged ...
     download_csv_btn.click(download_conversation_csv, state, download_csv_file)
     download_json_btn.click(download_current_dataset, None, download_json_file)