add application files

app.py

@@ -1,10 +1,7 @@
 import os
 import gradio as gr
 from model.controller import Controller
-import patoolib
-
-
-# import rarfile
+import zipfile
 
 # # Install unrar-free and make it available for rarfile to use
 # os.system('apt-get update && apt-get install -y unrar-free')
@@ -14,15 +11,11 @@ os.chdir("/home/user/app")
 
 # Download and prepare data files in the correct directory
 os.system('wget -O processed_cases.csv "https://drive.usercontent.google.com/download?id=1jMuQtywo0mbj7ZHCCsyE8xurbSyVVCst&export=download&confirm=t&uuid=2f681c98-86f8-4159-9e03-673cdcbc7cb51"')
-os.system('wget -O chromadb_collection.rar "https://drive.usercontent.google.com/download?id=1-3N-QjK6QMogy2mcNEKjTg_9yIrSjh4A&export=download&confirm=t&uuid=ceed27ad-0f81-4c16-b70f-68cebcc55279"')
-
-# Use patool to extract the .rar file
-patoolib.extract_archive("chromadb_collection.rar", outdir=".")
+os.system('wget -O chromadb_collection.zip "https://drive.usercontent.google.com/download?id=1gz5-gxSlySEtPTzL_VPQ9e8jxHFuL0ZJ&export=download&confirm=t&uuid=de946efb-47b3-435d-b432-3bd5c01c73fb"')
 
-# Use rarfile to extract the .rar file
-rarfile.UNRAR_TOOL = "unrar"
-with rarfile.RarFile("chromadb_collection.rar") as rf:
-    rf.extractall()
+# Use zipfile to extract the .zip file
+with zipfile.ZipFile("chromadb_collection.zip", 'r') as zip_ref:
+    zip_ref.extractall()  # Extracts all files into the current directory
 
 os.system('mv content/chromadb_collections chromadb_collections')
 os.system('rm -r content')