Update app.py
Browse files
app.py
CHANGED
|
@@ -43,23 +43,20 @@ def validate_token(token):
|
|
| 43 |
except Exception:
|
| 44 |
return False
|
| 45 |
|
| 46 |
-
def get_user_dir(username):
|
| 47 |
-
return f'chatbot_demo_{username}'
|
| 48 |
-
|
| 49 |
def process_files(token, pdf_files, chunk_limit, chunk_separator):
|
| 50 |
-
|
| 51 |
-
|
| 52 |
return "Invalid token. Please enter a valid Hugging Face token."
|
| 53 |
|
| 54 |
user_dir = get_user_dir(username)
|
| 55 |
|
| 56 |
# Initialize Pixeltable
|
| 57 |
-
pxt.drop_dir('
|
| 58 |
-
pxt.create_dir('
|
| 59 |
|
| 60 |
# Create a table to store the uploaded PDF documents
|
| 61 |
t = pxt.create_table(
|
| 62 |
-
|
| 63 |
{'document': pxt.DocumentType(nullable=True),
|
| 64 |
'question': pxt.StringType(nullable=True)}
|
| 65 |
)
|
|
@@ -69,7 +66,7 @@ def process_files(token, pdf_files, chunk_limit, chunk_separator):
|
|
| 69 |
|
| 70 |
# Create a view that splits the documents into smaller chunks
|
| 71 |
chunks_t = pxt.create_view(
|
| 72 |
-
|
| 73 |
t,
|
| 74 |
iterator=DocumentSplitter.create(
|
| 75 |
document=t.document,
|
|
@@ -124,13 +121,12 @@ def process_files(token, pdf_files, chunk_limit, chunk_separator):
|
|
| 124 |
return "Files processed successfully. You can start the discussion."
|
| 125 |
|
| 126 |
def get_answer(token, msg):
|
| 127 |
-
|
| 128 |
-
|
| 129 |
return "Invalid token. Please enter a valid Hugging Face token."
|
| 130 |
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
chunks_t = pxt.get_table(f'{user_dir}.chunks')
|
| 134 |
|
| 135 |
# Insert the question into the table
|
| 136 |
t.insert([{'question': msg}])
|
|
|
|
| 43 |
except Exception:
|
| 44 |
return False
|
| 45 |
|
|
|
|
|
|
|
|
|
|
| 46 |
def process_files(token, pdf_files, chunk_limit, chunk_separator):
|
| 47 |
+
if not validate_token(token):
|
| 48 |
+
|
| 49 |
return "Invalid token. Please enter a valid Hugging Face token."
|
| 50 |
|
| 51 |
user_dir = get_user_dir(username)
|
| 52 |
|
| 53 |
# Initialize Pixeltable
|
| 54 |
+
pxt.drop_dir('chatbot_demo', force=True)
|
| 55 |
+
pxt.create_dir('chatbot_demo')
|
| 56 |
|
| 57 |
# Create a table to store the uploaded PDF documents
|
| 58 |
t = pxt.create_table(
|
| 59 |
+
'chatbot_demo.documents',
|
| 60 |
{'document': pxt.DocumentType(nullable=True),
|
| 61 |
'question': pxt.StringType(nullable=True)}
|
| 62 |
)
|
|
|
|
| 66 |
|
| 67 |
# Create a view that splits the documents into smaller chunks
|
| 68 |
chunks_t = pxt.create_view(
|
| 69 |
+
'chatbot_demo.chunks',
|
| 70 |
t,
|
| 71 |
iterator=DocumentSplitter.create(
|
| 72 |
document=t.document,
|
|
|
|
| 121 |
return "Files processed successfully. You can start the discussion."
|
| 122 |
|
| 123 |
def get_answer(token, msg):
|
| 124 |
+
if not validate_token(token):
|
| 125 |
+
|
| 126 |
return "Invalid token. Please enter a valid Hugging Face token."
|
| 127 |
|
| 128 |
+
t = pxt.get_table('chatbot_demo.documents')
|
| 129 |
+
chunks_t = pxt.get_table('chatbot_demo.chunks')
|
|
|
|
| 130 |
|
| 131 |
# Insert the question into the table
|
| 132 |
t.insert([{'question': msg}])
|