Spaces:
Sleeping
Sleeping
File size: 1,299 Bytes
9ddeec6 e4b0154 9ddeec6 e4b0154 9ddeec6 4722847 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
import sqlite3
import json
class create_db:
def __init__(self, token, chunk_json1, filename, full_content):
conn = sqlite3.connect('my_database.db')
cursor = conn.cursor()
cursor.execute("""
CREATE TABLE IF NOT EXISTS token_data (
token_id TEXT PRIMARY KEY,
chunk_data TEXT,
filename TEXT,
full_content TEXT
)
""")
chunk_json = json.dumps(chunk_json1)
try:
cursor.execute(
"INSERT INTO token_data (token_id, chunk_data, filename, full_content) VALUES (?, ?, ?, ?)",
(token, chunk_json, filename, full_content)
)
conn.commit()
print({"message": f"✅ {filename} uploaded and stored successfully"})
except sqlite3.IntegrityError:
print({"error": f"❌ Token already exists for: {filename}"})
conn.close()
@staticmethod
def get_all_filenames():
conn = sqlite3.connect('my_database.db')
cursor = conn.cursor()
cursor.execute("SELECT filename FROM token_data")
rows = cursor.fetchall()
conn.close()
return {"pdfs": [{"filename": row[0]} for row in rows]}
|