Update app.py

app.py

@@ -43,17 +43,23 @@ def validate_token(token):
     except Exception:
         return False
 
+def get_user_dir(username):
+    return f'chatbot_demo_{username}'
+
 def process_files(token, pdf_files, chunk_limit, chunk_separator):
-    if not validate_token(token):
+    username = validate_token(token)
+    if not username:
         return "Invalid token. Please enter a valid Hugging Face token."
+
+    user_dir = get_user_dir(username)
     
     # Initialize Pixeltable
-    pxt.drop_dir('chatbot_demo', force=True)
-    pxt.create_dir('chatbot_demo')
+    pxt.drop_dir('user_dir', force=True)
+    pxt.create_dir('user_dir')
 
     # Create a table to store the uploaded PDF documents
     t = pxt.create_table(
-    'chatbot_demo.documents',
+    f'{user_dir}.documents',
     {'document': pxt.DocumentType(nullable=True),
      'question': pxt.StringType(nullable=True)}
     )
@@ -63,7 +69,7 @@ def process_files(token, pdf_files, chunk_limit, chunk_separator):
 
     # Create a view that splits the documents into smaller chunks
     chunks_t = pxt.create_view(
-        'chatbot_demo.chunks',
+        f'{user_dir}.chunks',
         t,
         iterator=DocumentSplitter.create(
             document=t.document,
@@ -118,11 +124,13 @@ def process_files(token, pdf_files, chunk_limit, chunk_separator):
     return "Files processed successfully. You can start the discussion."
 
 def get_answer(token, msg):
-    if not validate_token(token):
+    username = validate_token(token)
+    if not username:
         return "Invalid token. Please enter a valid Hugging Face token."
 
-    t = pxt.get_table('chatbot_demo.documents')
-    chunks_t = pxt.get_table('chatbot_demo.chunks')
+    user_dir = get_user_dir(username)
+    t = pxt.get_table(f'{user_dir}.documents')
+    chunks_t = pxt.get_table(f'{user_dir}.chunks')
 
     # Insert the question into the table
     t.insert([{'question': msg}])