added new features like view cache

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import redis
 import numpy as np
 import json
+from datetime import timedelta
 from openai import AzureOpenAI
 from sentence_transformers import SentenceTransformer
 
@@ -15,10 +16,7 @@ redis_client = redis.Redis(
     password=os.getenv("REDIS_PASSWORD")
 )
 
-# 🧹 Do NOT flush DB globally anymore, since multi-user support
-# redis_client.flushdb()
-
-# Azure OpenAI client (only for chat, not embeddings anymore)
+# Azure OpenAI client
 client = AzureOpenAI(
     api_key=os.getenv("AZURE_OPENAI_API_KEY").strip(),
     api_version="2025-01-01-preview",
@@ -30,7 +28,10 @@ CHAT_DEPLOYMENT = "gpt-4.1"
 # 🚀 Super lightweight multilingual embedding model
 embedder = SentenceTransformer("intfloat/multilingual-e5-small")
 
-# Helper: get embedding from HF
+# Cache expiration: 2 days (in seconds)
+CACHE_TTL = int(timedelta(days=2).total_seconds())
+
+# Helper: get embedding
 def get_embedding(text):
     return embedder.encode(text, convert_to_numpy=True).astype(np.float32)
 
@@ -61,11 +62,24 @@ def store_cache(user_id, user_input, output):
         "embedding": vec,
         "output": output
     }))
+    # ⏳ Reset TTL every time user stores something
+    redis_client.expire(cache_key, CACHE_TTL)
 
 def clear_user_cache(user_id):
     cache_key = f"cache:{user_id}"
     redis_client.delete(cache_key)
 
+def view_user_cache(user_id):
+    cache_key = f"cache:{user_id}"
+    entries = redis_client.hgetall(cache_key)
+    if not entries:
+        return "⚠️ No cache stored."
+    lines = []
+    for q, val in entries.items():
+        entry = json.loads(val)
+        lines.append(f"**Q:** {q}\n**A:** {entry['output']}")
+    return "\n\n---\n\n".join(lines)
+
 def chat_with_ai(user_id, user_input):
     if not user_input:
         return "Please type something."
@@ -91,9 +105,13 @@ def chat_with_ai(user_id, user_input):
 
 # Gradio UI
 with gr.Blocks(title="Azure OpenAI + Redis Cloud Chat") as demo:
-    gr.Markdown("# 💬 Azure OpenAI + Redis Cloud (Semantic Cache, Multi-User)")
+    gr.Markdown("# 💬 Azure OpenAI + Redis Cloud (Semantic Cache, Multi-User, Auto Clean)")
 
-    user_id = gr.Textbox(label="User ID", placeholder="Enter your username", value="guest")
+    user_id_state = gr.State("")
+
+    with gr.Row():
+        user_id_input = gr.Textbox(label="Enter Username (only once)", placeholder="Your username")
+        save_user = gr.Button("✅ Save Username")
 
     with gr.Row():
         chatbot = gr.Chatbot(type="messages")
@@ -101,21 +119,39 @@ with gr.Blocks(title="Azure OpenAI + Redis Cloud Chat") as demo:
     with gr.Row():
         msg = gr.Textbox(placeholder="Type your message here...")
         send = gr.Button("Send")
+
+    with gr.Row():
         clear = gr.Button("🧹 Clear Cache")
+        view = gr.Button("👀 View Cache")
+        cache_output = gr.Markdown("")
+
+    def set_user_id(uid):
+        return uid, f"✅ Username set as **{uid}**"
 
     def respond(message, history, user_id):
+        if not user_id:
+            return history, "⚠️ Please set username first!"
         bot_reply = chat_with_ai(user_id, message)
         history.append({"role": "user", "content": message})
         history.append({"role": "assistant", "content": bot_reply})
         return history, ""
 
     def clear_cache_ui(user_id, history):
+        if not user_id:
+            return history, "⚠️ Please set username first!"
         clear_user_cache(user_id)
         return [], f"✅ Cache cleared for {user_id}"
 
-    send.click(respond, [msg, chatbot, user_id], [chatbot, msg])
-    msg.submit(respond, [msg, chatbot, user_id], [chatbot, msg])
-    clear.click(clear_cache_ui, [user_id, chatbot], [chatbot, msg])
+    def view_cache_ui(user_id):
+        if not user_id:
+            return "⚠️ Please set username first!"
+        return view_user_cache(user_id)
+
+    save_user.click(set_user_id, user_id_input, [user_id_state, cache_output])
+    send.click(respond, [msg, chatbot, user_id_state], [chatbot, msg])
+    msg.submit(respond, [msg, chatbot, user_id_state], [chatbot, msg])
+    clear.click(clear_cache_ui, [user_id_state, chatbot], [chatbot, cache_output])
+    view.click(view_cache_ui, user_id_state, cache_output)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860, debug=True, pwa=True)