Update app.py

app.py

@@ -20,23 +20,31 @@ NAME_DICT = {'汤师爷': 'tangshiye', '慕容复': 'murongfu', '李云龙': 'li
              '乔峰': 'qiaofeng', '神里绫华': 'ayaka', '雷电将军': 'raidenShogun', '于谦': 'yuqian'}
 
 
-ai_roles_obj = {}
 
+try:
+  os.makedirs("characters_zip")
+except:
+  pass
+try:
+  os.makedirs("characters")
+except:
+  pass
+ai_roles_obj = {}
 for ai_role_en in NAME_DICT.values():
-    zip_file_path = f"/content/Haruhi-2-Dev/data/character_in_zip/{ai_role_en}.zip"
-    if not os.path.exists(zip_file_path):
-        # os.remove(zip_file_path)
-        print('unfound zip file ', zip_file_path)
-        continue
-
+  file_url = f"https://github.com/LC1332/Haruhi-2-Dev/raw/main/data/character_in_zip/{ai_role_en}.zip"
+  try:
+    os.makedirs(f"characters/{ai_role_en}")
+  except:
+    pass
+  if f"{ai_role_en}.zip" not in os.listdir(f"characters_zip"):
+    destination_file = f"characters_zip/{ai_role_en}.zip"
+    wget.download(file_url, destination_file)
     destination_folder = f"characters/{ai_role_en}"
-
-    with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
+    with zipfile.ZipFile(destination_file, 'r') as zip_ref:
         zip_ref.extractall(destination_folder)
-
-    db_folder = f"./characters/{ai_role_en}/content/{ai_role_en}"
-    system_prompt = f"./characters/{ai_role_en}/content/system_prompt.txt"
-    ai_roles_obj[ai_role_en] = ChatHaruhi(system_prompt=system_prompt,
+  db_folder = f"./characters/{ai_role_en}/content/{ai_role_en}"
+  system_prompt = f"./characters/{ai_role_en}/content/system_prompt.txt"
+  ai_roles_obj[ai_role_en] = ChatHaruhi(system_prompt=system_prompt,
                         llm="openai",
                         story_db=db_folder,
                         verbose=True)