Update app.py

app.py

@@ -1,36 +1,23 @@
 import os
-from flask import Flask, render_template, request, jsonify, session
+import uuid
+from flask import Flask, render_template, request, jsonify
 from gradio_client import Client
 
 app = Flask(__name__)
 
-# Required for Flask sessions
-app.secret_key = os.environ.get("FLASK_SECRET_KEY")
-
 client = Client("tonyassi/llm-api", token=os.environ.get("HF_TOKEN"))
 API_NAME = "/chat"
 
+# In-memory store: resets if Space restarts (fine for your use-case)
+HISTORIES = {}
 
-def get_history():
-    # per-session chat history
-    if "chat_history" not in session:
-        session["chat_history"] = []
-    return session["chat_history"]
-
-
-def set_history(history):
-    session["chat_history"] = history
-    session.modified = True
-
-
-def llm_response(user_input: str) -> str:
+def llm_response(session_id: str, user_input: str) -> str:
     user_input = (user_input or "").strip()
     if not user_input:
         return "Type something first"
 
-    history = get_history()
+    history = HISTORIES.get(session_id, [])
     history.append({"role": "user", "content": user_input})
-    set_history(history)
 
     try:
         assistant_text = client.predict(
@@ -41,18 +28,16 @@ def llm_response(user_input: str) -> str:
         if isinstance(assistant_text, str) and "-***-" in assistant_text:
             assistant_text = assistant_text.split("-***-")[0].strip()
 
-        history = get_history()
         history.append({"role": "assistant", "content": assistant_text})
-        set_history(history)
+        HISTORIES[session_id] = history
 
         return assistant_text
 
     except Exception as e:
         # rollback last user message on failure
-        history = get_history()
         if history and history[-1].get("role") == "user":
             history.pop()
-            set_history(history)
+            HISTORIES[session_id] = history
 
         print("LLM API error:", e)
         return "Something isn't working..."
@@ -60,22 +45,29 @@ def llm_response(user_input: str) -> str:
 
 @app.route("/")
 def index():
-    # Refreshing the page hits "/", so this clears the chat
-    session.pop("chat_history", None)
-    return render_template("index.html")
+    # New session id on every page load => refresh starts over
+    session_id = uuid.uuid4().hex
+    HISTORIES.pop(session_id, None)
+    return render_template("index.html", session_id=session_id)
 
 
 @app.route("/send_message", methods=["POST"])
 def send_message():
-    user_input = request.form["user_input"]
-    response = llm_response(user_input)
-    print("history len:", len(session.get("chat_history", [])))
+    user_input = request.form.get("user_input", "")
+    session_id = request.form.get("session_id", "")
+
+    if not session_id:
+        return jsonify({"response": "Missing session_id", "user_input": user_input}), 400
+
+    response = llm_response(session_id, user_input)
     return jsonify({"response": response, "user_input": user_input})
 
 
 @app.route("/reset", methods=["POST"])
 def reset():
-    session.pop("chat_history", None)
+    session_id = request.form.get("session_id", "")
+    if session_id:
+        HISTORIES.pop(session_id, None)
     return jsonify({"ok": True})