Update app.py

app.py

@@ -112,10 +112,29 @@ agent_with_chat_history = RunnableWithMessageHistory(
     input_messages_key="query",
     history_messages_key="chat_history",
 )
+
+def login(message, history):
+    if "Lina" in message.lower():
+        username = "Lina"
+        return username
+    elif "Guest" in message.lower():
+        username = "Guest"
+        return username
+    else:
+        return "Please ask about username.", None
+
 # Gradio chat response function
 
 def respond(message, history: list[dict]):
 
+    with gr.Column():
+            gr.ChatInterface(
+                login,
+                examples=["Guest", "Lina"],
+                additional_outputs=[code],
+                type="messages"
+            )
+
     # Step 1: Embed user message
     try:
         vectorized_input = embedding_model.embed_query(message)
@@ -139,15 +158,16 @@ def respond(message, history: list[dict]):
             matches = pinecone_response.get("matches", [])
             context = "\n".join([match['metadata']['text'] for match in matches if 'metadata' in match])
             print("From Pinecone:", context)
-            pinecone_journal_response = index.query(
-                namespace="Lina",
-                vector=vectorized_input,
-                top_k=3,
-                include_metadata=True
-            )
-            journal_matches = pinecone_journal_response.get("matches", [])
-            journal_context = "\n".join([match['metadata']['text'] for match in journal_matches if 'metadata' in match])
-            print("From Pinecone:", journal_context)
+            if username == "Lina":
+                pinecone_journal_response = index.query(
+                    namespace=username,
+                    vector=vectorized_input,
+                    top_k=3,
+                    include_metadata=True
+                )
+                journal_matches = pinecone_journal_response.get("matches", [])
+                journal_context = "\n".join([match['metadata']['text'] for match in journal_matches if 'metadata' in match])
+                print("From Pinecone:", journal_context)
         except:
             print("Pinecone query failed.")
     else: