wop commited on
Commit
b5fc4fe
·
verified ·
1 Parent(s): b35606a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -39,11 +39,15 @@ def generate(prompt, history, temperature=0.9, max_new_tokens=2048, top_p=0.95,
39
  def load_database():
40
  try:
41
  with open("database.json", "r", encoding="utf-8") as f:
42
- return json.load(f)
43
- except (FileNotFoundError, json.JSONDecodeError):
44
- print("Error loading database: File not found or invalid format. Creating an empty database.")
 
 
 
45
  return []
46
 
 
47
  def save_database(data):
48
  try:
49
  with open("database.json", "w", encoding="utf-8") as f:
 
39
  def load_database():
40
  try:
41
  with open("database.json", "r", encoding="utf-8") as f:
42
+ data = json.load(f)
43
+ if not isinstance(data, list):
44
+ raise ValueError("Invalid data format")
45
+ return data
46
+ except (FileNotFoundError, json.JSONDecodeError, ValueError):
47
+ print("Error loading database: File not found, invalid format, or empty. Creating an empty database.")
48
  return []
49
 
50
+
51
  def save_database(data):
52
  try:
53
  with open("database.json", "w", encoding="utf-8") as f: